Workflow Updates

src/streamlit_app.py

@@ -123,4 +123,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
+