Update app.py

app.py

@@ -11,21 +11,24 @@ def chat(message, history):
     }
     payload = json.dumps({"input": message})
 
+    output = ""
     try:
         response = requests.post(url, data=payload, headers=headers, stream=True)
         client = sseclient.SSEClient(response)
 
-        output = ""
-        try:
-            for event in client.events():
-                if event.data == "[DONE]":
-                    break
-                output += event.data
-                yield [{"role": "assistant", "content": output}]
-        except (StopIteration, StopAsyncIteration):
-            # Graceful end of generator
-            return
+        for event in client.events():
+            if event.data == "[DONE]":
+                break
+            output += event.data
+            # Always yield messages in OpenAI format
+            yield [{"role": "assistant", "content": output}]
+
+        # If no output was produced at all
+        if output.strip() == "":
+            yield [{"role": "assistant", "content": "⚠️ No response received from the MCP server."}]
+
     except Exception as e:
-        yield [{"role": "assistant", "content": f"Error: {e}"}]
+        yield [{"role": "assistant", "content": f"❌ Error: {str(e)}"}]
 
-gr.ChatInterface(fn=chat, title="Sentiment Tiny Agent", type="messages").launch(share=True)
+# Launch without share=True in a Space
+gr.ChatInterface(fn=chat, title="Sentiment Tiny Agent", type="messages").launch()