Update api_helper.py

api_helper.py

@@ -19,23 +19,26 @@ def call_api(message):
         "Authorization": f"Bearer {API_KEY}",
         "Content-Type": "application/json",
     }
-    payload = {"prompt": message}  # Ensure the key matches API's expected input
+    payload = {"prompt": message}  # Match the API's expected input key
 
     try:
         # Send request with streaming enabled
         response = requests.post(SPACE_URL, json=payload, headers=headers, stream=True)
         response.raise_for_status()
 
-        # Stream and aggregate the response
+        # Aggregate streamed response
         bot_response = ""
-        for chunk in response.iter_lines():
-            if chunk:  # Ensure the chunk is not empty
+        for chunk in response.iter_lines(decode_unicode=True):
+            if chunk.strip():  # Ensure the chunk is not empty
                 try:
-                    data = json.loads(chunk.decode("utf-8"))
-                    bot_response += data.get("response", "")
+                    # Parse JSON chunk
+                    data = json.loads(chunk)
+                    if "response" in data:
+                        bot_response += data["response"]
                 except json.JSONDecodeError:
                     bot_response += "\nError decoding response chunk."
-        return bot_response.strip() if bot_response else "Error: Empty response from API."
+        
+        return bot_response.strip() if bot_response else "Error: No response from API."
     except requests.exceptions.Timeout:
         return "Error: The API call timed out."
     except requests.exceptions.RequestException as e: