Update app.py

app.py

@@ -3,9 +3,8 @@ from gpt_researcher import GPTResearcher
 import asyncio
 import nest_asyncio
 import os
-from contextlib import contextmanager
-from io import StringIO
-import sys
+from concurrent.futures import ThreadPoolExecutor
+import threading
 
 # Access secrets
 openai_api_key = st.secrets["OPENAI_API_KEY"]
@@ -20,18 +19,15 @@ os.environ["DOC_PATH"] = "./local"  # Path to the folder with documents
 # Constants
 REPORT_TYPE = "research_report"
 
-
-# Function to handle asynchronous calls
-def run_async(coroutine):
-    loop = asyncio.get_event_loop()
-    return loop.run_until_complete(coroutine)
+# ThreadPoolExecutor to handle asynchronous tasks
+executor = ThreadPoolExecutor(max_workers=4)  # Adjust the number of workers as needed
 
 
 # Define the asynchronous function to fetch the report
 async def fetch_report(query, report_type):
     """
     Fetch a research report based on the provided query and report type.
-    Research is conducted on a local document.
+    Research is conducted on a local document related to Google Search Algorithm Leak.
     """
     try:
         researcher = GPTResearcher(
@@ -43,6 +39,13 @@ async def fetch_report(query, report_type):
         return f"Error during research: {str(e)}"
 
 
+# Function to run the asynchronous function in a separate thread
+def run_async(coroutine):
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    return loop.run_until_complete(coroutine)
+
+
 # Streamlit interface
 st.title("Google Leak Reporting Tool")
 
@@ -59,7 +62,10 @@ if st.button("Generate Report"):
         st.warning("Please enter a query to generate a report.")
     else:
         with st.spinner("Generating report..."):
-            report = run_async(fetch_report(query, REPORT_TYPE))
+            # Submit the task to the ThreadPoolExecutor
+            future = executor.submit(run_async, fetch_report(query, REPORT_TYPE))
+            # Wait for the result
+            report = future.result()
             # Display the report or error message
             if report and not report.startswith("Error"):
                 st.success("Report generated successfully!")
@@ -72,4 +78,4 @@ if st.button("Generate Report"):
                     mime="text/plain",
                 )
             else:
-                st.error(report)  # Show the error message if any
+                st.error(report)  # Show the error message if any