Update app.py

app.py

@@ -89,8 +89,13 @@ async def duckduckgo_chat_stream(model, messages):
                     if line:
                         decoded_line = line.decode('utf-8')
                         if decoded_line.startswith("data: "):
+                            content = decoded_line[5:].strip()
+                            # Check if this is the final marker
+                            if content == "[DONE]":
+                                yield f"data: [DONE]\n\n"
+                                break
                             try:
-                                json_data = json.loads(decoded_line[5:])
+                                json_data = json.loads(content)
                                 message_content = json_data.get("message", "")
                                 if message_content:
                                     # Build OpenAI style streaming response
@@ -101,11 +106,9 @@ async def duckduckgo_chat_stream(model, messages):
                                         "model": model,
                                         "choices": [
                                             {
-                                                "delta": {
-                                                    "content": message_content
-                                                },
+                                                "delta": {"content": message_content},
                                                 "index": 0,
-                                                "finish_reason": None
+                                                "finish_reason": None,
                                             }
                                         ],
                                     }