Update app.py

app.py

@@ -14,19 +14,25 @@ async def generate_video(api_key, prompt, loop=False, aspect_ratio="16:9", progr
         aspect_ratio=aspect_ratio
     )
     
-    progress(0.2, desc="Video generation started. Waiting for completion...")
+    progress(0.1, desc="Video generation started. Waiting for completion...")
     
     # Poll for completion
+    start_time = asyncio.get_event_loop().time()
     while True:
         status = await client.generations.get(id=generation.id)
         if status.status == "completed":
             break
         elif status.status == "failed":
             raise Exception("Video generation failed")
+        
+        # Update progress based on time elapsed (assuming 60 seconds total)
+        elapsed_time = asyncio.get_event_loop().time() - start_time
+        progress_value = min(0.1 + (elapsed_time / 60) * 0.8, 0.9)
+        progress(progress_value, desc="Generating video...")
+        
         await asyncio.sleep(5)
-        progress(0.5, desc="Still processing...")
 
-    progress(0.8, desc="Downloading generated video...")
+    progress(0.9, desc="Downloading generated video...")
     
     # Download the video
     video_url = status.assets.video
@@ -46,17 +52,17 @@ async def generate_video(api_key, prompt, loop=False, aspect_ratio="16:9", progr
 
 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."
+        raise gr.Error("Please enter your Luma AI API key.")
     
     try:
         video_path = await generate_video(api_key, prompt, loop, aspect_ratio, progress)
-        return video_path
+        return gr.Video.update(value=video_path), ""
     except Exception as e:
-        return f"An error occurred: {str(e)}"
+        return gr.Video.update(value=None), f"An error occurred: {str(e)}"
 
 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."
+        raise gr.Error("Please enter your Luma AI API key.")
     
     try:
         client = AsyncLumaAI(auth_token=api_key)
@@ -74,19 +80,25 @@ async def image_to_video(api_key, prompt, image_url, loop, aspect_ratio, progres
             }
         )
         
-        progress(0.2, desc="Video generation started. Waiting for completion...")
+        progress(0.1, desc="Video generation started. Waiting for completion...")
         
         # Poll for completion
+        start_time = asyncio.get_event_loop().time()
         while True:
             status = await client.generations.get(id=generation.id)
             if status.status == "completed":
                 break
             elif status.status == "failed":
                 raise Exception("Video generation failed")
+            
+            # Update progress based on time elapsed (assuming 60 seconds total)
+            elapsed_time = asyncio.get_event_loop().time() - start_time
+            progress_value = min(0.1 + (elapsed_time / 60) * 0.8, 0.9)
+            progress(progress_value, desc="Generating video...")
+            
             await asyncio.sleep(5)
-            progress(0.5, desc="Still processing...")
 
-        progress(0.8, desc="Downloading generated video...")
+        progress(0.9, desc="Downloading generated video...")
         
         # Download the video
         video_url = status.assets.video
@@ -102,9 +114,9 @@ async def image_to_video(api_key, prompt, image_url, loop, aspect_ratio, progres
                             fd.write(chunk)
         
         progress(1.0, desc="Video generation complete!")
-        return file_name
+        return gr.Video.update(value=file_name), ""
     except Exception as e:
-        return f"An error occurred: {str(e)}"
+        return gr.Video.update(value=None), f"An error occurred: {str(e)}"
 
 with gr.Blocks() as demo:
     gr.Markdown("# Luma AI Text-to-Video Demo")
@@ -115,6 +127,7 @@ with gr.Blocks() as demo:
         prompt = gr.Textbox(label="Prompt")
         generate_btn = gr.Button("Generate Video")
         video_output = gr.Video(label="Generated Video")
+        error_output = gr.Textbox(label="Error Messages", visible=True)
         
         with gr.Accordion("Advanced Options", open=False):
             loop = gr.Checkbox(label="Loop", value=False)
@@ -123,7 +136,7 @@ with gr.Blocks() as demo:
         generate_btn.click(
             text_to_video,
             inputs=[api_key, prompt, loop, aspect_ratio],
-            outputs=video_output
+            outputs=[video_output, error_output]
         )
     
     with gr.Tab("Image to Video"):
@@ -131,6 +144,7 @@ with gr.Blocks() as demo:
         img_url = gr.Textbox(label="Image URL")
         img_generate_btn = gr.Button("Generate Video from Image")
         img_video_output = gr.Video(label="Generated Video")
+        img_error_output = gr.Textbox(label="Error Messages", visible=True)
         
         with gr.Accordion("Advanced Options", open=False):
             img_loop = gr.Checkbox(label="Loop", value=False)
@@ -139,7 +153,7 @@ with gr.Blocks() as demo:
         img_generate_btn.click(
             image_to_video,
             inputs=[api_key, img_prompt, img_url, img_loop, img_aspect_ratio],
-            outputs=img_video_output
+            outputs=[img_video_output, img_error_output]
         )
 
 demo.queue().launch()