Update app.py

app.py

@@ -8,16 +8,20 @@ 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/generate/image"
+API_URL = "https://api-singapore.klingai.com/v1/image-to-image"
 
 def generate_jwt_token():
     """Generate JWT token for Kling AI API authentication."""
+    headers = {
+        "alg": "HS256",
+        "typ": "JWT"
+    }
     payload = {
-        "iat": int(time.time()),
+        "iss": ACCESS_KEY_ID,
         "exp": int(time.time()) + 1800,  # Token expires in 30 minutes
-        "access_key": ACCESS_KEY_ID
+        "nbf": int(time.time()) - 5  # Effective 5 seconds before current time
     }
-    token = jwt.encode(payload, ACCESS_KEY_SECRET, algorithm="HS256")
+    token = jwt.encode(payload, ACCESS_KEY_SECRET, headers=headers)
     return token
 
 def generate_image(image, prompt=""):
@@ -89,16 +93,56 @@ def generate_image(image, prompt=""):
         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
+        status_code = e.response.status_code if e.response else None
+        error_data = e.response.json() if e.response and e.response.text else {}
+        service_code = error_data.get("code", 0)
+        
         if status_code == 401:
-            return "Error: Authentication failed. Verify API keys are correct."
+            if service_code == 1000:
+                return "Error: Authentication failed (1000). Verify API keys are correct."
+            elif service_code == 1001:
+                return "Error: Authorization empty (1001). Ensure JWT token is included."
+            elif service_code == 1002:
+                return "Error: Authorization invalid (1002). Check key format."
+            elif service_code == 1003:
+                return "Error: Authorization not yet valid (1003). Adjust token start time (nbf)."
+            elif service_code == 1004:
+                return "Error: Authorization expired (1004). Generate a new token."
+        elif status_code == 429:
+            if service_code == 1100:
+                return "Error: Account exception (1100). Verify account configuration at https://klingai.com."
+            elif service_code == 1101:
+                return "Error: Account in arrears (1101). Recharge your account."
+            elif service_code == 1102:
+                return "Error: Resource pack depleted or expired (1102). Purchase additional resources."
+            elif service_code in (1302, 1303):
+                return "Error: Rate limit exceeded (1302/1303). Reduce request frequency or contact support."
+            elif service_code == 1304:
+                return "Error: IP whitelisting issue (1304). Contact Kling AI support."
+        elif status_code == 403 and service_code == 1103:
+            return "Error: Unauthorized access (1103). Verify account permissions."
+        elif status_code == 400:
+            if service_code == 1200:
+                return "Error: Invalid request parameters (1200). Check payload format."
+            elif service_code == 1201:
+                return "Error: Invalid parameters (1201). Use PNG/JPEG (<10 MB, ≥512x512) and a simple prompt."
+            elif service_code == 1300:
+                return "Error: Platform policy triggered (1300). Check input content."
+            elif service_code == 1301:
+                return "Error: Content security policy triggered (1301). Avoid NSFW or sensitive content."
         elif status_code == 404:
-            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 (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)}"
+            if service_code == 1202:
+                return "Error: Invalid request method (1202). Check https://klingai.com/global/dev for correct endpoint."
+            elif service_code == 1203:
+                return "Error: Resource does not exist (1203). Verify model or endpoint."
+        elif status_code in (500, 503, 504):
+            if service_code == 5000:
+                return "Error: Server internal error (5000). Try again later."
+            elif service_code == 5001:
+                return "Error: Server unavailable due to maintenance (5001). Try again later."
+            elif service_code == 5002:
+                return "Error: Server timeout (5002). Try during off-peak hours."
+        return f"Error: API request failed. HTTP {status_code}, Service Code {service_code}. Details: {str(e)}"
     except requests.exceptions.RequestException as e:
         return f"Error: Network issue. Ensure stable internet, disable VPN, and try again. Details: {str(e)}"