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/images/image2image"  # Correct endpoint
+API_URL = "https://api.klingai.com/v1/images/image2image"  # Your specified endpoint
 
 def generate_jwt_token():
     """Generate JWT token for Kling AI API authentication."""
@@ -62,13 +62,12 @@ def generate_image(image, prompt=""):
     
     payload = {
         "model_name": "kling-v2-1",  # V2.1 model for image-to-image
-        "task_type": "image_generation",  # Adapted from video task structure
-        "input": {
-            "image": image_base64,
-            "prompt": prompt or "Transform the face into a cartoon style while preserving identity",
-            "strength": 0.97,  # High reference strength
-            "output_format": "png"
-        }
+        "image": image_base64,
+        "prompt": prompt or "Transform the face into a cartoon style while preserving identity",
+        "strength": 0.97,  # High reference strength
+        "output_format": "png",
+        "access_key_id": ACCESS_KEY_ID,  # Fallback for authentication
+        "access_key_secret": ACCESS_KEY_SECRET
     }
     
     try:
@@ -88,6 +87,11 @@ def generate_image(image, prompt=""):
         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)
+            if status_response.status_code == 404:
+                # Try alternative status endpoint
+                status_url = f"https://api.klingai.com/api/v1/task/{task_id}"
+                logger.debug(f"Fallback: Polling status at {status_url}")
+                status_response = requests.get(status_url, headers=headers, timeout=30)
             status_response.raise_for_status()
             status_data = status_response.json()
             logger.debug(f"Status response: {status_data}")
@@ -127,15 +131,12 @@ def generate_image(image, prompt=""):
         error_message = error_data.get("message", str(e))
         logger.error(f"HTTP error: {status_code}, Service Code: {service_code}, Details: {error_message}")
         
-        if error_message.startswith("Create task failed"):
-            logger.error("Create task failed error detected")
-            return None, "Error: Create task failed. Try during off-peak hours (e.g., 2 AM IST), verify account status at https://app.klingai.com, or join Kling AI Discord for support."
-        elif status_code == 404:
+        if status_code == 404:
             if service_code == 1202:
                 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). Verify https://api.klingai.com/v1/images/image2image at https://app.klingai.com/global/dev."
+                return None, "Error: Resource does not exist (1203). Verify model or endpoint at https://app.klingai.com/global/dev."
+            return None, "Error: Endpoint not found (404). Verify https://api.klingai.com/v1/images/image2image at https://app.klingai.com/global/dev or try alternative endpoints like /v1/images/generations."
         elif status_code == 401:
             if service_code == 1000:
                 return None, "Error: Authentication failed (1000). Verify API keys are correct."
@@ -205,7 +206,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 V2.1 API (reference strength 0.97). Avoid NSFW or sensitive content.
-        If errors persist, check account status at https://app.klingai.com or join Kling AI Discord for support.
+        If errors persist, verify endpoint at https://app.klingai.com/global/dev or contact support via https://app.klingai.com.
         """
     )
     with gr.Row():