Update app.py

app.py

@@ -22,11 +22,7 @@ if not openai_api_key or not tavily_api_key:
 # Define the asynchronous function to get the report and capture logs
 async def get_report(query: str, report_type: str, sources: list, report_source: str):
     f = io.StringIO()
-    logs_container = st.empty()
-    logs = ''
-
-    # Create the text area widget once before the loop
-    logs_container.text_area("Agent Logs", logs, height=200, key="realtime_logs_display")
+    logs_container = st.empty()  # Create a placeholder for the logs
 
     with redirect_stdout(f):
         if report_source == 'local':
@@ -35,19 +31,22 @@ async def get_report(query: str, report_type: str, sources: list, report_source:
         else:
             researcher = GPTResearcher(query=query, report_type=report_type, source_urls=sources)
 
+        # Start the research process
         await researcher.conduct_research()
 
         while True:
             logs = f.getvalue()
-            # Update the content of the text area via session_state
-            st.session_state.realtime_logs_display = logs
-            await asyncio.sleep(1)  # Update every second
+            # Update the text area within the placeholder
+            logs_container.text_area("Agent Logs", logs, height=200)
+            await asyncio.sleep(1)  # Wait for a second before the next update
             if "Finalized research step" in logs:
                 break
 
+        # Generate the final report
         report = await researcher.write_report()
     return report, f.getvalue()
 
+
 # Function to create PDF using fpdf with UTF-8 encoding
 class PDF(FPDF):
     def header(self):