Update app.py

app.py

@@ -19,7 +19,6 @@ os.environ['DOC_PATH'] = './local'  # Path to the folder with documents
 
 # Constants
 REPORT_TYPE = "research_report"
-DOCUMENT_FILE = 'removed_code.txt'  # Name of the document file
 
 # Function to capture output to the standard output
 @contextmanager
@@ -41,21 +40,23 @@ def run_async(coroutine):
 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 specified by DOCUMENT_FILE.
+    Research is conducted on a local document.
     """
-    researcher = GPTResearcher(query=query, report_type=report_type, report_source='local')
-    await researcher.conduct_research()
-    report = await researcher.write_report()
-    return report
+    try:
+        researcher = GPTResearcher(query=query, report_type=report_type, report_source='local')
+        await researcher.conduct_research()
+        return await researcher.write_report()
+    except Exception as e:
+        return f"Error during research: {str(e)}"
 
 # Streamlit interface
 st.title("Google Leak Reporting Tool")
 
-# User input for the query using a text area for multi-line input
+# User input for the query using a text area
 query = st.text_area(
     "Enter your research query:",
     "Extract all the information about how the ranking for internal links works.",
-    height=150  # You can adjust the height as needed
+    height=150  # Adjustable height
 )
 
 # Placeholder for the progress expander
@@ -67,16 +68,15 @@ if st.button("Generate Report"):
     if not query:
         st.warning("Please enter a query to generate a report.")
     else:
-        # Start an asynchronous task for generating the report
+        # Display initial progress information
         progress_placeholder.info("Starting research...")
         # Run the research asynchronously and capture output
-        with st_redirect(progress_placeholder.code):
+        with st_capture(progress_placeholder.code):
             report = run_async(fetch_report(query, REPORT_TYPE))
         
-        if report:
+        if report and not report.startswith("Error"):
             st.success("Report generated successfully!")
             st.write(report)  # Display the report in the app
-
             # Create a download button for the report
             st.download_button(
                 label="Download Report as Text File",
@@ -85,4 +85,4 @@ if st.button("Generate Report"):
                 mime="text/plain"
             )
         else:
-            st.error("Failed to generate the report.")
+            st.error(report)  # Show the error message if any