Update app.py

app.py

@@ -1,33 +1,97 @@
-from flask import Flask, request, jsonify
-import os
+import gradio as gr
 import requests
+import os
+from pathlib import Path
+import time
 
-app = Flask(__name__)
-
-# Store your Kling API key in HF "Secrets" settings
-KLING_API_KEY = os.getenv("KLING_API_KEY")
+# Kling AI API configuration
+KLING_API_URL = "https://api.klingai.com/v1/video"  # Hypothetical endpoint; replace with actual Kling AI API endpoint
+KLING_SECRET_KEY = "dp3pAe4PpdmnAHCAPgEd3PyLmBQrkMde"  # Provided secret key
 
-@app.route("/generate", methods=["POST"])
-def generate():
-    data = request.get_json()
-    prompt = data.get("prompt", "")
+def generate_video(prompt, negative_prompt="", duration=5, aspect_ratio="16:9"):
+    """
+    Function to call Kling AI API for text-to-video generation.
+    
+    Args:
+        prompt (str): Text prompt for video generation
+        negative_prompt (str): Optional negative prompt to guide generation
+        duration (int): Video duration in seconds
+        aspect_ratio (str): Aspect ratio (e.g., '16:9')
+    
+    Returns:
+        str: Path to the downloaded video or error message
+    """
+    headers = {
+        "Authorization": f"Bearer {KLING_SECRET_KEY}",
+        "Content-Type": "application/json"
+    }
+    
+    payload = {
+        "prompt": prompt,
+        "negative_prompt": negative_prompt,
+        "duration": duration,
+        "aspect_ratio": aspect_ratio
+    }
     
     try:
-        response = requests.post(
-            "https://api.kling.ai/v1/images",  # Replace with actual Kling endpoint
-            headers={
-                "Authorization": f"Bearer {KLING_API_KEY}",
-                "Content-Type": "application/json"
-            },
-            json={
-                "model": "your-model-id",  # Put actual Kling model ID
-                "prompt": prompt,
-                "size": "1024x1024"
-            }
-        )
-        return jsonify(response.json()), response.status_code
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
+        # Make API request to Kling AI
+        response = requests.post(KLING_API_URL, json=payload, headers=headers, timeout=60)
+        response.raise_for_status()  # Raise exception for bad status codes
+        
+        # Assuming the API returns a JSON with a video URL or file
+        data = response.json()
+        video_url = data.get("video_url")  # Adjust based on actual API response structure
+        
+        if not video_url:
+            return "Error: No video URL returned by the API."
+        
+        # Download the video
+        video_response = requests.get(video_url, timeout=30)
+        video_response.raise_for_status()
+        
+        # Save the video to a temporary file
+        output_path = Path("output_video.mp4")
+        with open(output_path, "wb") as f:
+            f.write(video_response.content)
+        
+        return str(output_path)
+    
+    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):
+    """
+    Gradio interface function to handle user inputs and display results.
+    
+    Args:
+        prompt (str): Text prompt for video generation
+        negative_prompt (str): Negative prompt to avoid certain elements
+        duration (int): Video duration in seconds
+        aspect_ratio (str): Aspect ratio selection
+    
+    Returns:
+        Video or str: Path to the video file or error message
+    """
+    if not prompt.strip():
+        return "Error: Please provide a valid prompt."
+    
+    result = generate_video(prompt, negative_prompt, duration, aspect_ratio)
+    return result
+
+# Define Gradio interface
+iface = gr.Interface(
+    fn=chatbot_interface,
+    inputs=[
+        gr.Textbox(lines=2, placeholder="Enter your video generation prompt here...", 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.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, optionally a negative prompt, select duration and aspect ratio, and get your video!"
+)
 
+# Launch the interface
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+    iface.launch(server_name="0.0.0.0", server_port=7860)