Update app.py

app.py

@@ -440,9 +440,14 @@ def chat():
                 if response.status_code == 200:
                     for line in response.iter_lines():
                         if line:
-                            chunk = json.loads(line)
-                            if 'message' in chunk and 'content' in chunk['message']:
-                                yield json.dumps({"token": chunk['message']['content'], "source": "colab", "provider": "Ollama (Phi-3)", "tools_used": tools_needed}) + "\n"
+                            try:
+                                chunk = json.loads(line)
+                                if 'message' in chunk and 'content' in chunk['message']:
+                                    token = chunk['message']['content']
+                                    yield json.dumps({"token": token, "source": "colab", "provider": "Nexa AI (Colab)", "tools_used": tools_needed}) + "\n"
+                            except Exception as e:
+                                logger.error(f"Error parsing Ollama chunk: {e}")
+                                continue
                     return
             except Exception as colab_err:
                 logger.warning(f"Colab streaming failed: {colab_err}. Switching to HF...")