Update app.py

app.py

@@ -1,11 +1,14 @@
 import gradio as gr
-from lumaai import Lumaai
+from lumaai import LumaAI
 import requests
+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:
@@ -18,6 +21,7 @@ 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
@@ -29,6 +33,7 @@ def poll_generation(api_key, generation_id):
     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
@@ -47,17 +52,41 @@ def download_video(api_key, generation_id):
     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")
+    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")
-            aspect_ratio = gr.Dropdown(["16:9", "9:16", "1:1", "4:3", "3:4"], label="Aspect Ratio")
-            loop = gr.Checkbox(label="Loop")
+            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)
             generate_btn = gr.Button("Generate Video")
         
         with gr.Column(scale=3):
@@ -67,6 +96,7 @@ with gr.Blocks() as demo:
     
     generation_id = gr.State()
     
+    # Basic generate button action
     def on_generate(api_key, prompt, aspect_ratio, loop):
         gen_id, message = generate_video(api_key, prompt, aspect_ratio, loop)
         return gen_id, message
@@ -77,12 +107,16 @@ with gr.Blocks() as demo:
         outputs=[generation_id, status]
     )
     
+    # Poll status and download video
     def on_poll(api_key, gen_id):
-        status, thumb = poll_generation(api_key, gen_id)
-        if status == "completed":
+        status_text, thumb = poll_generation(api_key, gen_id)
+        if status_text.lower() == "completed":
             video_path = download_video(api_key, gen_id)
-            return status, thumb, video_path
-        return status, thumb, None
+            if video_path:
+                return status_text, thumb, video_path
+            else:
+                return status_text, thumb, None
+        return status_text, thumb, None
     
     poll_btn = gr.Button("Check Status")
     poll_btn.click(
@@ -90,5 +124,64 @@ with gr.Blocks() as demo:
         inputs=[api_key, generation_id],
         outputs=[status, thumbnail, video]
     )
-
-demo.launch()
+    
+    # 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", 
+                    default=None, 
+                    optional=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
+    - [Luma AI Python SDK](https://github.com/lumalabs/lumaai-python)
+    - [Get API Key](https://lumalabs.ai/dream-machine/api/keys)
+    """)
+    
+demo.launch()