Update app.py

app.py

@@ -25,42 +25,33 @@ def web_search(query):
     response = requests.post(url, json=payload)
 
     if response.status_code == 200:
-        results = response.json()
-        if isinstance(results, list):
-            return results
-        else:
-            return results.get('results')  # Assuming 'results' contain relevant info
+        return response.json().get('results')  # Assuming 'results' contain relevant info
     else:
         return {"error": f"Error: {response.status_code}"}
 
 # Main function
 def main():
-    st.title("AI Research Assistant")
-    st.write("This tool helps you generate a research report based on the text obtained from a web search.")
-    
-    query = st.text_area("Enter your search query:", value="", height=200, max_chars=None, key=None)
-    
-    if st.button("Generate Report"):
-        if query:
-            # Perform web search
-            search_results = web_search(query)
-            
-            # Extract relevant information and convert it into a message
-            message = ""  # Create an empty string to store the message
-            for result in search_results:
-                # Assuming each result is a string, you might adjust this part accordingly
-                message += result + ". "  # Concatenate results with a period
-            
-            # Combine prompt and message for AI conversation
-            full_message = f"Generate a research report based on the following text: {message}"
-            
-            # Pass the combined message to the AI for generating a report
-            report = chat_with_ai(full_message)
-            
-            # Display the report
-            st.write(report)
-        else:
-            st.write("Please enter a search query.")
+    query = st.text_area("Enter resume content:")
+    if query:
+        prompt = f"Please provide a comprehensive review of the following resume and suggest improvements. The goal is to enhance its effectiveness in conveying the candidate's skills and experiences. Here's the resume content: {query}"
+        
+        # Perform web search
+        search_results = web_search(query)
+        
+        # Extract relevant information and convert it into a message
+        message = ""  # Create an empty string to store the message
+        for result in search_results:
+            # Assuming each result is a string, you might adjust this part accordingly
+            message += result + ". "  # Concatenate results with a period
+        
+        # Combine prompt and message for AI conversation
+        full_message = prompt + " " + message
+        
+        # Pass the combined message to the AI for generating a report
+        report = chat_with_ai(full_message)
+        
+        # Display the report
+        st.write(report)
 
 if __name__ == "__main__":
-    main()
+    main()