Update app.py

app.py

@@ -585,52 +585,189 @@ def create_advanced_options():
             num_frames_slider,
         ]
 
-with gr.Blocks(theme=gr.themes.Soft()) as iface:
-
-    with gr.Column():
-        txt2vid_prompt = gr.Textbox(
-            label="Step 1: Enter Your Prompt (한글 또는 영어)",
-            placeholder="생성하고 싶은 비디오를 설명하세요 (최소 50자)...",
-            value="긴 갈색 머리와 밝은 피부를 가진 여성이 긴 금발 머리를 가진 다른 여성을 향해 미소 짓습니다. 갈색 머리 여성은 검은 재킷을 입고 있으며 오른쪽 뺨에 작고 거의 눈에 띄지 않는 점이 있습니다. 카메라 앵글은 갈색 머리 여성의 얼굴에 초점을 맞춘 클로즈업입니다. 조명은 따뜻하고 자연스러우며, 아마도 지는 해에서 나오는 것 같아 장면에 부드러운 빛을 비춥니다.",
-            lines=5,
-        )
-
-
-        txt2vid_enhance_toggle = Toggle(
-            label="Enhance Prompt",
-            value=False,
-            interactive=True,
-        )
-
-        txt2vid_negative_prompt = gr.Textbox(
-            label="Step 2: Enter Negative Prompt",
-            placeholder="비디오에서 원하지 않는 요소를 설명하세요...",
-            value="low quality, worst quality, deformed, distorted, damaged, motion blur, motion artifacts, fused fingers, incorrect anatomy, strange hands, ugly",
-            lines=2,
-        )
-
-        txt2vid_preset = gr.Dropdown(
-            choices=[p["label"] for p in preset_options],
-            value="512x512, 160 frames",
-            label="Step 3.1: Choose Resolution Preset",
-        )
-
-        txt2vid_frame_rate = gr.Slider(
-            label="Step 3.2: Frame Rate",
-            minimum=6,
-            maximum=60,
-            step=1,
-            value=20,
-        )
-
-        txt2vid_advanced = create_advanced_options()
-        txt2vid_generate = gr.Button(
-            "Step 5: Generate Video",
-            variant="primary",
-            size="lg",
-        )
-
-        txt2vid_output = gr.Video(label="Generated Output")
+# CSS 스타일 정의
+css = """
+[이전의 CSS 코드를 그대로 유지]
+"""
+
+# Gradio 인터페이스 생성
+with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
+    gr.HTML('<div class="title">AI Image & Video Generator</div>')
+
+    with gr.Tabs():
+        # 첫 번째 탭: Image Generation
+        with gr.Tab("Image Generation"):
+            with gr.Row():
+                with gr.Column(scale=3):
+                    img_prompt = gr.Textbox(
+                        label="Image Description",
+                        placeholder="이미지 설명을 입력하세요... (한글 입력 가능)",
+                        lines=3
+                    )
+                    
+                    with gr.Accordion("Advanced Settings", open=False):
+                        with gr.Row():
+                            height = gr.Slider(
+                                label="Height",
+                                minimum=256,
+                                maximum=1152,
+                                step=64,
+                                value=1024
+                            )
+                            width = gr.Slider(
+                                label="Width",
+                                minimum=256,
+                                maximum=1152,
+                                step=64,
+                                value=1024
+                            )
+                        
+                        with gr.Row():
+                            steps = gr.Slider(
+                                label="Inference Steps",
+                                minimum=6,
+                                maximum=25,
+                                step=1,
+                                value=8
+                            )
+                            scales = gr.Slider(
+                                label="Guidance Scale",
+                                minimum=0.0,
+                                maximum=5.0,
+                                step=0.1,
+                                value=3.5
+                            )
+                        
+                        def get_random_seed():
+                            return torch.randint(0, 1000000, (1,)).item()
+                        
+                        seed = gr.Number(
+                            label="Seed",
+                            value=get_random_seed(),
+                            precision=0
+                        )
+                        
+                        randomize_seed = gr.Button("🎲 Randomize Seed", elem_classes=["generate-btn"])
+                    
+                    generate_btn = gr.Button(
+                        "✨ Generate Image",
+                        elem_classes=["generate-btn"]
+                    )
+
+                with gr.Column(scale=4):
+                    img_output = gr.Image(
+                        label="Generated Image",
+                        type="pil",
+                        format="png"
+                    )
+                    img_gallery = gr.Gallery(
+                        label="Image Gallery",
+                        show_label=True,
+                        elem_id="gallery",
+                        columns=[4],
+                        rows=[2],
+                        height="auto",
+                        object_fit="cover"
+                    )
+                    img_gallery.value = load_gallery()
+
+        # 두 번째 탭: Video Generation
+        with gr.Tab("Video Generation"):
+            with gr.Row():
+                with gr.Column(scale=3):
+                    video_prompt = gr.Textbox(
+                        label="Video Description",
+                        placeholder="비디오 설명을 입력하세요... (한글 입력 가능)",
+                        lines=3
+                    )
+                    upload_image = gr.Image(
+                        type="filepath",
+                        label="Upload First Frame Image"
+                    )
+                    video_generate_btn = gr.Button(
+                        "🎬 Generate Video",
+                        elem_classes=["generate-btn"]
+                    )
+
+                with gr.Column(scale=4):
+                    video_output = gr.Video(label="Generated Video")
+                    video_gallery = gr.Gallery(
+                        label="Video Gallery",
+                        show_label=True,
+                        columns=[4],
+                        rows=[2],
+                        height="auto",
+                        object_fit="cover"
+                    )
+
+        # 세 번째 탭: AI Video Generation
+        with gr.Tab("AI Video Generation"):
+            with gr.Column():
+                txt2vid_prompt = gr.Textbox(
+                    label="Step 1: Enter Your Prompt (한글 또는 영어)",
+                    placeholder="생성하고 싶은 비디오를 설명하세요 (최소 50자)...",
+                    value="긴 갈색 머리와 밝은 피부를 가진 여성이 긴 금발 머리를 가진 다른 여성을 향해 미소 짓습니다. 갈색 머리 여성은 검은 재킷을 입고 있으며 오른쪽 뺨에 작고 거의 눈에 띄지 않는 점이 있습니다. 카메라 앵글은 갈색 머리 여성의 얼굴에 초점을 맞춘 클로즈업입니다. 조명은 따뜻하고 자연스러우며, 아마도 지는 해에서 나오는 것 같아 장면에 부드러운 빛을 비춥니다.",
+                    lines=5,
+                )
+
+                txt2vid_enhance_toggle = Toggle(
+                    label="Enhance Prompt",
+                    value=False,
+                    interactive=True,
+                )
+
+                txt2vid_negative_prompt = gr.Textbox(
+                    label="Step 2: Enter Negative Prompt",
+                    placeholder="비디오에서 원하지 않는 요소를 설명하세요...",
+                    value="low quality, worst quality, deformed, distorted, damaged, motion blur, motion artifacts, fused fingers, incorrect anatomy, strange hands, ugly",
+                    lines=2,
+                )
+
+                txt2vid_preset = gr.Dropdown(
+                    choices=[p["label"] for p in preset_options],
+                    value="512x512, 160 frames",
+                    label="Step 3.1: Choose Resolution Preset",
+                )
+
+                txt2vid_frame_rate = gr.Slider(
+                    label="Step 3.2: Frame Rate",
+                    minimum=6,
+                    maximum=60,
+                    step=1,
+                    value=20,
+                )
+
+                txt2vid_advanced = create_advanced_options()
+                txt2vid_generate = gr.Button(
+                    "Step 5: Generate Video",
+                    variant="primary",
+                    size="lg",
+                )
+
+                txt2vid_output = gr.Video(label="Generated Output")
+
+    # 이벤트 핸들러 연결
+    generate_btn.click(
+        process_and_save_image,
+        inputs=[height, width, steps, scales, img_prompt, seed],
+        outputs=[img_output, img_gallery]
+    )
+    
+    video_generate_btn.click(
+        process_and_generate_video,
+        inputs=[upload_image, video_prompt],
+        outputs=video_output
+    )
+    
+    randomize_seed.click(
+        update_seed,
+        outputs=[seed]
+    )
+    
+    generate_btn.click(
+        update_seed,
+        outputs=[seed]
+    )
 
     txt2vid_preset.change(
         fn=preset_changed,
@@ -653,5 +790,5 @@ with gr.Blocks(theme=gr.themes.Soft()) as iface:
         queue=True,
     )
 
-iface.queue(max_size=64, default_concurrency_limit=1, api_open=False).launch(share=True, show_api=False)
-# ===== Application Startup at 2024-12-20 01:30:34 =====
+if __name__ == "__main__":
+    demo.launch(allowed_paths=[PERSISTENT_DIR])