Update app.py

app.py

@@ -15,9 +15,9 @@ def chat_with_ai(message, user_input):
             if response.status_code == 200:
                 return response.json().get('response')  # Access 'response' key
             else:
-                return {"error": "Failed to get a response from the AI API. 😞"}
+                st.error("Failed to get a response from the AI API. 😞")
     except requests.RequestException as e:
-        return {"error": f"Error: {e}"}
+        st.error(f"Error: {e}")
 
 # Function to perform web search
 def web_search(query):
@@ -28,7 +28,7 @@ def web_search(query):
     if response.status_code == 200:
         return response.json()
     else:
-        return {"error": f"Error: {response.status_code}"}
+        st.error(f"Error: {response.status_code}")
 
 # Main function
 def main():
@@ -65,12 +65,15 @@ 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}. The user's query was: '{query}'. If the search results are insufficient, answer the user's question using the information available."
+            prompt = f"Generate a detailed research report based on the following information: {search_results}. The user's query was: '{query}'. Please include an overview, key findings, and any relevant details in the report. 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, query)
             
             # Display the report
-            st.write(report)
+            if isinstance(report, dict) and 'error' in report:
+                st.error(report['error'])
+            else:
+                st.write(report)
 
             # Update recent reports
             recent_reports.text(query)