Update app.py

app.py

@@ -112,7 +112,21 @@ def respond(message, api_key, max_tokens, top_p, temperature):
 
                 print(f"Received line: {line}")
 
-                if line.startswith("data:"):
+              if line.startswith("data:"):
+                  line_content = line[5:].strip()  
+                  if not line_content:
+                     continue  
+
+                  try:
+                      chunk = json.loads(line_content)  
+                      chunk_message = chunk.get("chunk", {}).get("content", "")
+                      assistant_reply += chunk_message 
+                     yield assistant_reply
+                 except json.JSONDecodeError as e:
+                      print(f"Stream chunk error: {e} with line: {line_content}")
+                     yield f"Error in stream chunk: {e}"  # Optionally notify the user
+                     continue
+
                     line_content = line[5:].strip()  
                     if not line_content:
                         continue