Update app.py

app.py

@@ -8,7 +8,7 @@ import time
 # Kling AI API configuration (keys hardcoded as requested)
 ACCESS_KEY_ID = "AGBGmadNd9hakFYfahytyQQJtN8CJmDJ"
 ACCESS_KEY_SECRET = "dp3pAe4PpdmnAHCAPgEd3PyLmBQrkMde"
-API_URL = "https://api-singapore.klingai.com/v1/predictions"
+API_URL = "https://api-singapore.klingai.com/v1/generate/image"
 
 def generate_jwt_token():
     """Generate JWT token for Kling AI API authentication."""
@@ -39,7 +39,7 @@ def generate_image(image, prompt=""):
         with open(image, "rb") as img_file:
             image_base64 = base64.b64encode(img_file.read()).decode("utf-8")
     except Exception as e:
-        return f"Error: Failed to process image. Details: {str(e)}"
+        return f"Error: Failed to process image. Ensure it’s a valid PNG/JPEG. Details: {str(e)}"
     
     headers = {
         "Authorization": f"Bearer {generate_jwt_token()}",
@@ -47,13 +47,10 @@ def generate_image(image, prompt=""):
     }
     
     payload = {
-        "inputs": [{"name": "input_image", "inputType": "URL", "url": image_base64}],
-        "arguments": [
-            {"name": "prompt", "value": prompt or "Transform the image"},
-            {"name": "strength", "value": "0.7"},
-            {"name": "output_format", "value": "png"},
-            {"name": "task_type", "value": "image-to-image"}
-        ]
+        "image": image_base64,
+        "prompt": prompt or "Transform the image into a cartoon style",
+        "strength": 0.7,  # Controls transformation intensity (0.0 to 1.0)
+        "output_format": "png"
     }
     
     try:
@@ -61,9 +58,9 @@ def generate_image(image, prompt=""):
         response.raise_for_status()
         
         data = response.json()
-        task_id = data.get("task_id") or data.get("taskId")
+        task_id = data.get("task_id") or data.get("taskId") or data.get("id")
         if not task_id:
-            return "Error: No task ID returned by the API. Check endpoint or keys."
+            return "Error: No task ID returned. Check API endpoint or keys at https://klingai.com/global/dev."
         
         # Poll for task completion
         status_url = f"https://api-singapore.klingai.com/v1/predictions/{task_id}"
@@ -71,10 +68,11 @@ def generate_image(image, prompt=""):
             status_response = requests.get(status_url, headers=headers, timeout=30)
             status_response.raise_for_status()
             status_data = status_response.json()
-            if status_data.get("status") == "succeeded":
-                image_url = status_data.get("image_url") or status_data.get("result", {}).get("image_url")
+            status = status_data.get("status")
+            if status == "succeeded":
+                image_url = status_data.get("image_url") or status_data.get("result", {}).get("image_url") or status_data.get("output")
                 if not image_url:
-                    return "Error: No image URL in API response."
+                    return "Error: No image URL in API response. Check API documentation."
                 # Download the image
                 image_response = requests.get(image_url, timeout=30)
                 image_response.raise_for_status()
@@ -82,23 +80,27 @@ def generate_image(image, prompt=""):
                 with open(output_path, "wb") as f:
                     f.write(image_response.content)
                 return str(output_path)
-            elif status_data.get("status") == "failed":
-                return "Error: Image generation failed. Check image content (no NSFW, celebrities, or sensitive topics)."
+            elif status == "failed":
+                return "Error: Image generation failed. Avoid NSFW, celebrities, or sensitive content (e.g., politics, violence)."
+            elif status == "processing" and status_data.get("progress", 0) >= 0.99:
+                return "Error: Generation stuck at 99%. Check account credits or upgrade to a paid plan at https://klingai.com."
             time.sleep(5)
         
-        return "Error: Image generation timed out. Try a paid account if using free credits."
+        return "Error: Image generation timed out. Try during off-peak hours or check account status."
     
     except requests.exceptions.HTTPError as e:
         status_code = e.response.status_code
         if status_code == 401:
-            return "Error: Authentication failed. Verify API keys."
+            return "Error: Authentication failed. Verify API keys are correct."
         elif status_code == 404:
-            return "Error: API endpoint not found. Check https://klingai.com/global/dev for correct endpoint."
+            return "Error: API endpoint not found. Check https://klingai.com/global/dev for correct endpoint (e.g., /v1/predictions)."
         elif status_code == 422:
-            return "Error: Invalid input. Use PNG/JPEG (<10 MB, ≥512x512 pixels) and a simple prompt."
+            return "Error: Invalid input. Use PNG/JPEG (<10 MB, ≥512x512 pixels) and a simple prompt (e.g., 'Turn this into a sketch')."
+        elif status_code == 500:
+            return "Error: Server issue. Try again later or during off-peak hours."
         return f"Error: API request failed. Details: {str(e)}"
     except requests.exceptions.RequestException as e:
-        return f"Error: Network issue. Ensure internet access and try again. Details: {str(e)}"
+        return f"Error: Network issue. Ensure stable internet, disable VPN, and try again. Details: {str(e)}"
 
 def chatbot_interface(image, prompt):
     """
@@ -122,7 +124,7 @@ iface = gr.Interface(
     ],
     outputs=gr.Image(label="Generated Image"),
     title="Kling AI Image-to-Image Generator",
-    description="Upload a PNG/JPEG image to generate a transformed image using Kling AI API. Ensure image is <10 MB, ≥512x512 pixels, and avoid NSFW or sensitive content."
+    description="Upload a PNG/JPEG image (<10 MB, ≥512x512 pixels) to generate a transformed image using Kling AI API. Avoid NSFW, celebrities, or sensitive content."
 )
 
 # Launch the interface