Update app.py

app.py

@@ -315,13 +315,20 @@ async def run_query(query: str):
         instrumentor.flush()
 
 # Gradio interface function
-async def gradio_query(user_input, chat_history=None):
+async def gradio_query_stream(user_input, chat_history=None):
     history = chat_history or []
     history.append({"role": "user", "content": user_input})
-    result = await run_query(user_input)
-    text = result.response if isinstance(result.response, str) else str(result.response)
-    history.append({"role": "assistant", "content": text})
-    return history, history
+    # Start streaming response from the agent
+    result = await web_agent.run(user_input, ctx=ctx, streaming=True)
+    partial = ""
+    async for chunk in result.response_gen:
+        # If chunk is an object, use chunk.delta or chunk.text
+        partial += chunk
+        if len(history) > 1 and history[-1]["role"] == "assistant":
+            history[-1]["content"] = partial
+        else:
+            history.append({"role": "assistant", "content": partial})
+        yield history, history
 
 # Build and launch Gradio app
 grb = gr.Blocks()
@@ -332,17 +339,18 @@ with grb:
         This bot can check the news, tell you the weather, and even browse websites to answer follow-up questions — all powered by a team of tiny AI tools working behind the scenes.\n\n
         🧪 Built for fun during the [AI Agents course](https://huggingface.co/learn/agents-course/unit0/introduction) — it's just a demo to show what agents can do.\n
         🙌 Got ideas or improvements? PRs welcome!\n\n
-        👉 _Try asking 'What's the weather in Montreal?' or 'What's in the news today?'
+        👉 Try asking 'What's the weather in Montreal?' or 'What's in the news today?'
         """
     )
     chatbot = gr.Chatbot(type="messages")
     txt = gr.Textbox(placeholder="Ask me anything...", show_label=False)
     txt.submit(
-    gradio_query, 
+    gradio_query_stream, 
     inputs=[txt, chatbot], 
-    outputs=[chatbot, chatbot]  # first for display, second for state
+    outputs=[chatbot, chatbot],
+    stream=True
     )   
-    gr.Button("Send").click(gradio_query, [txt, chatbot], [chatbot, chatbot])
+    gr.Button("Send").click(gradio_query_stream, [txt, chatbot], [chatbot, chatbot])
 
 if __name__ == "__main__":
     grb.launch()