Update app.py

app.py

@@ -2,15 +2,14 @@ import gradio as gr
 import requests
 import os
 from pathlib import Path
-import time
 
-# 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
+# Retrieve Kling AI API key from environment variable (set as a Space secret)
+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"):
     """
-    Function to call Kling AI API for text-to-video generation.
+    Call Kling AI API to generate a video based on a text prompt.
     
     Args:
         prompt (str): Text prompt for video generation
@@ -21,6 +20,9 @@ def generate_video(prompt, negative_prompt="", duration=5, aspect_ratio="16:9"):
     Returns:
         str: Path to the downloaded video or error message
     """
+    if not KLING_SECRET_KEY:
+        return "Error: KLING_SECRET_KEY environment variable not set."
+    
     headers = {
         "Authorization": f"Bearer {KLING_SECRET_KEY}",
         "Content-Type": "application/json"
@@ -36,9 +38,9 @@ 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()  # Raise exception for bad status codes
+        response.raise_for_status()
         
-        # Assuming the API returns a JSON with a video URL or file
+        # 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
         
@@ -49,7 +51,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 the video to a temporary file
+        # Save video to a temporary file
         output_path = Path("output_video.mp4")
         with open(output_path, "wb") as f:
             f.write(video_response.content)
@@ -61,7 +63,7 @@ def generate_video(prompt, negative_prompt="", duration=5, aspect_ratio="16:9"):
 
 def chatbot_interface(prompt, negative_prompt, duration, aspect_ratio):
     """
-    Gradio interface function to handle user inputs and display results.
+    Gradio interface to handle user inputs and display results.
     
     Args:
         prompt (str): Text prompt for video generation
@@ -92,6 +94,6 @@ iface = gr.Interface(
     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
+# Launch the interface (Hugging Face Spaces handles this automatically)
 if __name__ == "__main__":
-    iface.launch(server_name="0.0.0.0", server_port=7860)
+    iface.launch()