Update app.py

app.py

@@ -1,52 +1,68 @@
 import gradio as gr
 import os
-from lumaai import LumaAI
-import time
-import requests
+from lumaai import AsyncLumaAI
+import asyncio
+import aiohttp
 
-def generate_video(api_key, prompt, loop=False, aspect_ratio="16:9"):
-    client = LumaAI(auth_token=api_key)
+async def generate_video(api_key, prompt, loop=False, aspect_ratio="16:9", progress=gr.Progress()):
+    client = AsyncLumaAI(auth_token=api_key)
     
-    generation = client.generations.create(
+    progress(0, desc="Initiating video generation...")
+    generation = await client.generations.create(
         prompt=prompt,
         loop=loop,
         aspect_ratio=aspect_ratio
     )
     
+    progress(0.2, desc="Video generation started. Waiting for completion...")
+    
     # Poll for completion
     while True:
-        status = client.generations.get(id=generation.id)
+        status = await client.generations.get(id=generation.id)
         if status.status == "completed":
             break
-        time.sleep(5)
+        elif status.status == "failed":
+            raise Exception("Video generation failed")
+        await asyncio.sleep(5)
+        progress(0.5, desc="Still processing...")
+
+    progress(0.8, desc="Downloading generated video...")
     
     # Download the video
     video_url = status.assets.video
-    response = requests.get(video_url, stream=True)
-    
-    file_name = f"luma_ai_generated_{generation.id}.mp4"
-    with open(file_name, 'wb') as file:
-        file.write(response.content)
+    async with aiohttp.ClientSession() as session:
+        async with session.get(video_url) as resp:
+            if resp.status == 200:
+                file_name = f"luma_ai_generated_{generation.id}.mp4"
+                with open(file_name, 'wb') as fd:
+                    while True:
+                        chunk = await resp.content.read(1024)
+                        if not chunk:
+                            break
+                        fd.write(chunk)
     
+    progress(1.0, desc="Video generation complete!")
     return file_name
 
-def text_to_video(api_key, prompt, loop, aspect_ratio):
+async def text_to_video(api_key, prompt, loop, aspect_ratio, progress=gr.Progress()):
     if not api_key:
         return "Please enter your Luma AI API key."
     
     try:
-        video_path = generate_video(api_key, prompt, loop, aspect_ratio)
+        video_path = await generate_video(api_key, prompt, loop, aspect_ratio, progress)
         return video_path
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
-def image_to_video(api_key, prompt, image_url, loop, aspect_ratio):
+async def image_to_video(api_key, prompt, image_url, loop, aspect_ratio, progress=gr.Progress()):
     if not api_key:
         return "Please enter your Luma AI API key."
     
     try:
-        client = LumaAI(auth_token=api_key)
-        generation = client.generations.create(
+        client = AsyncLumaAI(auth_token=api_key)
+        
+        progress(0, desc="Initiating video generation from image...")
+        generation = await client.generations.create(
             prompt=prompt,
             loop=loop,
             aspect_ratio=aspect_ratio,
@@ -58,21 +74,34 @@ def image_to_video(api_key, prompt, image_url, loop, aspect_ratio):
             }
         )
         
+        progress(0.2, desc="Video generation started. Waiting for completion...")
+        
         # Poll for completion
         while True:
-            status = client.generations.get(id=generation.id)
+            status = await client.generations.get(id=generation.id)
             if status.status == "completed":
                 break
-            time.sleep(5)
+            elif status.status == "failed":
+                raise Exception("Video generation failed")
+            await asyncio.sleep(5)
+            progress(0.5, desc="Still processing...")
+
+        progress(0.8, desc="Downloading generated video...")
         
         # Download the video
         video_url = status.assets.video
-        response = requests.get(video_url, stream=True)
-        
-        file_name = f"luma_ai_generated_{generation.id}.mp4"
-        with open(file_name, 'wb') as file:
-            file.write(response.content)
+        async with aiohttp.ClientSession() as session:
+            async with session.get(video_url) as resp:
+                if resp.status == 200:
+                    file_name = f"luma_ai_generated_{generation.id}.mp4"
+                    with open(file_name, 'wb') as fd:
+                        while True:
+                            chunk = await resp.content.read(1024)
+                            if not chunk:
+                                break
+                            fd.write(chunk)
         
+        progress(1.0, desc="Video generation complete!")
         return file_name
     except Exception as e:
         return f"An error occurred: {str(e)}"
@@ -113,4 +142,4 @@ with gr.Blocks() as demo:
             outputs=img_video_output
         )
 
-demo.launch()
+demo.queue().launch()