Update app.py

app.py

@@ -228,6 +228,7 @@ preset_options = [
 def preset_changed(preset):
     if preset != "Custom":
         selected = next(item for item in preset_options if item["label"] == preset)
+        # height, width, num_frames 값을 global 변수로 업데이트
         return (
             selected["height"],
             selected["width"],
@@ -246,6 +247,7 @@ def preset_changed(preset):
             gr.update(visible=True),
         )
 
+
 def generate_video_from_text(
     prompt="",
     enhance_prompt_toggle=False,
@@ -502,6 +504,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                         visible=False
                     )
 
+                    # 현재 선택된 값들을 저장할 상태 변수들
+                    txt2vid_current_height = gr.State(value=512)
+                    txt2vid_current_width = gr.State(value=320)
+                    txt2vid_current_num_frames = gr.State(value=257)
+
                     txt2vid_preset = gr.Dropdown(
                         choices=[p["label"] for p in preset_options],
                         value="512x320, 10.3초",
@@ -555,9 +562,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                         visible=False
                     )
 
+                    # 현재 선택된 값들을 저장할 상태 변수들
+                    img2vid_current_height = gr.State(value=512)
+                    img2vid_current_width = gr.State(value=768)
+                    img2vid_current_num_frames = gr.State(value=97)
+
                     img2vid_preset = gr.Dropdown(
                         choices=[p["label"] for p in preset_options],
-                        value="768x512, 97 frames",
+                        value="768x512, 3.9초",
                         label="Step 3: 해상도 프리셋 선택",
                     )
 
@@ -582,9 +594,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
 
     # Event handlers
     txt2vid_preset.change(
-        fn=preset_changed, 
-        inputs=[txt2vid_preset], 
-        outputs=txt2vid_advanced[3:]
+        fn=preset_changed,
+        inputs=[txt2vid_preset],
+        outputs=[
+            txt2vid_current_height,
+            txt2vid_current_width,
+            txt2vid_current_num_frames,
+            *txt2vid_advanced[3:]
+        ]
     )
 
     txt2vid_enhance_toggle.change(
@@ -600,7 +617,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
             txt2vid_enhance_toggle,
             txt2vid_negative_prompt,
             txt2vid_frame_rate,
-            *txt2vid_advanced,
+            *txt2vid_advanced[:3],  # seed, inference_steps, guidance_scale
+            txt2vid_current_height,
+            txt2vid_current_width,
+            txt2vid_current_num_frames,
         ],
         outputs=txt2vid_output,
         concurrency_limit=1,
@@ -609,9 +629,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
     )
 
     img2vid_preset.change(
-        fn=preset_changed, 
-        inputs=[img2vid_preset], 
-        outputs=img2vid_advanced[3:]
+        fn=preset_changed,
+        inputs=[img2vid_preset],
+        outputs=[
+            img2vid_current_height,
+            img2vid_current_width,
+            img2vid_current_num_frames,
+            *img2vid_advanced[3:]
+        ]
     )
 
     img2vid_enhance_toggle.change(
@@ -628,7 +653,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
             img2vid_enhance_toggle,
             img2vid_negative_prompt,
             img2vid_frame_rate,
-            *img2vid_advanced,
+            *img2vid_advanced[:3],  # seed, inference_steps, guidance_scale
+            img2vid_current_height,
+            img2vid_current_width,
+            img2vid_current_num_frames,
         ],
         outputs=img2vid_output,
         concurrency_limit=1,
@@ -639,4 +667,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
 if __name__ == "__main__":
     iface.queue(max_size=64, default_concurrency_limit=1, api_open=False).launch(
         share=True, show_api=False
-    )            
+    )