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-video"
+API_URL = "https://api-singapore.klingai.com/v1/image-to-image"
 
 def generate_jwt_token():
     """Generate JWT token for Kling AI API authentication."""
@@ -20,16 +20,16 @@ def generate_jwt_token():
     token = jwt.encode(payload, ACCESS_KEY_SECRET, algorithm="HS256")
     return token
 
-def generate_video(image, prompt=""):
+def generate_image(image, prompt=""):
     """
-    Call Kling AI API to generate a video from an input image.
+    Call Kling AI API to generate an image from an input image.
     
     Args:
         image: Uploaded image file (from Gradio)
-        prompt (str): Optional text prompt to guide video generation
+        prompt (str): Optional text prompt to guide image transformation
     
     Returns:
-        str: Path to the downloaded video or error message
+        str: Path to the generated image or error message
     """
     if not image:
         return "Error: Please upload a valid image (PNG/JPEG)."
@@ -49,8 +49,8 @@ def generate_video(image, prompt=""):
     payload = {
         "image": image_base64,
         "prompt": prompt,
-        "duration": 5,
-        "aspect_ratio": "16:9"
+        "strength": 0.7,  # Controls transformation intensity (0.0 to 1.0)
+        "output_format": "png"
     }
     
     try:
@@ -69,21 +69,21 @@ def generate_video(image, prompt=""):
             status_response.raise_for_status()
             status_data = status_response.json()
             if status_data.get("status") == "succeeded":
-                video_url = status_data.get("video_url") or status_data.get("result", {}).get("video_url")
-                if not video_url:
-                    return "Error: No video URL in API response."
-                # Download the video
-                video_response = requests.get(video_url, timeout=30)
-                video_response.raise_for_status()
-                output_path = Path("output_video.mp4")
+                image_url = status_data.get("image_url") or status_data.get("result", {}).get("image_url")
+                if not image_url:
+                    return "Error: No image URL in API response."
+                # Download the image
+                image_response = requests.get(image_url, timeout=30)
+                image_response.raise_for_status()
+                output_path = Path("output_image.png")
                 with open(output_path, "wb") as f:
-                    f.write(video_response.content)
+                    f.write(image_response.content)
                 return str(output_path)
             elif status_data.get("status") == "failed":
-                return "Error: Video generation failed."
+                return "Error: Image generation failed."
             time.sleep(5)
         
-        return "Error: Video generation timed out."
+        return "Error: Image generation timed out."
     
     except requests.exceptions.HTTPError as e:
         return f"Error: API request failed. Details: {str(e)}"
@@ -92,27 +92,27 @@ def generate_video(image, prompt=""):
 
 def chatbot_interface(image, prompt):
     """
-    Gradio interface for image-to-video generation.
+    Gradio interface for image-to-image generation.
     
     Args:
         image: Uploaded image file
         prompt (str): Optional text prompt
     
     Returns:
-        Video file path or error message
+        Image file path or error message
     """
-    return generate_video(image, prompt)
+    return generate_image(image, prompt)
 
 # Define Gradio interface
 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., 'A cat running')", label="Prompt")
+        gr.Textbox(lines=2, placeholder="Enter an optional prompt (e.g., 'Turn this into a painting')", label="Prompt")
     ],
-    outputs=gr.Video(label="Generated Video"),
-    title="Kling AI Image-to-Video Generator",
-    description="Upload a PNG/JPEG image to generate a 5-second video using Kling AI API."
+    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."
 )
 
 # Launch the interface