Update app.py

app.py

@@ -4,21 +4,18 @@ from lumaai import AsyncLumaAI
 import asyncio
 import aiohttp
 
-# 전역 변수로 카메라 모션 목록을 저장합니다.
-camera_motions = []
-
 async def get_camera_motions():
-    global camera_motions
     api_key = os.getenv("LMGEN_KEY")
     if not api_key:
         raise gr.Error("LMGEN_KEY environment variable is not set.")
     
     client = AsyncLumaAI(auth_token=api_key)
-    motions = await client.generations.camera_motion.list()
-    camera_motions = [motion.name for motion in motions]
-
-# 앱 시작 시 카메라 모션 목록을 가져옵니다.
-asyncio.run(get_camera_motions())
+    try:
+        motions = await client.generations.camera_motion.list()
+        return [motion.name for motion in motions]
+    except Exception as e:
+        print(f"Error fetching camera motions: {str(e)}")
+        return []
 
 async def generate_video(client, prompt, loop=False, aspect_ratio="16:9", camera_motion=None, extend_id=None, reverse_extend_id=None, interpolate_ids=None, progress=gr.Progress()):
     generation_params = {
@@ -179,6 +176,8 @@ async def image_to_video(prompt, image_url, loop, aspect_ratio, camera_motion, p
 with gr.Blocks() as demo:
     gr.Markdown("# Luma AI Text-to-Video Demo")
     
+    camera_motions = gr.State([])
+    
     with gr.Tab("Text to Video"):
         prompt = gr.Textbox(label="Prompt")
         generate_btn = gr.Button("Generate Video")
@@ -188,7 +187,7 @@ with gr.Blocks() as demo:
         with gr.Accordion("Advanced Options", open=False):
             loop = gr.Checkbox(label="Loop", value=False)
             aspect_ratio = gr.Dropdown(label="Aspect Ratio", choices=["16:9", "1:1", "9:16", "4:3", "3:4"], value="16:9")
-            camera_motion = gr.Dropdown(label="Camera Motion", choices=camera_motions)
+            camera_motion = gr.Dropdown(label="Camera Motion", choices=[])
             extend_id = gr.Textbox(label="Extend Video ID (optional)")
             reverse_extend_id = gr.Textbox(label="Reverse Extend Video ID (optional)")
             with gr.Row():
@@ -211,7 +210,7 @@ with gr.Blocks() as demo:
         with gr.Accordion("Advanced Options", open=False):
             img_loop = gr.Checkbox(label="Loop", value=False)
             img_aspect_ratio = gr.Dropdown(label="Aspect Ratio", choices=["16:9", "1:1", "9:16", "4:3", "3:4"], value="16:9")
-            img_camera_motion = gr.Dropdown(label="Camera Motion", choices=camera_motions)
+            img_camera_motion = gr.Dropdown(label="Camera Motion", choices=[])
         
         img_generate_btn.click(
             image_to_video,
@@ -219,4 +218,10 @@ with gr.Blocks() as demo:
             outputs=[img_video_output, img_error_output]
         )
 
-demo.queue().launch(share=True)
+    async def update_camera_motions():
+        motions = await get_camera_motions()
+        return gr.Dropdown.update(choices=motions), gr.Dropdown.update(choices=motions)
+
+    demo.load(update_camera_motions, outputs=[camera_motion, img_camera_motion])
+
+demo.queue().launch(debug=True)