handle the async generator

app.py

@@ -495,7 +495,7 @@ def create_application():
         updated_history = chat_history + [{"role": "user", "content": user_input}]
         return "", updated_history
     
-    def bot_response(chat_history: List[Dict]) -> Tuple[List[Dict], Dict]:
+    async def bot_response(chat_history: List[Dict]) -> Tuple[List[Dict], Dict]:
         """Get bot response and update chat history."""
         if not chat_history or not chat_history[-1].get("role") == "user":
             return chat_history, gr.update(visible=False)
@@ -513,7 +513,11 @@ def create_application():
                 old_format[-1][1] = msg["content"]
         
         # Call the agent and get the response
-        return stream_agent_response(question, old_format[:-1])
+        # We need to consume the async generator and return the last value
+        last_response = None
+        async for response in stream_agent_response(question, old_format[:-1]):
+            last_response = response
+        return last_response
     
     # Event handlers
     with demo: