Update app.py

app.py

@@ -107,14 +107,21 @@ def respond(message, api_key, max_tokens, top_p, temperature):
             assistant_reply = ""
             for line in response.iter_lines(decode_unicode=True):
                 print(f"Received line: {line}")  # Debugging
-                if line:
+        
+                if line.strip():  # Ignore empty or whitespace-only lines
                     try:
-                        chunk = json.loads(line)
-                        chunk_message = chunk.get("chunk", {}).get("content", "")
-                        assistant_reply += chunk_message
-                        yield assistant_reply
-                    except Exception as e:
-                        print(f"Stream chunk error: {e}")
+                        # Remove 'data: ' prefix if present
+                        if line.startswith("data:"):
+                            line = line[5:].strip()
+                
+                        if line:  # Ensure there's content after stripping
+                            chunk = json.loads(line)  # Parse JSON
+                            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}")
+
 
 
             history.append((message, assistant_reply, "You", "P-ALPLE", sessions[api_key]["avatar"], ASSISTANT_PIC_PATH))