Update app.py

app.py

@@ -7,6 +7,7 @@ from contextlib import redirect_stdout
 import io
 from fpdf import FPDF
 from datetime import datetime
+import uuid
 
 # Apply nest_asyncio for asyncio support in Streamlit
 nest_asyncio.apply()
@@ -22,7 +23,8 @@ 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()
+    unique_key = str(uuid.uuid4())  # Generate a unique key for this run
+    
     with redirect_stdout(f):
         if report_source == 'local':
             # Set the DOC_PATH environment variable
@@ -33,15 +35,21 @@ async def get_report(query: str, report_type: str, sources: list, report_source:
 
         await researcher.conduct_research()
 
-        while True:
+        max_attempts = 30  # Prevent infinite loop
+        attempts = 0
+        while attempts < max_attempts:
             logs = f.getvalue()
-            logs_container.text_area("Agent Logs", logs, height=200, key="realtime_logs_display_unique")
-            await asyncio.sleep(1)  # Update every second
+            
+            # Break condition
             if "Finalized research step" in logs:
                 break
+            
+            await asyncio.sleep(1)  # Update every second
+            attempts += 1
 
         report = await researcher.write_report()
-    return report, f.getvalue()
+    
+    return report, logs
 
 # Function to create PDF using fpdf with UTF-8 encoding
 class PDF(FPDF):
@@ -54,16 +62,6 @@ class PDF(FPDF):
         self.set_font("Arial", "I", 8)
         self.cell(0, 10, f"Page {self.page_no()}", 0, 0, "C")
 
-    def chapter_title(self, title):
-        self.set_font("Arial", "B", 12)
-        self.cell(0, 10, title, 0, 1, "L")
-        self.ln(5)
-
-    def chapter_body(self, body):
-        self.set_font("Arial", "", 12)
-        self.multi_cell(0, 10, body)
-        self.ln()
-
 def create_pdf(report_text, pdf_path):
     pdf = PDF()
     pdf.add_page()
@@ -187,14 +185,14 @@ if 'report' in st.session_state:
 
 st.markdown("### Agent Logs")
 if 'logs' in st.session_state:
-    st.text_area("Logs will appear here during the research process.", 
-                 st.session_state.logs, 
+    st.text_area("Logs will appear here during the research process:", 
+                 value=st.session_state.logs, 
                  height=200, 
-                 key="final_logs_display_unique")
+                 key=f"logs_{uuid.uuid4()}")
 else:
-    st.text_area("Logs will appear here during the research process.", 
+    st.text_area("Logs will appear here during the research process", 
                  height=200, 
-                 key="final_logs_display_alternative")
+                 key=f"logs_{uuid.uuid4()}")
 
 # Hide Streamlit's default footer and menu
 hide_streamlit_style = """