Update app.py

app.py

@@ -1,34 +1,43 @@
 import gradio as gr
 import requests
-import os
 from pathlib import Path
+import base64
+import io
 
-# Retrieve Kling AI API key from environment variable (set as a secret in Hugging Face Spaces)
-KLING_SECRET_KEY = os.environ.get("KLING_SECRET_KEY")
-KLING_API_URL = "https://api.klingai.com/v1/video"  # Replace with actual Kling AI API endpoint
+# Kling AI API configuration (keys hardcoded as requested)
+KLING_ACCESS_KEY_ID = "AGBGmadNd9hakFYfahytyQQJtN8CJmDJ"
+KLING_ACCESS_KEY_SECRET = "dp3pAe4PpdmnAHCAPgEd3PyLmBQrkMde"
+KLING_API_URL = "https://api-singapore.klingai.com/v1/image-to-video"  # Assumed endpoint for image-to-image video generation
 
-def generate_video(prompt, negative_prompt="", duration=5, aspect_ratio="16:9"):
+def generate_video(image, prompt="", negative_prompt="", duration=5, aspect_ratio="16:9"):
     """
-    Call Kling AI API to generate a video from a text prompt.
+    Call Kling AI API to generate a video from an input image.
     
     Args:
-        prompt (str): Text prompt for video generation
+        image: Uploaded image file (from Gradio)
+        prompt (str): Optional text prompt to guide video generation
         negative_prompt (str): Optional negative prompt
-        duration (int): Video duration in seconds
+        duration (int): Video duration in seconds (5 or 10)
         aspect_ratio (str): Aspect ratio (e.g., '16:9')
     
     Returns:
         str: Path to the downloaded video or error message
     """
-    if not KLING_SECRET_KEY:
-        return "Error: KLING_SECRET_KEY not set in environment variables."
+    # Convert image to base64 for API submission
+    try:
+        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 input image. Details: {str(e)}"
     
     headers = {
-        "Authorization": f"Bearer {KLING_SECRET_KEY}",
+        "Authorization": f"Bearer {KLING_ACCESS_KEY_SECRET}",
+        "X-Kling-Access-Key-Id": KLING_ACCESS_KEY_ID,
         "Content-Type": "application/json"
     }
     
     payload = {
+        "image": image_base64,
         "prompt": prompt,
         "negative_prompt": negative_prompt,
         "duration": duration,
@@ -60,12 +69,13 @@ def generate_video(prompt, negative_prompt="", duration=5, aspect_ratio="16:9"):
     except requests.exceptions.RequestException as e:
         return f"Error: Failed to generate video. Details: {str(e)}"
 
-def chatbot_interface(prompt, negative_prompt, duration, aspect_ratio):
+def chatbot_interface(image, prompt, negative_prompt, duration, aspect_ratio):
     """
     Gradio interface for user inputs and video output.
     
     Args:
-        prompt (str): Text prompt
+        image: Uploaded image file
+        prompt (str): Optional text prompt
         negative_prompt (str): Negative prompt
         duration (int): Video duration
         aspect_ratio (str): Aspect ratio
@@ -73,23 +83,24 @@ def chatbot_interface(prompt, negative_prompt, duration, aspect_ratio):
     Returns:
         Video file path or error message
     """
-    if not prompt.strip():
-        return "Error: Please provide a valid prompt."
+    if not image:
+        return "Error: Please upload an image."
     
-    return generate_video(prompt, negative_prompt, duration, aspect_ratio)
+    return generate_video(image, prompt, negative_prompt, duration, aspect_ratio)
 
 # Define Gradio interface
 iface = gr.Interface(
     fn=chatbot_interface,
     inputs=[
-        gr.Textbox(lines=2, placeholder="Enter your video generation prompt...", label="Prompt"),
+        gr.Image(type="filepath", label="Input Image"),
+        gr.Textbox(lines=2, placeholder="Enter an optional text prompt...", label="Prompt"),
         gr.Textbox(lines=2, placeholder="Enter negative prompt (optional)", label="Negative Prompt"),
-        gr.Slider(minimum=1, maximum=30, step=1, value=5, label="Video Duration (seconds)"),
+        gr.Slider(minimum=5, maximum=10, step=5, value=5, label="Video Duration (seconds)"),
         gr.Dropdown(choices=["16:9", "4:3", "1:1"], value="16:9", label="Aspect Ratio")
     ],
     outputs=gr.Video(label="Generated Video"),
-    title="Kling AI Video Generator",
-    description="Generate videos using Kling AI API. Enter a prompt and settings to create your video!"
+    title="Kling AI Image-to-Video Generator",
+    description="Generate videos from an input image using Kling AI API. Upload an image, optionally add a prompt, and set parameters."
 )
 
 # Launch the interface