Update app.py

app.py

@@ -6,9 +6,9 @@ import requests
 session = requests.Session()
 
 # Function to interact with the AI
-def chat_with_ai(message):
+def chat_with_ai(message, user_input):
     api_url = os.getenv("CHAT_API_URL")
-    payload = {"message": message}
+    payload = {"message": message, "user_input": user_input}
 
     try:
         with session.post(api_url, json=payload) as response:
@@ -65,9 +65,9 @@ def main():
                 search_results = web_search(query)
             
             # Pass the search results to the AI for generating a report
-            prompt = f"Generate a research report based on the following information: {search_results}. If the search results are insufficient, answer the user's question using the information available."
+            prompt = f"Generate a research report based on the following information: {search_results}. The user's query was: '{query}'. If the search results are insufficient, answer the user's question using the information available."
             with st.spinner('🔄 Generating report...'):
-                report = chat_with_ai(prompt)
+                report = chat_with_ai(prompt, query)
             
             # Display the report
             st.write(report)