Update app.py

app.py

@@ -56,10 +56,9 @@ def call_api(history, max_tokens, temperature, top_p, system_message="", seed=42
     full_response = ""
     for line in response.iter_lines():
         if line:
-            decoded_line = line.decode("utf-8")
-            print(f"API response line: {decoded_line}")
-            if "data:" in decoded_line:
-                json_data = json.loads(decoded_line.replace("data:", ""))
+            decoded_line = line.decode("utf-8").strip()
+            if decoded_line.startswith("data:"):
+                json_data = json.loads(decoded_line[5:])
                 if "choices" in json_data and len(json_data["choices"]) > 0:
                     deltas = json_data["choices"][0].get("delta", {})
                     if "content" in deltas:
@@ -67,6 +66,7 @@ def call_api(history, max_tokens, temperature, top_p, system_message="", seed=42
     print(f"Full API response: {full_response}")
     return full_response
 
+    
 def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetition_penalty):
     print("Starting chat...")
     system_message_to_use = system_message if system_message.strip() else BASE_SYSTEM_MESSAGE