Update app.py

app.py

@@ -3,17 +3,17 @@ import requests
 import os
 from pathlib import Path
 
-# Retrieve Kling AI API key from environment variable (set as a Space secret)
+# 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
 
 def generate_video(prompt, negative_prompt="", duration=5, aspect_ratio="16:9"):
     """
-    Call Kling AI API to generate a video based on a text prompt.
+    Call Kling AI API to generate a video from a text prompt.
     
     Args:
         prompt (str): Text prompt for video generation
-        negative_prompt (str): Optional negative prompt to guide generation
+        negative_prompt (str): Optional negative prompt
         duration (int): Video duration in seconds
         aspect_ratio (str): Aspect ratio (e.g., '16:9')
     
@@ -21,7 +21,7 @@ def generate_video(prompt, negative_prompt="", duration=5, aspect_ratio="16:9"):
         str: Path to the downloaded video or error message
     """
     if not KLING_SECRET_KEY:
-        return "Error: KLING_SECRET_KEY environment variable not set."
+        return "Error: KLING_SECRET_KEY not set in environment variables."
     
     headers = {
         "Authorization": f"Bearer {KLING_SECRET_KEY}",
@@ -36,13 +36,12 @@ def generate_video(prompt, negative_prompt="", duration=5, aspect_ratio="16:9"):
     }
     
     try:
-        # Make API request to Kling AI
         response = requests.post(KLING_API_URL, json=payload, headers=headers, timeout=60)
         response.raise_for_status()
         
         # Assuming API returns a JSON with a video URL
         data = response.json()
-        video_url = data.get("video_url")  # Adjust based on actual API response structure
+        video_url = data.get("video_url")  # Adjust based on actual API response
         
         if not video_url:
             return "Error: No video URL returned by the API."
@@ -51,7 +50,7 @@ def generate_video(prompt, negative_prompt="", duration=5, aspect_ratio="16:9"):
         video_response = requests.get(video_url, timeout=30)
         video_response.raise_for_status()
         
-        # Save video to a temporary file
+        # Save video to temporary file
         output_path = Path("output_video.mp4")
         with open(output_path, "wb") as f:
             f.write(video_response.content)
@@ -63,37 +62,36 @@ def generate_video(prompt, negative_prompt="", duration=5, aspect_ratio="16:9"):
 
 def chatbot_interface(prompt, negative_prompt, duration, aspect_ratio):
     """
-    Gradio interface to handle user inputs and display results.
+    Gradio interface for user inputs and video output.
     
     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
+        prompt (str): Text prompt
+        negative_prompt (str): Negative prompt
+        duration (int): Video duration
+        aspect_ratio (str): Aspect ratio
     
     Returns:
-        Video or str: Path to the video file or error message
+        Video file path 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
+    return generate_video(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 here...", label="Prompt"),
+        gr.Textbox(lines=2, placeholder="Enter your video generation 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.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!"
+    description="Generate videos using Kling AI API. Enter a prompt and settings to create your video!"
 )
 
-# Launch the interface (Hugging Face Spaces handles this automatically)
+# Launch the interface
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(server_name="0.0.0.0", server_port=7860)