Update app.py

app.py

@@ -99,34 +99,28 @@ def respond(message, api_key, max_tokens, top_p, temperature):
         "input": message,
         "stream": True  
     }
-    line_content = None
-    
+
     try:
         response = requests.post(API_URL, headers=headers, data=json.dumps(data), stream=True)
 
         if response.status_code == 200:
             assistant_reply = ""
             for line in response.iter_lines(decode_unicode=True):
-                line = line.strip()  
-                if not line:  
-                    continue
-
-                print(f"Received line: {line}")
-
-                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
+                print(f"Received line: {line}")  
+        
+                if line.strip(): 
+                    try:
+                        if line.startswith("data:"):
+                            line = line[5:].strip()
+                
+                        if line:  
+                            chunk = json.loads(line)  
+                            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))