Update app.py

app.py

@@ -180,27 +180,57 @@ workflow = AgentWorkflow(
 )
 ctx = Context(workflow)
 
-# Sync wrapper
-async def respond(query: str) -> str:
-    out = await workflow.run(user_msg=query, ctx=ctx, memory=memory)
-    return out.response.blocks[0].text
+# # Sync wrapper
+# async def respond(query: str) -> str:
+#     out = await workflow.run(user_msg=query, ctx=ctx, memory=memory)
+#     return out.response.blocks[0].text
     
-# Async response handler for Gradio
-async def respond_gradio(query, chat_history):
-    answer = await respond(query)
-    return chat_history + [[query, answer]]
+# # Async response handler for Gradio
+# async def respond_gradio(query, chat_history):
+#     answer = await respond(query)
+#     return chat_history + [[query, answer]]
+
+# New unified respond() function
+async def respond(message, history):
+    # (Optional) Format history if needed by your agent system
+    formatted_history = []
+    for user_msg, bot_msg in history:
+        if user_msg:  # User input
+            formatted_history.append({"role": "user", "content": user_msg})
+        if bot_msg:  # Bot output
+            formatted_history.append({"role": "assistant", "content": bot_msg})
+
+    # Run the agent workflow
+    out = await workflow.run(user_msg=message, ctx=ctx, memory=memory)
+    return out.response.blocks[0].text
 
 # --- Gradio UI ---
+# with gr.Blocks() as demo:
+#     gr.Markdown("## 🤖 Perspicacity")
+#     gr.Markdown(
+#     "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 agents working behind the scenes.  \n\n"
+#     "🧪 Built for fun during the [AI Agents course](https://huggingface.co/learn/agents-course/unit0/introduction) at Hugging Face — it's just a demo to show what agents can do.  \n"
+#     "🙌 Got ideas or improvements? PRs welcome!"
+#     )
+#     chatbot = gr.Chatbot()
+#     txt = gr.Textbox(placeholder="Ask me about news, weather, etc…")
+
+#     txt.submit(respond_gradio, inputs=[txt, chatbot], outputs=chatbot)
+
+# demo.launch()
+
 with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 Perspicacity")
+    gr.Markdown("## 🗞️ Multi‐Agent News & Weather Chatbot")
     gr.Markdown(
-    "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 agents working behind the scenes.  \n\n"
-    "🧪 Built for fun during the [AI Agents course](https://huggingface.co/learn/agents-course/unit0/introduction) at Hugging Face — it's just a demo to show what agents can do.  \n"
-    "🙌 Got ideas or improvements? PRs welcome!"
+        "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 agents 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?”_"
     )
+
     chatbot = gr.Chatbot()
-    txt = gr.Textbox(placeholder="Ask me about news, weather, etc…")
+    txt = gr.Textbox(placeholder="Ask me about news, weather or anything…")
 
-    txt.submit(respond_gradio, inputs=[txt, chatbot], outputs=chatbot)
+    txt.submit(respond, inputs=[txt, chatbot], outputs=chatbot)
 
 demo.launch()