Create app.py

app.py

@@ -0,0 +1,94 @@
+import gradio as gr
+from lumaai import Lumaai
+import requests
+import time
+
+def create_client(api_key):
+    return Lumaai(auth_token=api_key)
+
+def generate_video(api_key, prompt, aspect_ratio, loop):
+    client = create_client(api_key)
+    try:
+        generation = client.generations.create(
+            prompt=prompt,
+            aspect_ratio=aspect_ratio,
+            loop=loop
+        )
+        return generation.id, "Generation started..."
+    except Exception as e:
+        return None, f"Error: {str(e)}"
+
+def poll_generation(api_key, generation_id):
+    if not generation_id:
+        return "No generation in progress", None
+    
+    client = create_client(api_key)
+    try:
+        generation = client.generations.get(id=generation_id)
+        return generation.status, generation.assets.thumbnail
+    except Exception as e:
+        return f"Error: {str(e)}", None
+
+def download_video(api_key, generation_id):
+    if not generation_id:
+        return None
+    
+    client = create_client(api_key)
+    try:
+        generation = client.generations.get(id=generation_id)
+        video_url = generation.assets.video
+        response = requests.get(video_url, stream=True)
+        
+        file_name = f"generated_video_{generation_id}.mp4"
+        with open(file_name, 'wb') as file:
+            file.write(response.content)
+        
+        return file_name
+    except Exception as e:
+        return None
+
+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")
+    
+    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")
+            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")
+    
+    generation_id = gr.State()
+    
+    def on_generate(api_key, prompt, aspect_ratio, loop):
+        gen_id, message = generate_video(api_key, prompt, aspect_ratio, loop)
+        return gen_id, message
+    
+    generate_btn.click(
+        on_generate,
+        inputs=[api_key, prompt, aspect_ratio, loop],
+        outputs=[generation_id, status]
+    )
+    
+    def on_poll(api_key, gen_id):
+        status, thumb = poll_generation(api_key, gen_id)
+        if status == "completed":
+            video_path = download_video(api_key, gen_id)
+            return status, thumb, video_path
+        return status, thumb, None
+    
+    poll_btn = gr.Button("Check Status")
+    poll_btn.click(
+        on_poll,
+        inputs=[api_key, generation_id],
+        outputs=[status, thumbnail, video]
+    )
+
+demo.launch()