Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,7 +1,8 @@
-import asyncio
-import streamlit as st
-from typing import Dict, Any, List
-from agents import Agent, Runner, trace, set_default_openai_key
+import asyncio # This will helpus to handle tasks without blocking execution
+import streamlit as st  # Build the web application
+from typing import Dict, Any, List  
+from agents import Agent, Runner, trace
+from agents import set_default_openai_key
 from firecrawl import FirecrawlApp
 from agents.tool import function_tool
 
@@ -38,4 +39,44 @@ with st.sidebar:
         st.session_state.firecrawl_api_key = firecrawl_api_key
         set_default_openai_key(firecrawl_api_key)
     
-    
+st.title("Openai Deep Research Agent")
+st.markdown("This openai agent from openi agent SDK performs deep search on any topic")
+
+reasearch_topic = st.text_input("Enter research topic: ")
+
+@function_tool
+async def deep_research(query: str, max_depth: int, time_limit: int, max_urls: int):
+    """
+    Perform comprehensive web research using Firecrawl's deep research endpoint.
+    """
+
+    try:
+        firecrawl_app = FirecrawlApp(api_key = st.session_state.firecrawl_api_key)
+        params = {
+            "maxDepth": max_depth,
+            "timeLimit": time_limit,
+            "maxUrls": max_urls
+        }
+
+    def on_activity(activity):
+        st.write(f"[{activity['type']}]{activity['message']}")
+
+    with st.spinner("Performing Deep Research..."):
+        resp = firecrawl_app.deep_research(
+            query = query,
+            param = param,
+            on_activity = on_activity
+        )
+     return {
+            "success" : True,
+            "final_analysis" : resp["data"]["finalAnalysis"]
+            "sources_count": len(resp["data"]["sources"]),
+            "sources":resp["data"]["sources"]
+        }
+    except Exception as e:
+        st.error(f"Deep Research Error: {str(e)}")
+        return {
+            "error" : str(e),
+            "success" : False
+        }
+