Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import requests
+import time
 
 # Create a session for reusing connections
 session = requests.Session()
@@ -25,33 +26,20 @@ def web_search(query):
     response = requests.post(url, json=payload)
 
     if response.status_code == 200:
-        return response.json().get('results')  # Assuming 'results' contain relevant info
+        return response.json()
     else:
         return {"error": f"Error: {response.status_code}"}
 
 # Main function
 def main():
-    query = st.text_area("Enter resume content:")
+    query = st.text_input("Enter your research query: ")
     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)
-        
+        # Pass the search results to the AI for generating a report
+        report = chat_with_ai(search_results)
         # Display the report
         st.write(report)
 
 if __name__ == "__main__":
-    main()
+    main()