Update app.py

app.py

@@ -347,228 +347,225 @@ def leffa_predict_pt(src_image_path, ref_image_path):
 # 초기 설정 실행
 setup()    
     
-# Gradio 인터페이스
-with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange") as demo:
-    gr.Markdown("# 🎭 FitGen:Fashion Studio & Virtual Try-on")
-    
-    with gr.Tabs():
-        # 패션 생성 탭
-
-# 패션 생성 탭
-        with gr.Tab("Fashion Generation"):
-            with gr.Column():
-                mode = gr.Radio(
-                    choices=["Generate Model", "Generate Clothes"],
-                    label="Generation Mode",
-                    value="Generate Model"
-                )
-        
-        # 예제 프롬프트 설정
-                example_model_prompts = [
-                    "professional fashion model, full body shot, standing pose, natural lighting, studio background, high fashion, elegant pose",
-                    "fashion model portrait, upper body, confident pose, fashion photography, neutral background, professional lighting",
-                    "stylish fashion model, three-quarter view, editorial pose, high-end fashion magazine style, minimal background"
-                ]
-
-                example_clothes_prompts = [
-                    "luxury designer sweater, cashmere material, cream color, cable knit pattern, high-end fashion, product photography",
-                    "elegant business blazer, tailored fit, charcoal grey, premium wool fabric, professional wear",
-                    "modern streetwear hoodie, oversized fit, minimalist design, premium cotton, urban style"
-                ]
-        
-                prompt = gr.TextArea(
-                    label="Fashion Description (한글 또는 영어)",
-                    placeholder="패션 모델이나 의류를 설명하세요..."
-                )
-        
-        # 예제 섹션 추가
-                gr.Examples(
-                    examples=example_model_prompts + example_clothes_prompts,
-                    inputs=prompt,
-                    label="Example Prompts"
-                )
-        
-                with gr.Row():
-                    with gr.Column():
-                        result = gr.Image(label="Generated Result")
-                        generate_button = gr.Button("Generate Fashion")
+def create_interface():
+    with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange") as demo:
+        gr.Markdown("# 🎭 FitGen:Fashion Studio & Virtual Try-on")
         
-
-                
-                with gr.Accordion("Advanced Options", open=False):
-                    with gr.Group():
-                        with gr.Row():
-                            with gr.Column():
-                                cfg_scale = gr.Slider(
-                                    label="CFG Scale",
-                                    minimum=1,
-                                    maximum=20,
-                                    step=0.5,
-                                    value=7.0
-                                )
-                                steps = gr.Slider(
-                                    label="Steps",
-                                    minimum=1,
-                                    maximum=50,  # 최대값 감소
-                                    step=1,
-                                    value=30
-                                )
-                                lora_scale = gr.Slider(
-                                    label="LoRA Scale",
-                                    minimum=0,
-                                    maximum=1,
-                                    step=0.01,
-                                    value=0.85
-                                )
-                        
-                        with gr.Row():
-                            width = gr.Slider(
-                                label="Width",
-                                minimum=256,
-                                maximum=1024,  # 최대값 감소
-                                step=64,
-                                value=512
-                            )
-                            height = gr.Slider(
-                                label="Height",
-                                minimum=256,
-                                maximum=1024,  # 최대값 감소
-                                step=64,
-                                value=768
-                            )
-                        
-                        with gr.Row():
-                            randomize_seed = gr.Checkbox(
-                                True,
-                                label="Randomize seed"
-                            )
-                            seed = gr.Slider(
-                                label="Seed",
-                                minimum=0,
-                                maximum=MAX_SEED,
-                                step=1,
-                                value=42
-                            )
-
-        # 가상 피팅 탭
-        with gr.Tab("Virtual Try-on"):
-            with gr.Row():
+        with gr.Tabs():
+            # 패션 생성 탭
+            with gr.Tab("Fashion Generation"):
                 with gr.Column():
-                    gr.Markdown("#### Person Image")
-                    vt_src_image = gr.Image(
-                        sources=["upload"],
-                        type="filepath",
-                        label="Person Image",
-                        width=512,
-                        height=512,
-                    )
-                    gr.Examples(
-                        inputs=vt_src_image,
-                        examples_per_page=5,
-                        examples=["a1.webp",
-                                "a2.webp",
-                                "a3.webp",
-                                "a4.webp",
-                                "a5.webp"]
+                    mode = gr.Radio(
+                        choices=["Generate Model", "Generate Clothes"],
+                        label="Generation Mode",
+                        value="Generate Model"
                     )
-
-                with gr.Column():
-                    gr.Markdown("#### Garment Image")
-                    vt_ref_image = gr.Image(
-                        sources=["upload"],
-                        type="filepath",
-                        label="Garment Image",
-                        width=512,
-                        height=512,
+            
+                    # 예제 프롬프트 설정
+                    example_model_prompts = [
+                        "professional fashion model, full body shot, standing pose, natural lighting, studio background, high fashion, elegant pose",
+                        "fashion model portrait, upper body, confident pose, fashion photography, neutral background, professional lighting", 
+                        "stylish fashion model, three-quarter view, editorial pose, high-end fashion magazine style, minimal background"
+                    ]
+
+                    example_clothes_prompts = [
+                        "luxury designer sweater, cashmere material, cream color, cable knit pattern, high-end fashion, product photography",
+                        "elegant business blazer, tailored fit, charcoal grey, premium wool fabric, professional wear",
+                        "modern streetwear hoodie, oversized fit, minimalist design, premium cotton, urban style"
+                    ]
+            
+                    prompt = gr.TextArea(
+                        label="Fashion Description (한글 또는 영어)",
+                        placeholder="패션 모델이나 의류를 설명하세요..."
                     )
+            
+                    # 예제 섹션 추가
                     gr.Examples(
-                        inputs=vt_ref_image,
-                        examples_per_page=5,
-                        examples=["b1.webp",
-                                "b2.webp",
-                                "b3.webp",
-                                "b4.webp",
-                                "b5.webp"]
+                        examples=example_model_prompts + example_clothes_prompts,
+                        inputs=prompt,
+                        label="Example Prompts"
                     )
+            
+                    with gr.Row():
+                        with gr.Column():
+                            result = gr.Image(label="Generated Result")
+                            generate_button = gr.Button("Generate Fashion")
+            
+                    with gr.Accordion("Advanced Options", open=False):
+                        with gr.Group():
+                            with gr.Row():
+                                with gr.Column():
+                                    cfg_scale = gr.Slider(
+                                        label="CFG Scale",
+                                        minimum=1,
+                                        maximum=20,
+                                        step=0.5,
+                                        value=7.0
+                                    )
+                                    steps = gr.Slider(
+                                        label="Steps",
+                                        minimum=1,
+                                        maximum=30,
+                                        step=1,
+                                        value=30
+                                    )
+                                    lora_scale = gr.Slider(
+                                        label="LoRA Scale",
+                                        minimum=0,
+                                        maximum=1,
+                                        step=0.01,
+                                        value=0.85
+                                    )
+                            
+                            with gr.Row():
+                                width = gr.Slider(
+                                    label="Width",
+                                    minimum=256,
+                                    maximum=768,
+                                    step=64,
+                                    value=512
+                                )
+                                height = gr.Slider(
+                                    label="Height", 
+                                    minimum=256,
+                                    maximum=768,
+                                    step=64,
+                                    value=768
+                                )
+                            
+                            with gr.Row():
+                                randomize_seed = gr.Checkbox(
+                                    True,
+                                    label="Randomize seed"
+                                )
+                                seed = gr.Slider(
+                                    label="Seed",
+                                    minimum=0,
+                                    maximum=2**32-1,
+                                    step=1,
+                                    value=42
+                                )
 
-                with gr.Column():
-                    gr.Markdown("#### Generated Image")
-                    vt_gen_image = gr.Image(
-                        label="Generated Image",
-                        width=512,
-                        height=512,
-                    )
-                    vt_gen_button = gr.Button("Try-on")
+            # 가상 피팅 탭
+            with gr.Tab("Virtual Try-on"):
+                with gr.Row():
+                    with gr.Column():
+                        gr.Markdown("#### Person Image")
+                        vt_src_image = gr.Image(
+                            sources=["upload"],
+                            type="filepath",
+                            label="Person Image",
+                            width=512,
+                            height=512,
+                        )
+                        gr.Examples(
+                            inputs=vt_src_image,
+                            examples_per_page=5,
+                            examples=["a1.webp",
+                                    "a2.webp",
+                                    "a3.webp",
+                                    "a4.webp",
+                                    "a5.webp"]
+                        )
 
-        # 포즈 전송 탭
-        with gr.Tab("Pose Transfer"):
-            with gr.Row():
-                with gr.Column():
-                    gr.Markdown("#### Person Image")
-                    pt_ref_image = gr.Image(
-                        sources=["upload"],
-                        type="filepath",
-                        label="Person Image",
-                        width=512,
-                        height=512,
-                    )
-                    gr.Examples(
-                        inputs=pt_ref_image,
-                        examples_per_page=5,
-                        examples=["a1.webp",
-                                "a2.webp",
-                                "a3.webp",
-                                "a4.webp",
-                                "a5.webp"]
-                    )
+                    with gr.Column():
+                        gr.Markdown("#### Garment Image")
+                        vt_ref_image = gr.Image(
+                            sources=["upload"],
+                            type="filepath",
+                            label="Garment Image",
+                            width=512,
+                            height=512,
+                        )
+                        gr.Examples(
+                            inputs=vt_ref_image,
+                            examples_per_page=5,
+                            examples=["b1.webp",
+                                    "b2.webp",
+                                    "b3.webp",
+                                    "b4.webp",
+                                    "b5.webp"]
+                        )
 
-                with gr.Column():
-                    gr.Markdown("#### Target Pose Person Image")
-                    pt_src_image = gr.Image(
-                        sources=["upload"],
-                        type="filepath",
-                        label="Target Pose Person Image",
-                        width=512,
-                        height=512,
-                    )
-                    gr.Examples(
-                        inputs=pt_src_image,
-                        examples_per_page=5,
-                        examples=["d1.webp",
-                                "d2.webp",
-                                "d3.webp",
-                                "d4.webp",
-                                "d5.webp"]
-                    )
+                    with gr.Column():
+                        gr.Markdown("#### Generated Image")
+                        vt_gen_image = gr.Image(
+                            label="Generated Image",
+                            width=512,
+                            height=512,
+                        )
+                        vt_gen_button = gr.Button("Try-on")
+
+            # 포즈 전송 탭
+            with gr.Tab("Pose Transfer"):
+                with gr.Row():
+                    with gr.Column():
+                        gr.Markdown("#### Person Image")
+                        pt_ref_image = gr.Image(
+                            sources=["upload"],
+                            type="filepath",
+                            label="Person Image",
+                            width=512,
+                            height=512,
+                        )
+                        gr.Examples(
+                            inputs=pt_ref_image,
+                            examples_per_page=5,
+                            examples=["a1.webp",
+                                    "a2.webp",
+                                    "a3.webp",
+                                    "a4.webp",
+                                    "a5.webp"]
+                        )
 
-                with gr.Column():
-                    gr.Markdown("#### Generated Image")
-                    pt_gen_image = gr.Image(
-                        label="Generated Image",
-                        width=512,
-                        height=512,
-                    )
-                    pose_transfer_gen_button = gr.Button("Generate")
-
-    # 이벤트 핸들러
-    generate_button.click(
-        generate_fashion,
-        inputs=[prompt, mode, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale],
-        outputs=[result, seed]
-    )
-    
-    vt_gen_button.click(
-        fn=leffa_predict_vt,
-        inputs=[vt_src_image, vt_ref_image],
-        outputs=[vt_gen_image]
-    )
-    
-    pose_transfer_gen_button.click(
-        fn=leffa_predict_pt,
-        inputs=[pt_src_image, pt_ref_image],
-        outputs=[pt_gen_image]
-    )
+                    with gr.Column():
+                        gr.Markdown("#### Target Pose Person Image")
+                        pt_src_image = gr.Image(
+                            sources=["upload"],
+                            type="filepath",
+                            label="Target Pose Person Image",
+                            width=512,
+                            height=512,
+                        )
+                        gr.Examples(
+                            inputs=pt_src_image,
+                            examples_per_page=5,
+                            examples=["d1.webp",
+                                    "d2.webp",
+                                    "d3.webp",
+                                    "d4.webp",
+                                    "d5.webp"]
+                        )
+
+                    with gr.Column():
+                        gr.Markdown("#### Generated Image")
+                        pt_gen_image = gr.Image(
+                            label="Generated Image",
+                            width=512,
+                            height=512,
+                        )
+                        pose_transfer_gen_button = gr.Button("Generate")
+
+        # 이벤트 핸들러
+        generate_button.click(
+            fn=generate_image,
+            inputs=[prompt, mode, cfg_scale, steps, seed, width, height, lora_scale],
+            outputs=[result, seed]
+        )
+        
+        vt_gen_button.click(
+            fn=leffa_predict_vt,
+            inputs=[vt_src_image, vt_ref_image],
+            outputs=[vt_gen_image]
+        )
+        
+        pose_transfer_gen_button.click(
+            fn=leffa_predict_pt,
+            inputs=[pt_src_image, pt_ref_image],
+            outputs=[pt_gen_image]
+        )
 
+        return demo
 
 if __name__ == "__main__":
     # 환경 설정
@@ -580,4 +577,4 @@ if __name__ == "__main__":
         server_name="0.0.0.0",
         server_port=7860,
         share=False
-    ) 
+    )