Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,14 +6,13 @@ from agents import set_default_openai_key
 from firecrawl import FirecrawlApp
 from agents.tool import function_tool
 
-#Setup page configuration using streamlit
+# Setup the page configuration (done using streamlit)
 st.set_page_config(
-    page_title = "OpenAI based deep search agent",
-    page_icon = "",
+    page_title = "OpenAI based Deep Research Agent",
+    page_icon = "📚",
     layout = "wide"
 )
 
-
 # Initialize session state for API Key if don't exist
 if "openai_api_key" not in st.session_state:
     st.session_state.openai_api_key = ""
@@ -121,4 +120,34 @@ elaboration_agent = Agent(
     4. Preserve the original structure while making it more comprehensive
     5. Ensure all additions are relevant and valuable to the topic.
     """
-)
+)
+
+async def run_research_process(topic : str):
+    """Run the complete research process"""
+    # Step 1 - Intial Research
+    with st.spinner("Conducting initial research..."):
+        research_results = await Runner.run(research_agent, topic)
+        initial_report = research_results.final_output
+    # Display initial report 
+    with st.expander("View Initial Research Report"):
+        st.markdown(initial_report)
+
+        
+    # Step 2 - Enhance the report
+    with st.spinner("Enhancing the report with additional information..."):
+        elaboration_input = f"""
+        RESEARCH_TOPIC = {topic}
+        INITIAL RESEARCH REPORT:
+        {initial_report}
+        Please enhance this research report with additional information,
+        examples, case studies, and deeper insights while maintaining its
+        academic rigor and factual accuracy.
+        """
+
+        elaboration_results = await Runner.run(elaboration_agent, elaboration_input)
+        enhanced_report = elaboration_results.final_output
+        
+    return enhanced_report
+
+
+