Update app.py

app.py

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