Update app.py

app.py

@@ -14,7 +14,7 @@ logger = logging.getLogger(__name__)
 # Kling AI API configuration (keys hardcoded as requested)
 ACCESS_KEY_ID = "AGBGmadNd9hakFYfahytyQQJtN8CJmDJ"
 ACCESS_KEY_SECRET = "dp3pAe4PpdmnAHCAPgEd3PyLmBQrkMde"
-API_URL = "https://api.klingai.com/v1/predictions"  # Updated endpoint
+API_URL = "https://api.klingai.com/v1/generations"  # Updated endpoint
 
 def generate_jwt_token():
     """Generate JWT token for Kling AI API authentication."""
@@ -114,7 +114,7 @@ def generate_image(image, prompt=""):
             time.sleep(5)
         
         logger.error("Image generation timed out")
-        return None, "Error: Image generation timed out. Try during off-peak hours or check account status."
+        return None, "Error: Image generation timed out. Try during off-peak hours (e.g., 2 AM IST) or check account status."
     
     except requests.exceptions.HTTPError as e:
         status_code = e.response.status_code if e.response else None
@@ -127,7 +127,7 @@ def generate_image(image, prompt=""):
                 return None, "Error: Invalid request method (1202). Check https://klingai.com/global/dev for correct endpoint."
             elif service_code == 1203:
                 return None, "Error: Resource does not exist (1203). Verify model or endpoint."
-            return None, f"Error: Endpoint not found (404). Try https://api.klingai.com/v1/predictions or check https://klingai.com/global/dev."
+            return None, f"Error: Endpoint not found (404). Try https://api.klingai.com/v1/image or check https://klingai.com/global/dev."
         elif status_code == 401:
             if service_code == 1000:
                 return None, "Error: Authentication failed (1000). Verify API keys are correct."
@@ -197,6 +197,7 @@ with gr.Blocks() as iface:
         """
         # Kling AI Single-Reference Face Generator
         Upload a PNG/JPEG image (<10 MB, ≥512x512 pixels) with a clear face to generate a transformed image using Kling AI API (reference strength 0.97). Avoid NSFW or sensitive content.
+        If you see a 404 error, check the correct API endpoint at https://klingai.com/global/dev.
         """
     )
     with gr.Row():