Update Gradio_UI.py

Gradio_UI.py

@@ -193,11 +193,19 @@ class GradioUI:
         import gradio as gr
 
         messages.append(gr.ChatMessage(role="user", content=prompt))
-        yield messages
-        for msg in stream_to_gradio(self.agent, task=prompt, reset_agent_memory=False):
-            messages.append(msg)
-            yield messages
-        yield messages
+        yield messages  # Show user message first
+
+        try:
+            for msg in stream_to_gradio(self.agent, task=prompt, reset_agent_memory=False):
+                messages.append(msg)
+                yield messages  # Stream agent messages one by one
+        except Exception as e:
+            error_msg = f"⚠️ Error: {str(e)}"
+            print(error_msg)  # Print in logs
+            messages.append(gr.ChatMessage(role="system", content=error_msg))
+            yield messages  # Display error in chat
+
+        yield messages  # Final yield to ensure UI updates
 
     def upload_file(
         self,