Update app.py

app.py

@@ -16,15 +16,16 @@ def chat(message, history):
         client = sseclient.SSEClient(response)
 
         output = ""
-        for event in client.events():
-            if event.data == "[DONE]":
-                break
-            output += event.data
-            # Yield as a list of message dicts in the new format
-            yield [{"role": "assistant", "content": 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
     except Exception as e:
         yield [{"role": "assistant", "content": f"Error: {e}"}]
 
-# Launch with type="messages" to use the new message format
 gr.ChatInterface(fn=chat, title="Sentiment Tiny Agent", type="messages").launch(share=True)
-