Update api_helper.py

api_helper.py

@@ -19,7 +19,7 @@ def call_api(message):
         "Authorization": f"Bearer {API_KEY}",
         "Content-Type": "application/json",
     }
-    payload = {"prompt": message}  # Updated to match API's expected key
+    payload = {"prompt": message}  # Ensure the key matches API's expected input
 
     try:
         # Send request with streaming enabled
@@ -35,6 +35,8 @@ def call_api(message):
                     bot_response += data.get("response", "")
                 except json.JSONDecodeError:
                     bot_response += "\nError decoding response chunk."
-        return bot_response
+        return bot_response.strip() if bot_response else "Error: Empty response from API."
+    except requests.exceptions.Timeout:
+        return "Error: The API call timed out."
     except requests.exceptions.RequestException as e:
         return f"Error: {str(e)}"