Update app.py

app.py

@@ -2,6 +2,20 @@
 import os
 import logging
 import asyncio
+import nest_asyncio
+from datetime import datetime
+import uuid
+import aiohttp
+import gradio as gr
+
+from langfuse.llama_index import LlamaIndexInstrumentor
+from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
+from llama_index.tools.weather import OpenWeatherMapToolSpec
+from llama_index.tools.playwright import PlaywrightToolSpec
+from llama_index.core.tools import FunctionTool
+from llama_index.core.agent.workflow import AgentWorkflow
+from llama_index.core.workflow import Context
+from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 from llama_index.core.memory import ChatMemoryBuffer
 from llama_index.readers.web import RssReader
 
@@ -15,6 +29,7 @@ nest_asyncio.apply()
 instrumentor = LlamaIndexInstrumentor(
     public_key=os.environ.get("LANGFUSE_PUBLIC_KEY"),
     secret_key=os.environ.get("LANGFUSE_SECRET_KEY"),
+    host=os.environ.get("LANGFUSE_HOST"),
 )
 instrumentor.start()
 
@@ -51,7 +66,6 @@ search_tool = FunctionTool.from_defaults(duck_spec.duckduckgo_full_search)
 # Weather
 openweather_api_key=OPENWEATHERMAP_KEY
 weather_tool_spec = OpenWeatherMapToolSpec(key=openweather_api_key)
-weather_tool_spec = OpenWeatherMapToolSpec(key=openweather_api_key)
 weather_tool = FunctionTool.from_defaults(
     weather_tool_spec.weather_at_location,
     name="current_weather",
@@ -67,10 +81,6 @@ forecast_tool = FunctionTool.from_defaults(
 async def _start_browser():
     return await PlaywrightToolSpec.create_async_playwright_browser(headless=True)
 browser = asyncio.get_event_loop().run_until_complete(_start_browser())
-
-
-
-
 playwright_tool_spec = PlaywrightToolSpec.from_async_browser(browser)
 
 navigate_tool = FunctionTool.from_defaults(
@@ -93,10 +103,6 @@ extract_links_tool = FunctionTool.from_defaults(
 def fetch_google_news_rss():
     docs = RssReader(html_to_text=True).load_data(["https://news.google.com/rss"])
     return [{"title":d.metadata.get("title",""), "url":d.metadata.get("link","")} for d in docs]
-
-
-
-
 google_rss_tool = FunctionTool.from_defaults(
     fn=fetch_google_news_rss,
     name="fetch_google_news_rss",
@@ -107,6 +113,13 @@ google_rss_tool = FunctionTool.from_defaults(
 async def fetch_serper_news(query: str):
     if not serper_api_key:
         raise ValueError("Missing SERPER_API_KEY environment variable")
+    url = f"https://google.serper.dev/news?q={query}&tbs=qdr%3Ad"
+    headers = {"X-API-KEY": serper_api_key, "Content-Type": "application/json"}
+    async with aiohttp.ClientSession() as session:
+        async with session.get(url, headers=headers) as resp:
+            resp.raise_for_status()
+            return await resp.json()
+
 serper_news_tool = FunctionTool.from_defaults(
     fetch_serper_news,
     name="fetch_news_from_serper",
@@ -119,14 +132,41 @@ tools = [
     navigate_tool,
     extract_text_tool,
     extract_links_tool,
+    weather_tool,
+    forecast_tool,
+    google_rss_tool,
+    serper_news_tool,
+]
+web_agent = AgentWorkflow.from_tools_or_functions(tools, llm=llm)
+ctx = Context(web_agent)
+
+# Async helper to run agent queries
+def run_query_sync(query: str):
+    """Helper to run async agent.run in sync context."""
+    return asyncio.get_event_loop().run_until_complete(
+        web_agent.run(query, ctx=ctx)
+    )
+
+async def run_query(query: str):
+    trace_id = f"agent-run-{uuid.uuid4().hex}"
+    try:
+        with instrumentor.observe(
+            trace_id=trace_id,
+            session_id="web-agent-session",
+            user_id=ANON_USER_ID,
+        ):
+            return await web_agent.run(query, ctx=ctx)
+    finally:
+        instrumentor.flush()
 
 # Gradio interface function
 async def gradio_query(user_input, chat_history=None):
-    chat_history = chat_history or []
+    history = chat_history or []
+    history.append({"role": "user", "content": user_input})
     result = await run_query(user_input)
-    response = result.response
-    chat_history.append((user_input, response))
-    return chat_history, chat_history
+    text = result.response if isinstance(result.response, str) else str(result.response)
+    history.append({"role": "assistant", "content": text})
+    return history, history
 
 # Build and launch Gradio app
 grb = gr.Blocks()
@@ -138,9 +178,13 @@ with grb:
         "🙌 Got ideas or improvements? PRs welcome!  \n\n"
         "👉 _Try asking “What’s the weather in Montreal?” or “What’s in the news today?”_"
     )
-    chatbot = gr.Chatbot()  # conversation UI
+    chatbot = gr.Chatbot(type="messages")
     txt = gr.Textbox(placeholder="Ask me anything...", show_label=False)
-    txt.submit(gradio_query, [txt, chatbot], [chatbot, chatbot])
+    txt.submit(
+    gradio_query, 
+    inputs=[txt, chatbot], 
+    outputs=[chatbot, chatbot]  # first for display, second for state
+    )   
     gr.Button("Send").click(gradio_query, [txt, chatbot], [chatbot, chatbot])
 
 if __name__ == "__main__":