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/image-to-image"
+API_URL = "https://api-singapore.klingai.com/v1/predictions"
 
 def generate_jwt_token():
     """Generate JWT token for Kling AI API authentication."""
@@ -32,7 +32,7 @@ def generate_image(image, prompt=""):
         str: Path to the generated image or error message
     """
     if not image:
-        return "Error: Please upload a valid image (PNG/JPEG)."
+        return "Error: Please upload a valid image (PNG/JPEG, <10 MB, ≥512x512 pixels)."
     
     # Convert image to base64
     try:
@@ -47,10 +47,13 @@ def generate_image(image, prompt=""):
     }
     
     payload = {
-        "image": image_base64,
-        "prompt": prompt,
-        "strength": 0.7,  # Controls transformation intensity (0.0 to 1.0)
-        "output_format": "png"
+        "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"}
+        ]
     }
     
     try:
@@ -60,7 +63,7 @@ def generate_image(image, prompt=""):
         data = response.json()
         task_id = data.get("task_id") or data.get("taskId")
         if not task_id:
-            return "Error: No task ID returned by the API."
+            return "Error: No task ID returned by the API. Check endpoint or keys."
         
         # Poll for task completion
         status_url = f"https://api-singapore.klingai.com/v1/predictions/{task_id}"
@@ -80,15 +83,22 @@ def generate_image(image, prompt=""):
                     f.write(image_response.content)
                 return str(output_path)
             elif status_data.get("status") == "failed":
-                return "Error: Image generation failed."
+                return "Error: Image generation failed. Check image content (no NSFW, celebrities, or sensitive topics)."
             time.sleep(5)
         
-        return "Error: Image generation timed out."
+        return "Error: Image generation timed out. Try a paid account if using free credits."
     
     except requests.exceptions.HTTPError as e:
+        status_code = e.response.status_code
+        if status_code == 401:
+            return "Error: Authentication failed. Verify API keys."
+        elif status_code == 404:
+            return "Error: API endpoint not found. Check https://klingai.com/global/dev for correct endpoint."
+        elif status_code == 422:
+            return "Error: Invalid input. Use PNG/JPEG (<10 MB, ≥512x512 pixels) and a simple prompt."
         return f"Error: API request failed. Details: {str(e)}"
     except requests.exceptions.RequestException as e:
-        return f"Error: Network issue. Details: {str(e)}"
+        return f"Error: Network issue. Ensure internet access and try again. Details: {str(e)}"
 
 def chatbot_interface(image, prompt):
     """
@@ -107,12 +117,12 @@ def chatbot_interface(image, prompt):
 iface = gr.Interface(
     fn=chatbot_interface,
     inputs=[
-        gr.Image(type="filepath", label="Upload Image (PNG/JPEG)"),
-        gr.Textbox(lines=2, placeholder="Enter an optional prompt (e.g., 'Turn this into a painting')", label="Prompt")
+        gr.Image(type="filepath", label="Upload Image (PNG/JPEG, <10 MB, ≥512x512)"),
+        gr.Textbox(lines=2, placeholder="Enter an optional prompt (e.g., 'Turn this into a cartoon')", label="Prompt")
     ],
     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."
+    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."
 )
 
 # Launch the interface