Update app.py

app.py

@@ -1,11 +1,8 @@
 import streamlit as st
 from gpt_researcher import GPTResearcher
-import asyncio
 import nest_asyncio
+import asyncio
 import os
-from contextlib import contextmanager
-from io import StringIO
-import sys
 
 # Access secrets
 openai_api_key = st.secrets["OPENAI_API_KEY"]
@@ -21,14 +18,8 @@ os.environ["DOC_PATH"] = "./local"  # Path to the folder with documents
 REPORT_TYPE = "research_report"
 
 
-# Function to handle asynchronous calls
-def run_async(coroutine):
-    loop = asyncio.get_event_loop()
-    return loop.run_until_complete(coroutine)
-
-
 # Define the asynchronous function to fetch the report
-async def fetch_report(query, report_type):
+async def fetch_report(query: str, report_type: str) -> str:
     """
     Fetch a research report based on the provided query and report type.
     Research is conducted on a local document.
@@ -37,12 +28,22 @@ async def fetch_report(query, report_type):
         researcher = GPTResearcher(
             query=query, report_type=report_type, report_source="local"
         )
-        await researcher.conduct_research()
-        return await researcher.write_report()
+        research_result = await researcher.conduct_research()
+        report = await researcher.write_report()
+        return report
     except Exception as e:
         return f"Error during research: {str(e)}"
 
 
+def run_report_generation(query, report_type):
+    """
+    Helper function to run async fetch_report function.
+    """
+    loop = asyncio.get_event_loop()
+    report = loop.run_until_complete(fetch_report(query, report_type))
+    return report
+
+
 # Streamlit interface
 st.title("Google Leak Reporting Tool")
 
@@ -55,13 +56,12 @@ query = st.text_area(
 
 # Start the report generation process
 if st.button("Generate Report"):
-    if not query:
+    if not query.strip():
         st.warning("Please enter a query to generate a report.")
     else:
         with st.spinner("Generating report..."):
-            report = run_async(fetch_report(query, REPORT_TYPE))
-            # Display the report or error message
-            if report and not report.startswith("Error"):
+            report = run_report_generation(query, REPORT_TYPE)
+            if "Error during research" not in report:
                 st.success("Report generated successfully!")
                 st.write(report)  # Display the report in the app
                 # Create a download button for the report
@@ -72,4 +72,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