Update app.py

app.py

@@ -58,22 +58,26 @@ query = st.text_area(
     height=150  # You can adjust the height as needed
 )
 
+# Placeholder for the progress expander
+progress_expander = st.expander("See research progress", expanded=True)
+progress_placeholder = progress_expander.empty()
 
-# Button to generate report
+# Start the report generation process
 if st.button("Generate Report"):
     if not query:
         st.warning("Please enter a query to generate a report.")
     else:
-        # Collapsible expander to show progress
-        with st.expander("See research progress"):
-            with st_capture(st.write):
-                report = run_async(fetch_report(query, "research_report"))
+        # Start an asynchronous task for generating the report
+        progress_placeholder.info("Starting research...")
+        # Run the research asynchronously and capture output
+        with st_redirect(progress_placeholder.code):
+            report = run_async(fetch_report(query, REPORT_TYPE))
         
         if report:
             st.success("Report generated successfully!")
             st.write(report)  # Display the report in the app
-            
-            # Create a download button and provide the report as a downloadable file
+
+            # Create a download button for the report
             st.download_button(
                 label="Download Report as Text File",
                 data=report,