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/generations"  # Updated endpoint
+API_URL = "https://api-singapore.klingai.com/v1/images/image2image"  # Updated endpoint
 
 def generate_jwt_token():
     """Generate JWT token for Kling AI API authentication."""
@@ -61,6 +61,7 @@ def generate_image(image, prompt=""):
     }
     
     payload = {
+        "model_name": "kling-v2-1",  # V2.1 model for image-to-image
         "image": image_base64,
         "prompt": prompt or "Transform the face into a cartoon style while preserving identity",
         "strength": 0.97,  # High reference strength for face preservation
@@ -77,10 +78,10 @@ def generate_image(image, prompt=""):
         task_id = data.get("task_id") or data.get("taskId") or data.get("id")
         if not task_id:
             logger.error("No task ID returned in API response")
-            return None, "Error: No task ID returned. Check API endpoint or keys at https://klingai.com/global/dev."
+            return None, "Error: No task ID returned. Check API endpoint or keys at https://app.klingai.com/global/dev."
         
         # Poll for task completion
-        status_url = f"{API_URL}/{task_id}"
+        status_url = f"https://api-singapore.klingai.com/v1/tasks/{task_id}"
         for _ in range(60):  # Poll for up to 5 minutes
             logger.debug(f"Polling status at {status_url}")
             status_response = requests.get(status_url, headers=headers, timeout=30)
@@ -92,7 +93,7 @@ def generate_image(image, prompt=""):
                 image_url = status_data.get("image_url") or status_data.get("result", {}).get("image_url") or status_data.get("output") or status_data.get("data", {}).get("image")
                 if not image_url:
                     logger.error("No image URL in API response")
-                    return None, "Error: No image URL in API response. Check API documentation at https://klingai.com/global/dev."
+                    return None, "Error: No image URL in API response. Check API documentation at https://app.klingai.com/global/dev."
                 # Download the image
                 logger.debug(f"Downloading image from {image_url}")
                 image_response = requests.get(image_url, timeout=30)
@@ -110,7 +111,7 @@ def generate_image(image, prompt=""):
                 return None, "Error: Image generation failed. Ensure the image contains a clear face and avoid NSFW or sensitive content."
             elif status == "processing" and status_data.get("progress", 0) >= 0.99:
                 logger.error("Generation stuck at 99%")
-                return None, "Error: Generation stuck at 99%. Check account credits or upgrade to a paid plan at https://klingai.com."
+                return None, "Error: Generation stuck at 99%. Check account credits or upgrade to a paid plan at https://app.klingai.com."
             time.sleep(5)
         
         logger.error("Image generation timed out")
@@ -124,10 +125,10 @@ def generate_image(image, prompt=""):
         
         if status_code == 404:
             if service_code == 1202:
-                return None, "Error: Invalid request method (1202). Check https://klingai.com/global/dev for correct endpoint."
+                return None, "Error: Invalid request method (1202). Check https://app.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/image or check https://klingai.com/global/dev."
+            return None, f"Error: Endpoint not found (404). Check https://app.klingai.com/global/dev for the correct endpoint."
         elif status_code == 401:
             if service_code == 1000:
                 return None, "Error: Authentication failed (1000). Verify API keys are correct."
@@ -141,7 +142,7 @@ def generate_image(image, prompt=""):
                 return None, "Error: Authorization expired (1004). Generate a new token."
         elif status_code == 429:
             if service_code == 1100:
-                return None, "Error: Account exception (1100). Verify account configuration at https://klingai.com."
+                return None, "Error: Account exception (1100). Verify account configuration at https://app.klingai.com."
             elif service_code == 1101:
                 return None, "Error: Account in arrears (1101). Recharge your account."
             elif service_code == 1102:
@@ -196,8 +197,8 @@ with gr.Blocks() as iface:
     gr.Markdown(
         """
         # 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.
+        Upload a PNG/JPEG image (<10 MB, ≥512x512 pixels) with a clear face to generate a transformed image using Kling AI V2.1 API (reference strength 0.97). Avoid NSFW or sensitive content.
+        If errors persist, check the correct API endpoint at https://app.klingai.com/global/dev.
         """
     )
     with gr.Row():