Update app.py

app.py

@@ -53,37 +53,35 @@ system_prompt_i2v = """당신은 이미지 기반 비디오 생성을 위한 프
 # Default preset
 default_preset = "16:9 (512x320)"
 
-# 초기 default_preset "16:9 (512x320)"에 맞추어 num_frames 초기값 설정
-txt2vid_current_height = gr.State(value=320)
-txt2vid_current_width = gr.State(value=512)
-txt2vid_current_num_frames = gr.State(value=257)  # 10.3초 (257 프레임)
-
-img2vid_current_height = gr.State(value=320)
-img2vid_current_width = gr.State(value=512)
-img2vid_current_num_frames = gr.State(value=257)  # 10.3초 (257 프레임)
-
-
-# preset_options 수정 - 16:9 최소 크기 변경
+# preset_options 수정 - 각 해상도별 정확한 프레임 수 적용
 preset_options = [
     # 16:9 비율 (최대/최소)
-    {"label": "16:9 HD (1216x684)", "width": 1216, "height": 684, "num_frames": 41, "aspect": "16:9"},
-    {"label": "16:9 (512x320)", "width": 512, "height": 320, "num_frames": 81, "aspect": "16:9"},
+    {"label": "16:9 HD (1216x684)", "width": 1216, "height": 684, "num_frames": 41, "aspect": "16:9"},  # 1.6초
+    {"label": "16:9 (512x320)", "width": 512, "height": 320, "num_frames": 257, "aspect": "16:9"},  # 10.3초
     
     # 4:3 비율 (최대/최소)
-    {"label": "4:3 (1024x768)", "width": 1024, "height": 768, "num_frames": 49, "aspect": "4:3"},
-    {"label": "4:3 (640x480)", "width": 640, "height": 480, "num_frames": 73, "aspect": "4:3"},
+    {"label": "4:3 (1024x768)", "width": 1024, "height": 768, "num_frames": 49, "aspect": "4:3"},  # 2.0초
+    {"label": "4:3 (640x480)", "width": 640, "height": 480, "num_frames": 121, "aspect": "4:3"},  # 4.8초
     
     # 1:1 비율 (최대/최소)
-    {"label": "1:1 (896x896)", "width": 896, "height": 896, "num_frames": 57, "aspect": "1:1"},
-    {"label": "1:1 (512x512)", "width": 512, "height": 512, "num_frames": 81, "aspect": "1:1"},
+    {"label": "1:1 (896x896)", "width": 896, "height": 896, "num_frames": 73, "aspect": "1:1"},  # 2.9초
+    {"label": "1:1 (512x512)", "width": 512, "height": 512, "num_frames": 233, "aspect": "1:1"},  # 9.3초
     
     # 3:2 비율 (최대)
-    {"label": "3:2 (1200x800)", "width": 1200, "height": 800, "num_frames": 49, "aspect": "3:2"},
+    {"label": "3:2 (1200x800)", "width": 1200, "height": 800, "num_frames": 41, "aspect": "3:2"},  # 1.6초
     
     # 9:16 비율 (최소)
-    {"label": "9:16 (432x768)", "width": 432, "height": 768, "num_frames": 81, "aspect": "9:16"}
+    {"label": "9:16 (432x768)", "width": 432, "height": 768, "num_frames": 241, "aspect": "9:16"}  # 9.6초
 ]
 
+# State 변수들의 초기값도 수정 (512x320 기준)
+txt2vid_current_height = gr.State(value=320)
+txt2vid_current_width = gr.State(value=512)
+txt2vid_current_num_frames = gr.State(value=257)  # 10.3초
+
+img2vid_current_height = gr.State(value=320)
+img2vid_current_width = gr.State(value=512)
+img2vid_current_num_frames = gr.State(value=257)  # 10.3초
 
 
 # Load Hugging Face token if needed