Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import requests
 import sseclient
 
-def stream_response(prompt):
+def chat(prompt):
     url = "https://wlchee-mcp-sentiment.hf.space/gradio_api/mcp/sse"
     headers = {
         "Accept": "text/event-stream",
@@ -11,18 +11,19 @@ def stream_response(prompt):
     payload = {"input": prompt}
 
     try:
-        # Make a streaming request
-        response = requests.post(url, json=payload, stream=True, headers=headers)
+        # Send streaming request to MCP server
+        response = requests.post(url, json=payload, headers=headers, stream=True)
         client = sseclient.SSEClient(response)
 
-        full_output = ""
+        output = ""
         for event in client.events():
             if event.data == "[DONE]":
                 break
-            full_output += event.data
-            yield full_output  # Gradio can stream this in real-time
+            output += event.data
+            yield output
     except Exception as e:
         yield f"Error: {e}"
 
-demo = gr.Interface(fn=stream_response, inputs="text", outputs="text", live=True)
-demo.launch()
+# Launch Tiny Agent UI
+gr.ChatInterface(chat, title="Sentiment Tiny Agent").launch()
+