Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 from lumaai import LumaAI
 import requests
+import time
 
 def generate_content(api_key, prompt, aspect_ratio, loop):
     """
@@ -32,11 +33,18 @@ def generate_content(api_key, prompt, aspect_ratio, loop):
         generation_id = generation.id
         video_url = None
         
-        # Poll until the video is ready
-        while video_url is None:
+        # Poll until the video is ready (adding a timeout of 60 seconds max)
+        timeout = 60  # seconds
+        start_time = time.time()
+        
+        while video_url is None and time.time() - start_time < timeout:
             generation_status = client.generations.get(id=generation_id)
             video_url = generation_status.assets.get('video', None)
+            if video_url is None:
+                time.sleep(5)  # Wait for 5 seconds before polling again
 
+        if video_url is None:
+            return "The video generation timed out. Please try again later."
         return video_url
     except Exception as e:
         return f"An error occurred: {str(e)}"