Update app.py

app.py

@@ -22,6 +22,9 @@ async def get_camera_motions():
         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()):
+    # aspect_ratio에서 실제 비율만 추출
+    aspect_ratio = aspect_ratio.split()[0]
+
     generation_params = {
         "prompt": prompt,
         "loop": loop,
@@ -117,6 +120,9 @@ async def text_to_video(prompt, loop, aspect_ratio, camera_motion, extend_id, re
         print(f"원본 프롬프트: {prompt}")
         print(f"번역된 프롬프트: {translated_prompt}")
 
+        # aspect_ratio에서 실제 비율만 추출
+        aspect_ratio = aspect_ratio.split()[0]
+
         interpolate_ids = None
         if interpolate_id1 and interpolate_id2:
             interpolate_ids = [interpolate_id1, interpolate_id2]
@@ -142,6 +148,9 @@ async def image_to_video(prompt, image_url, loop, aspect_ratio, camera_motion, p
         print(f"원본 프롬프트: {prompt}")
         print(f"번역된 프롬프트: {translated_prompt}")
 
+        # aspect_ratio에서 실제 비율만 추출
+        aspect_ratio = aspect_ratio.split()[0]
+
         generation_params = {
             "prompt": translated_prompt + (f" {camera_motion}" if camera_motion else ""),
             "loop": loop,
@@ -208,7 +217,7 @@ with gr.Blocks() as demo:
             loop = gr.Checkbox(label="루프 (비디오를 반복 재생할지 설정)", value=False)
             aspect_ratio = gr.Dropdown(
                 label="화면 비율", 
-                choices=["16:9 (와이드스크린)", "1:1 (정사각형)", "9:16 (세로 영상)", "4:3 (표준)", "3:4 (세로 표준)"], 
+                choices=["16:9 (와이드스크린)", "1:1 (정사각형)", "9:16 (세로 영상)", "4:3 (표준)", "3:4 (세로 표준)", "21:9 (울트라와이드)", "9:21 (세로 울트라와이드)"], 
                 value="16:9 (와이드스크린)"
             )
             camera_motion = gr.Dropdown(label="카메라 모션 (카메라 움직임 효과 선택)")
@@ -235,7 +244,7 @@ with gr.Blocks() as demo:
             img_loop = gr.Checkbox(label="루프 (비디오를 반복 재생할지 설정)", value=False)
             img_aspect_ratio = gr.Dropdown(
                 label="화면 비율", 
-                choices=["16:9 (와이드스크린)", "1:1 (정사각형)", "9:16 (세로 영상)", "4:3 (표준)", "3:4 (세로 표준)"], 
+                choices=["16:9 (와이드스크린)", "1:1 (정사각형)", "9:16 (세로 영상)", "4:3 (표준)", "3:4 (세로 표준)", "21:9 (울트라와이드)", "9:21 (세로 울트라와이드)"], 
                 value="16:9 (와이드스크린)"
             )
             img_camera_motion = gr.Dropdown(label="카메라 모션 (카메라 움직임 효과 선택)")