Update app.py

app.py

@@ -1,15 +1,11 @@
 import gradio as gr
-from lumaai.client import LumaAI  # Adjust based on your findings
+from lumaai import Lumaai
 import requests
-from PIL import Image
-from io import BytesIO
-import os
+import time
 
-# Initialize LumaAI client
 def create_client(api_key):
-    return LumaAI(auth_token=api_key)
+    return Lumaai(auth_token=api_key)
 
-# Basic video generation
 def generate_video(api_key, prompt, aspect_ratio, loop):
     client = create_client(api_key)
     try:
@@ -22,7 +18,6 @@ def generate_video(api_key, prompt, aspect_ratio, loop):
     except Exception as e:
         return None, f"Error: {str(e)}"
 
-# Polling generation status
 def poll_generation(api_key, generation_id):
     if not generation_id:
         return "No generation in progress", None
@@ -30,17 +25,10 @@ def poll_generation(api_key, generation_id):
     client = create_client(api_key)
     try:
         generation = client.generations.get(id=generation_id)
-        status = generation.status
-        thumbnail_url = generation.assets.thumbnail
-        
-        # Fetch the thumbnail image
-        thumbnail_response = requests.get(thumbnail_url)
-        thumbnail = thumbnail_response.content if thumbnail_response.status_code == 200 else None
-        return status, thumbnail
+        return generation.status, generation.assets.thumbnail
     except Exception as e:
         return f"Error: {str(e)}", None
 
-# Download the generated video
 def download_video(api_key, generation_id):
     if not generation_id:
         return None
@@ -51,14 +39,9 @@ def download_video(api_key, generation_id):
         video_url = generation.assets.video
         response = requests.get(video_url, stream=True)
         
-        if response.status_code != 200:
-            return None
-        
         file_name = f"generated_video_{generation_id}.mp4"
         with open(file_name, 'wb') as file:
-            for chunk in response.iter_content(chunk_size=8192):
-                if chunk:
-                    file.write(chunk)
+            file.write(response.content)
         
         return file_name
     except Exception as e:
@@ -68,53 +51,23 @@ with gr.Blocks() as demo:
     gr.Markdown("# Luma AI Video Generation Demo")
     gr.Markdown("Generate videos using Luma AI based on text prompts.")
     
-    api_key = gr.Textbox(
-        label="Enter your Luma AI API Key",
-        type="password",
-        placeholder="sk-..."
-    )
+    api_key = gr.Textbox(label="Enter your Luma AI API Key", type="password")
     
     with gr.Row():
         with gr.Column(scale=2):
-            prompt = gr.Textbox(
-                label="Prompt",
-                placeholder="Describe your video...",
-                lines=2
-            )
-            aspect_ratio = gr.Dropdown(
-                choices=["16:9", "9:16", "1:1", "4:3", "3:4"],
-                label="Aspect Ratio",
-                value="16:9"
-            )
-            loop = gr.Checkbox(
-                label="Loop",
-                value=False
-            )
+            prompt = gr.Textbox(label="Prompt")
+            aspect_ratio = gr.Dropdown(["16:9", "9:16", "1:1", "4:3", "3:4"], label="Aspect Ratio")
+            loop = gr.Checkbox(label="Loop")
             generate_btn = gr.Button("Generate Video")
         
         with gr.Column(scale=3):
-            status = gr.Textbox(
-                label="Status",
-                interactive=False  # Replaces readonly=True
-            )
-            thumbnail = gr.Image(
-                label="Thumbnail",
-                interactive=False
-            )
-            video = gr.Video(
-                label="Generated Video",
-                interactive=False
-            )
+            status = gr.Textbox(label="Status")
+            thumbnail = gr.Image(label="Thumbnail")
+            video = gr.Video(label="Generated Video")
     
     generation_id = gr.State()
     
-    # Generate Video Action
     def on_generate(api_key, prompt, aspect_ratio, loop):
-        if not api_key:
-            return None, "Please provide a valid API key."
-        if not prompt:
-            return None, "Prompt cannot be empty."
-        
         gen_id, message = generate_video(api_key, prompt, aspect_ratio, loop)
         return gen_id, message
     
@@ -124,30 +77,12 @@ with gr.Blocks() as demo:
         outputs=[generation_id, status]
     )
     
-    # Check Status and Display Results
     def on_poll(api_key, gen_id):
-        if not api_key:
-            return "Please provide a valid API key.", None, None
-        if not gen_id:
-            return "No generation in progress.", None, None
-        
-        status_text, thumb_data = poll_generation(api_key, gen_id)
-        if status_text.lower() == "completed":
+        status, thumb = poll_generation(api_key, gen_id)
+        if status == "completed":
             video_path = download_video(api_key, gen_id)
-            if video_path:
-                video_output = video_path
-            else:
-                video_output = None
-        else:
-            video_output = None
-        
-        # Convert thumbnail bytes to image format
-        if thumb_data:
-            thumb_image = Image.open(BytesIO(thumb_data))
-        else:
-            thumb_image = None
-        
-        return status_text, thumb_image, video_output
+            return status, thumb, video_path
+        return status, thumb, None
     
     poll_btn = gr.Button("Check Status")
     poll_btn.click(
@@ -155,12 +90,5 @@ with gr.Blocks() as demo:
         inputs=[api_key, generation_id],
         outputs=[status, thumbnail, video]
     )
-    
-    gr.Markdown("""
-    ---
-    ### Documentation & Resources
-    - [Luma AI Python SDK](https://github.com/lumalabs/lumaai-python)
-    - [Get API Key](https://lumalabs.ai/dream-machine/api/keys)
-    """)
 
-demo.launch()
+demo.launch()