Update app.py

app.py

@@ -19,6 +19,8 @@ TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")
 tavily = TavilyClient(api_key=TAVILY_API_KEY)
 
 # --- Helper Functions ---
+import json
+
 def call_llm(messages, model="deepseek/deepseek-chat-v3-0324:free", max_tokens=3500, temperature=0.7):
     url = "https://openrouter.ai/api/v1/chat/completions"
     headers = {
@@ -34,15 +36,18 @@ def call_llm(messages, model="deepseek/deepseek-chat-v3-0324:free", max_tokens=3
         "stream": True
     }
     with requests.post(url, headers=headers, json=data, stream=True) as response:
-        content = ""
-        for line in response.iter_lines():
             if line:
                 decoded = line.decode("utf-8")
                 if decoded.startswith("data: "):
                     piece = decoded.replace("data: ", "").strip()
                     if piece != "[DONE]":
                         try:
-                            content += eval(piece)['choices'][0]['delta'].get('content', '')
+                            parsed = json.loads(piece)
+                            delta = parsed.get("choices", [{}])[0].get("delta", {})
+                            token = delta.get("content", "")
+                            if token:
+                                content += token
+                                yield content
                             yield content
                         except:
                             continue