Update app-backup1.py

app-backup1.py

@@ -196,39 +196,39 @@ pipeline = XoraVideoPipeline(
 ).to(device)
 
 # Preset options for resolution and frame configuration
+# Convert frames to seconds assuming 25 FPS
 preset_options = [
-    {"label": "1216x704, 41 frames", "width": 1216, "height": 704, "num_frames": 41},
-    {"label": "1088x704, 49 frames", "width": 1088, "height": 704, "num_frames": 49},
-    {"label": "1056x640, 57 frames", "width": 1056, "height": 640, "num_frames": 57},
-    {"label": "992x608, 65 frames", "width": 992, "height": 608, "num_frames": 65},
-    {"label": "896x608, 73 frames", "width": 896, "height": 608, "num_frames": 73},
-    {"label": "896x544, 81 frames", "width": 896, "height": 544, "num_frames": 81},
-    {"label": "832x544, 89 frames", "width": 832, "height": 544, "num_frames": 89},
-    {"label": "800x512, 97 frames", "width": 800, "height": 512, "num_frames": 97},
-    {"label": "768x512, 97 frames", "width": 768, "height": 512, "num_frames": 97},
-    {"label": "800x480, 105 frames", "width": 800, "height": 480, "num_frames": 105},
-    {"label": "736x480, 113 frames", "width": 736, "height": 480, "num_frames": 113},
-    {"label": "704x480, 121 frames", "width": 704, "height": 480, "num_frames": 121},
-    {"label": "704x448, 129 frames", "width": 704, "height": 448, "num_frames": 129},
-    {"label": "672x448, 137 frames", "width": 672, "height": 448, "num_frames": 137},
-    {"label": "640x416, 153 frames", "width": 640, "height": 416, "num_frames": 153},
-    {"label": "672x384, 161 frames", "width": 672, "height": 384, "num_frames": 161},
-    {"label": "640x384, 169 frames", "width": 640, "height": 384, "num_frames": 169},
-    {"label": "608x384, 177 frames", "width": 608, "height": 384, "num_frames": 177},
-    {"label": "576x384, 185 frames", "width": 576, "height": 384, "num_frames": 185},
-    {"label": "608x352, 193 frames", "width": 608, "height": 352, "num_frames": 193},
-    {"label": "576x352, 201 frames", "width": 576, "height": 352, "num_frames": 201},
-    {"label": "544x352, 209 frames", "width": 544, "height": 352, "num_frames": 209},
-    {"label": "512x352, 225 frames", "width": 512, "height": 352, "num_frames": 225},
-    {"label": "512x352, 233 frames", "width": 512, "height": 352, "num_frames": 233},
-    {"label": "544x320, 241 frames", "width": 544, "height": 320, "num_frames": 241},
-    {"label": "512x320, 249 frames", "width": 512, "height": 320, "num_frames": 249},
-    {"label": "512x320, 257 frames", "width": 512, "height": 320, "num_frames": 257},
+    {"label": "1216x704, 1.6초", "width": 1216, "height": 704, "num_frames": 41},
+    {"label": "1088x704, 2.0초", "width": 1088, "height": 704, "num_frames": 49},
+    {"label": "1056x640, 2.3초", "width": 1056, "height": 640, "num_frames": 57},
+    {"label": "992x608, 2.6초", "width": 992, "height": 608, "num_frames": 65},
+    {"label": "896x608, 2.9초", "width": 896, "height": 608, "num_frames": 73},
+    {"label": "896x544, 3.2초", "width": 896, "height": 544, "num_frames": 81},
+    {"label": "832x544, 3.6초", "width": 832, "height": 544, "num_frames": 89},
+    {"label": "800x512, 3.9초", "width": 800, "height": 512, "num_frames": 97},
+    {"label": "768x512, 3.9초", "width": 768, "height": 512, "num_frames": 97},
+    {"label": "800x480, 4.2초", "width": 800, "height": 480, "num_frames": 105},
+    {"label": "736x480, 4.5초", "width": 736, "height": 480, "num_frames": 113},
+    {"label": "704x480, 4.8초", "width": 704, "height": 480, "num_frames": 121},
+    {"label": "704x448, 5.2초", "width": 704, "height": 448, "num_frames": 129},
+    {"label": "672x448, 5.5초", "width": 672, "height": 448, "num_frames": 137},
+    {"label": "640x416, 6.1초", "width": 640, "height": 416, "num_frames": 153},
+    {"label": "672x384, 6.4초", "width": 672, "height": 384, "num_frames": 161},
+    {"label": "640x384, 6.8초", "width": 640, "height": 384, "num_frames": 169},
+    {"label": "608x384, 7.1초", "width": 608, "height": 384, "num_frames": 177},
+    {"label": "576x384, 7.4초", "width": 576, "height": 384, "num_frames": 185},
+    {"label": "608x352, 7.7초", "width": 608, "height": 352, "num_frames": 193},
+    {"label": "576x352, 8.0초", "width": 576, "height": 352, "num_frames": 201},
+    {"label": "544x352, 8.4초", "width": 544, "height": 352, "num_frames": 209},
+    {"label": "512x352, 9.3초", "width": 512, "height": 352, "num_frames": 233},
+    {"label": "544x320, 9.6초", "width": 544, "height": 320, "num_frames": 241},
+    {"label": "512x320, 10.3초", "width": 512, "height": 320, "num_frames": 257},
 ]
 
 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"],
@@ -247,17 +247,18 @@ def preset_changed(preset):
             gr.update(visible=True),
         )
 
+
 def generate_video_from_text(
     prompt="",
     enhance_prompt_toggle=False,
-    negative_prompt="",
+    negative_prompt="low quality, worst quality, deformed, distorted, warped, motion smear, motion artifacts, fused fingers, incorrect anatomy, strange hands, unattractive",
     frame_rate=25,
     seed=171198,
-    num_inference_steps=30,
-    guidance_scale=3,
+    num_inference_steps=41,
+    guidance_scale=4,
     height=512,
-    width=768,
-    num_frames=121,
+    width=320,
+    num_frames=257,
     progress=gr.Progress(),
 ):
     if len(prompt.strip()) < 50:
@@ -331,11 +332,11 @@ def generate_video_from_image(
     image_path,
     prompt="",
     enhance_prompt_toggle=False,
-    negative_prompt="",
+    negative_prompt="low quality, worst quality, deformed, distorted, warped, motion smear, motion artifacts, fused fingers, incorrect anatomy, strange hands, unattractive",
     frame_rate=25,
     seed=171198,
-    num_inference_steps=30,
-    guidance_scale=3,
+    num_inference_steps=50,
+    guidance_scale=4,
     height=512,
     width=768,
     num_frames=121,
@@ -420,15 +421,28 @@ def generate_video_from_image(
 def create_advanced_options():
     with gr.Accordion("Step 4: Advanced Options (Optional)", open=False):
         seed = gr.Slider(
-            label="4.1 Seed", minimum=0, maximum=1000000, step=1, value=171198
+            label="Seed",
+            minimum=0,
+            maximum=1000000,
+            step=1,
+            value=171198
         )
         inference_steps = gr.Slider(
-            label="4.2 Inference Steps", minimum=1, maximum=50, step=1, value=30
+            label="4.2 Inference Steps",
+            minimum=1,
+            maximum=50,
+            step=1,
+            value=50,
+            visible=False
         )
         guidance_scale = gr.Slider(
-            label="4.3 Guidance Scale", minimum=1.0, maximum=5.0, step=0.1, value=3.0
+            label="4.3 Guidance Scale",
+            minimum=1.0,
+            maximum=5.0,
+            step=0.1,
+            value=4.0,
+            visible=False
         )
-
         height_slider = gr.Slider(
             label="4.4 Height",
             minimum=256,
@@ -473,7 +487,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                     txt2vid_prompt = gr.Textbox(
                         label="Step 1: 프롬프트 입력",
                         placeholder="생성하고 싶은 비디오를 설명하세요 (최소 50자)...",
-                        value="갈색 긴 머리를 가진 여성이 금발의 긴 머리를 가진 다른 여성을 향해 미소짓습니다. 갈색 머리의 여성은 검은색 자켓을 입고 있으며 오른쪽 뺨에 작은 점이 있습니다. 카메라 각도는 갈색 머리 여성의 얼굴에 클로즈업되어 있습니다. 조명은 자연스럽고 따뜻하며, 석양에서 오는 듯한 부드러운 빛이 장면을 비춥니다. 장면은 실제 영상처럼 보입니다.",
+                        value="귀여운 고양이",
                         lines=5,
                     )
                     txt2vid_enhance_toggle = Toggle(
@@ -487,25 +501,32 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                         placeholder="비디오에서 원하지 않는 요소를 설명하세요...",
                         value="low quality, worst quality, deformed, distorted, warped, motion smear, motion artifacts, fused fingers, incorrect anatomy, strange hands, unattractive",
                         lines=2,
+                        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="768x512, 97 frames",
-                        label="Step 3.1: 해상도 프리셋 선택",
+                        value="512x320, 10.3초",
+                        label="Step 2: 해상도 프리셋 선택",
                     )
 
                     txt2vid_frame_rate = gr.Slider(
-                        label="Step 3.2: 프레임 레이트",
+                        label="Step 3: 프레임 레이트",
                         minimum=21,
                         maximum=30,
                         step=1,
                         value=25,
+                        visible=False
                     )
 
                     txt2vid_advanced = create_advanced_options()
                     txt2vid_generate = gr.Button(
-                        "Step 5: 비디오 생성",
+                        "Step 3: 비디오 생성",
                         variant="primary",
                         size="lg",
                     )
@@ -513,29 +534,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                 with gr.Column():
                     txt2vid_output = gr.Video(label="생성된 비디오")
 
-            with gr.Row():
-                gr.Examples(
-                    examples=[
-                        [
-                            "전통적인 몽골 드레스를 입은 젊은 여성이 얇은 흰색 커튼을 통해 호기심과 긴장이 섞인 표정으로 들여다보고 있습니다. 여성은 흰 구슬로 장식된 두 개의 땋은 머리로 스타일링된 긴 검은 머리를 하고 있으며, 눈은 놀람을 띄며 크게 떠져 있습니다. 그녀의 드레스는 화려한 금색 자수가 새겨진 선명한 파란색이며, 비슷한 디자인의 머리띠를 하고 있습니다. 배경은 신비로움과 호기심을 자아내는 단순한 흰색 커튼입니다.",
-                            "low quality, worst quality, deformed, distorted, warped, motion smear, motion artifacts, fused fingers, incorrect anatomy, strange hands, unattractive",
-                            "assets/t2v_2.mp4",
-                        ],
-                        [
-                            "노란색 재킷을 입은 금발 머리의 젊은 남자가 숲에 서서 주위를 둘러봅니다. 그는 밝은 피부를 가졌고 머리는 가운데 가르마로 스타일링되어 있습니다. 그는 왼쪽을 보고 난 후 오른쪽을 보며, 각 방향을 잠시 응시합니다. 카메라는 낮은 각도에서 남자를 올려다보며 고정되어 있습니다. 배경은 약간 흐릿하며, 녹색 나무들과 남자의 뒤에서 밝게 비치는 태양이 보입니다. 조명은 자연스럽고 따뜻하며, 태양 빛이 남자의 얼굴을 가로지르는 렌즈 플레어를 만듭니다. 장면은 실제 영상처럼 촬영되었습니다.",
-                            "low quality, worst quality, deformed, distorted, warped, motion smear, motion artifacts, fused fingers, incorrect anatomy, strange hands, unattractive",
-                            "assets/t2v_1.mp4",
-                        ],
-                        [
-                            "한 사이클리스트가 굽이진 산길을 따라 달립니다. 공기역학적인 장비를 입은 그는 강하게 페달을 밟고 있으며, 이마에는 땀방울이 반짝입니다. 카메라는 그의 결연한 표정과 숨 막히는 풍경을 번갈아가며 보여줍니다. 소나무들이 스쳐 지나가고, 하늘은 선명한 파란색입니다. 이 장면은 활기차고 경쟁적인 분위기를 자아냅니다.",
-                            "low quality, worst quality, deformed, distorted, warped, motion smear, motion artifacts, fused fingers, incorrect anatomy, strange hands, unattractive",
-                            "assets/t2v_0.mp4",
-                        ],
-                    ],
-                    inputs=[txt2vid_prompt, txt2vid_negative_prompt, txt2vid_output],
-                    label="텍스트-비디오 생성 예시",
-                )
-
         # Image to Video Tab
         with gr.TabItem("이미지로 비디오 만들기"):
             with gr.Row():
@@ -548,11 +546,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                     img2vid_prompt = gr.Textbox(
                         label="Step 2: 프롬프트 입력",
                         placeholder="이미지를 어떻게 애니메이션화할지 설명하세요 (최소 50자)...",
-                        value="갈색 긴 머리를 가진 여성이 금발의 긴 머리를 가진 다른 여성을 향해 미소짓습니다. 갈색 머리의 여성은 검은색 자켓을 입고 있으며 오른쪽 뺨에 작은 점이 있습니다. 카메라 각도는 갈색 머리 여성의 얼굴에 클로즈업되어 있습니다. 조명은 자연스럽고 따뜻하며, 석양에서 오는 듯한 부드러운 빛이 장면을 비춥니다. 장면은 실제 영상처럼 보입니다.",
+                        value="귀여운 고양이",
                         lines=5,
                     )
                     img2vid_enhance_toggle = Toggle(
-                        label="프롬프트 개선",
+                        label="프롬프트 증강",
                         value=False,
                         interactive=True,
                     )
@@ -561,67 +559,49 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
                         placeholder="비디오에서 원하지 않는 요소를 설명하세요...",
                         value="low quality, worst quality, deformed, distorted, warped, motion smear, motion artifacts, fused fingers, incorrect anatomy, strange hands, unattractive",
                         lines=2,
+                        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",
-                        label="Step 3.1: 해상도 프리셋 선택",
+                        value="512x320, 10.3초",
+                        label="Step 3: 해상도 프리셋 선택",
                     )
 
                     img2vid_frame_rate = gr.Slider(
-                        label="Step 3.2: 프레임 레이트",
+                        label="Step 4: 프레임 레이트",
                         minimum=21,
                         maximum=30,
                         step=1,
                         value=25,
+                        visible=False
                     )
 
                     img2vid_advanced = create_advanced_options()
                     img2vid_generate = gr.Button(
-                        "Step 6: 비디오 생성", variant="primary", size="lg"
+                        "Step 4: 비디오 생성",
+                        variant="primary",
+                        size="lg",
                     )
 
                 with gr.Column():
                     img2vid_output = gr.Video(label="생성된 비디오")
 
-            with gr.Row():
-                gr.Examples(
-                    examples=[
-                        [
-                            "assets/i2v_i2.png",
-                            "여성이 흰색 전기 버너 위에서 끓는 물이 담긴 냄비를 젓고 있습니다. 보라색 매니큐어를 바른 그녀의 손이 하얀 냄비 안에서 나무 숟가락을 원형으로 움직입니다. 냄비는 검은색 버튼과 디지털 디스플레이가 있는 흰색 전기 버너 위에 놓여 있습니다. 버너는 오른쪽 아래 모서리에 빨간색과 흰색 체크무늬 천이 부분적으로 보이는 흰색 조리대 위에 놓여 있습니다. 카메라 각도는 정확히 위에서 내려다보는 각도이며 장면 내내 고정되어 있습니다. 조명은 밝고 고른 중성적인 흰색 빛으로 장면을 비춥니다. 장면은 실제 영상처럼 보입니다.",
-                            "low quality, worst quality, deformed, distorted, warped, motion smear, motion artifacts, fused fingers, incorrect anatomy, strange hands, unattractive",
-                            "assets/i2v_2.mp4",
-                        ],
-                        [
-                            "assets/i2v_i0.png",
-                            "긴 흐르는 드레스를 입은 여성이 들판에 서서 등을 카메라를 향한 채 지평선을 바라보고 있습니다. 그녀의 머리카락은 길고 밝으며 등 아래로 흘러내립니다. 그녀는 큰 참나무의 넓게 퍼진 가지 아래에 서 있습니다. 왼쪽으로는 말라붙은 잔디 위에 클래식한 미국 자동차가 주차되어 있습니다. 멀리서는 한 대의 부서진 자동차가 옆으로 누워 있습니다. 위의 하늘은 어두운 하늘을 배경으로 밝은 흰 구름이 극적인 캔버스를 이루고 있습니다. 전체 이미지는 흑백으로, 빛과 그림자의 대비를 강조합니다. 여성이 천천히 자동차를 향해 걸어가고 있습니다.",
-                            "low quality, worst quality, deformed, distorted, warped, motion smear, motion artifacts, fused fingers, incorrect anatomy, strange hands, unattractive",
-                            "assets/i2v_0.mp4",
-                        ],
-                        [
-                            "assets/i2v_i1.png",
-                            "한 쌍의 손이 도자기 물레 위에서 점토 조각을 모양 잡아 점차적으로 원뿔 모양을 만들어가고 있습니다. 프레임 밖의 사람의 손이 점토로 덮여 있으며, 회전하는 도자기 물레 중앙에 점토 덩어리를 부드럽게 누르고 있습니다. 손은 원형으로 움직이며, 점토 위쪽에 점차적으로 원뿔 모양을 만들어갑니다. 카메라는 도자기 물레 바로 위에 위치하여 점토가 모양 잡혀가는 것을 조감도로 보여줍니다. 조명은 밝고 고르며, 점토와 그것을 다루는 손을 밝게 비춥니다. 장면은 실제 영상처럼 촬영되었습니다.",
-                            "low quality, worst quality, deformed, distorted, warped, motion smear, motion artifacts, fused fingers, incorrect anatomy, strange hands, unattractive",
-                            "assets/i2v_1.mp4",
-                        ],
-                    ],
-                    inputs=[
-                        img2vid_image,
-                        img2vid_prompt,
-                        img2vid_negative_prompt,
-                        img2vid_output,
-                    ],
-                    label="이미지-비디오 생성 예시",
-                )
-
     # Event handlers
-# 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(
@@ -637,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,
@@ -646,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(
@@ -665,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,