Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
-from lumaai.client import LumaAI  # Adjusted import path
+from lumaai import LumaAI
 import requests
 import os
-import time
 
 # Initialize LumaAI client
 def create_client(api_key):
@@ -29,7 +28,11 @@ def poll_generation(api_key, generation_id):
     client = create_client(api_key)
     try:
         generation = client.generations.get(id=generation_id)
-        return generation.status, generation.assets.thumbnail
+        status = generation.status
+        thumbnail_url = generation.assets.thumbnail
+        thumbnail_response = requests.get(thumbnail_url)
+        thumbnail = thumbnail_response.content if thumbnail_response.status_code == 200 else None
+        return status, thumbnail
     except Exception as e:
         return f"Error: {str(e)}", None
 
@@ -44,60 +47,70 @@ 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:
-            file.write(response.content)
+            for chunk in response.iter_content(chunk_size=8192):
+                if chunk:
+                    file.write(chunk)
         
         return file_name
     except Exception as e:
         return None
 
-# Advanced video generation with keyframes and camera motions
-def generate_video_advanced(api_key, prompt, aspect_ratio, loop, keyframes, camera_motion):
-    client = create_client(api_key)
-    try:
-        generation_params = {
-            "prompt": prompt,
-            "aspect_ratio": aspect_ratio,
-            "loop": loop
-        }
-        
-        # Parse keyframes from JSON string
-        if keyframes:
-            import json
-            generation_params["keyframes"] = json.loads(keyframes)
-        
-        # Add camera motion if provided
-        if camera_motion:
-            generation_params["camera_motion"] = camera_motion
-        
-        generation = client.generations.create(**generation_params)
-        return generation.id, "Advanced generation started..."
-    except Exception as e:
-        return None, f"Error: {str(e)}"
-
 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",
+        placeholder="sk-..."
+    )
     
     with gr.Row():
         with gr.Column(scale=2):
-            prompt = gr.Textbox(label="Prompt", placeholder="Describe your video...")
-            aspect_ratio = gr.Dropdown(["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",
+                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
+            )
             generate_btn = gr.Button("Generate Video")
         
         with gr.Column(scale=3):
-            status = gr.Textbox(label="Status")
-            thumbnail = gr.Image(label="Thumbnail")
-            video = gr.Video(label="Generated Video")
+            status = gr.Textbox(
+                label="Status",
+                readonly=True
+            )
+            thumbnail = gr.Image(
+                label="Thumbnail",
+                interactive=False
+            )
+            video = gr.Video(
+                label="Generated Video",
+                interactive=False
+            )
     
     generation_id = gr.State()
     
-    # Basic generate button action
+    # 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
     
@@ -107,16 +120,32 @@ with gr.Blocks() as demo:
         outputs=[generation_id, status]
     )
     
-    # Poll status and download video
+    # Check Status and Display Results
     def on_poll(api_key, gen_id):
-        status_text, thumb = poll_generation(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":
             video_path = download_video(api_key, gen_id)
             if video_path:
-                return status_text, thumb, video_path
+                video_output = video_path
             else:
-                return status_text, thumb, None
-        return status_text, thumb, None
+                video_output = None
+        else:
+            video_output = None
+        
+        # Convert thumbnail bytes to image format
+        if thumb_data:
+            from PIL import Image
+            from io import BytesIO
+            thumb_image = Image.open(BytesIO(thumb_data))
+        else:
+            thumb_image = None
+        
+        return status_text, thumb_image, video_output
     
     poll_btn = gr.Button("Check Status")
     poll_btn.click(
@@ -125,59 +154,6 @@ with gr.Blocks() as demo:
         outputs=[status, thumbnail, video]
     )
     
-    # Accordion for Advanced Features
-    with gr.Accordion("Advanced Features", open=False):
-        with gr.Row():
-            with gr.Column(scale=1):
-                keyframes = gr.Textbox(
-                    label="Keyframes (JSON)",
-                    placeholder='{"frame0": {"type": "image", "url": "https://example.com/image1.jpg"}, "frame1": {"type": "image", "url": "https://example.com/image2.jpg"}}'
-                )
-                camera_motion = gr.Dropdown(
-                    ["pan", "tilt", "zoom", "orbit"], 
-                    label="Camera Motion", 
-                    choices=["pan", "tilt", "zoom", "orbit"],  # Ensure choices are correctly passed
-                    value=None,
-                    allow_none=True
-                )
-                advanced_generate_btn = gr.Button("Generate with Advanced Options")
-            
-            with gr.Column(scale=2):
-                advanced_status = gr.Textbox(label="Advanced Status")
-                advanced_thumbnail = gr.Image(label="Advanced Thumbnail")
-                advanced_video = gr.Video(label="Advanced Generated Video")
-        
-        advanced_gen_id = gr.State()
-        
-        # Advanced generate button action
-        def on_generate_advanced(api_key, prompt, aspect_ratio, loop, keyframes, camera_motion):
-            gen_id, message = generate_video_advanced(api_key, prompt, aspect_ratio, loop, keyframes, camera_motion)
-            return gen_id, message
-        
-        advanced_generate_btn.click(
-            on_generate_advanced,
-            inputs=[api_key, prompt, aspect_ratio, loop, keyframes, camera_motion],
-            outputs=[advanced_gen_id, advanced_status]
-        )
-        
-        # Poll status and download video for advanced generation
-        def on_poll_advanced(api_key, gen_id):
-            status_text, thumb = poll_generation(api_key, gen_id)
-            if status_text.lower() == "completed":
-                video_path = download_video(api_key, gen_id)
-                if video_path:
-                    return status_text, thumb, video_path
-                else:
-                    return status_text, thumb, None
-            return status_text, thumb, None
-        
-        advanced_poll_btn = gr.Button("Check Advanced Status")
-        advanced_poll_btn.click(
-            on_poll_advanced,
-            inputs=[api_key, advanced_gen_id],
-            outputs=[advanced_status, advanced_thumbnail, advanced_video]
-        )
-    
     gr.Markdown("""
     ---
     ### Documentation & Resources