Update app.py

app.py

@@ -179,118 +179,115 @@ body {
 }
 """
 
-# Gradio 인터페이스 구성
-with gr.Blocks(css=custom_css, theme="Yntec/HaleyCH_Theme_Orange") as app:
-    with gr.Column(elem_classes="container"):
-        gr.Markdown("# 🎭 Fashion Generation Studio", elem_classes="header")
-        
-        # 모드 선택
-        with gr.Group(elem_classes="mode-box box-common"):
-            mode = gr.Radio(
-                choices=["Generate Model", "Generate Clothes"],
-                label="Generation Mode",
-                value="Generate Model"
-            )
-        
-        # 이미지 출력 영역
-        with gr.Group(elem_classes="result-box box-common"):
-            gr.Markdown("### 🖼️ Generated Fashion")
-            result = gr.Image(label="Result", elem_classes="image-output")
-        
-        # 프롬프트 입력 영역
-        with gr.Group(elem_classes="prompt-box box-common"):
-            prompt = gr.TextArea(
-                label="✍️ Fashion Description (한글 또는 영어)",
-                placeholder="패션 모델이나 의류를 설명하세요...",
-                lines=5
-            )
-            generate_button = gr.Button(
-                "🚀 Generate Fashion",
-                elem_classes="generate-btn"
-            )
-
-        # 예제 탭
-        with gr.Tabs():
-            with gr.TabItem("Model Examples"):
-                gr.Examples(
-                    examples=example_model_prompts,
-                    inputs=prompt,
-                    label="Model Prompts"
-                )
-            with gr.TabItem("Clothes Examples"):
-                gr.Examples(
-                    examples=example_clothes_prompts,
-                    inputs=prompt,
-                    label="Clothes Prompts"
-                )
-        
-        # 고급 옵션
-        with gr.Accordion("🎨 Advanced Options", open=False, elem_classes="accordion box-common"):
-            with gr.Group(elem_classes="parameter-box"):
-                gr.Markdown("### 🎛️ Generation Parameters")
-                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=100,
-                            step=1,
-                            value=30
-                        )
-                        lora_scale = gr.Slider(
-                            label="LoRA Scale",
-                            minimum=0,
-                            maximum=1,
-                            step=0.01,
-                            value=0.85
-                        )
-            
-            with gr.Group(elem_classes="parameter-box"):
-                gr.Markdown("### 📐 Image Dimensions")
-                with gr.Row():
-                    width = gr.Slider(
-                        label="Width",
-                        minimum=256,
-                        maximum=1536,
-                        step=64,
-                        value=512
-                    )
-                    height = gr.Slider(
-                        label="Height",
-                        minimum=256,
-                        maximum=1536,
-                        step=64,
-                        value=768
+
+with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange") as app:
+    gr.Markdown("# 🎭 Fashion AI Studio")
+    gr.Markdown("Generate fashion images and try on virtual clothing using AI")
+    
+    with gr.Tabs():
+        # Virtual Try-On 탭
+        with gr.TabItem("👔 Virtual Try-On"):
+            with gr.Row():
+                with gr.Column():
+                    prompt_input = gr.Textbox(
+                        label="Style Description", 
+                        placeholder="Describe the desired style (e.g., 'person wearing elegant dress')"
                     )
-            
-            with gr.Group(elem_classes="parameter-box"):
-                gr.Markdown("### 🎲 Seed Settings")
-                with gr.Row():
-                    randomize_seed = gr.Checkbox(
-                        True,
-                        label="Randomize seed"
+                    with gr.Row():
+                        with gr.Group():
+                            structure_image = gr.Image(
+                                label="Your Photo (Full-body)", 
+                                type="filepath"
+                            )
+                            gr.Markdown("*Upload a clear, well-lit full-body photo*")
+                            depth_strength = gr.Slider(
+                                minimum=0, 
+                                maximum=50, 
+                                value=15, 
+                                label="Fitting Strength"
+                            )
+                        with gr.Group():
+                            style_image = gr.Image(
+                                label="Clothing Item", 
+                                type="filepath"
+                            )
+                            gr.Markdown("*Upload the clothing item you want to try on*")
+                            style_strength = gr.Slider(
+                                minimum=0, 
+                                maximum=1, 
+                                value=0.5, 
+                                label="Style Transfer Strength"
+                            )
+                    tryon_btn = gr.Button("Generate Try-On")
+                    
+                    gr.Examples(
+                        examples=examples,
+                        inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
+                        outputs=[output_image],
+                        fn=generate_image,
+                        cache_examples=True
                     )
-                    seed = gr.Slider(
-                        label="Seed",
-                        minimum=0,
-                        maximum=MAX_SEED,
-                        step=1,
-                        value=42
+                
+                with gr.Column():
+                    output_image.render()
+
+        # Fashion Generation 탭
+        with gr.TabItem("👗 Fashion Generation"):
+            with gr.Column():
+                # 모드 선택
+                with gr.Group():
+                    mode = gr.Radio(
+                        choices=["Generate Model", "Generate Clothes"],
+                        label="Generation Mode",
+                        value="Generate Model"
                     )
+                
+                # 프롬프트 입력
+                prompt = gr.TextArea(
+                    label="✍️ Fashion Description (한글 또는 영어)",
+                    placeholder="패션 모델이나 의류를 설명하세요...",
+                    lines=5
+                )
+                
+                # 결과 이미지
+                result = gr.Image(label="Generated Fashion")
+                
+                generate_button = gr.Button("🚀 Generate Fashion")
+                
+                # 고급 설정 아코디언
+                with gr.Accordion("🎨 Advanced Options", open=False):
+                    with gr.Row():
+                        cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7.0)
+                        steps = gr.Slider(label="Steps", minimum=1, maximum=100, value=30)
+                        lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=1, value=0.85)
+                    
+                    with gr.Row():
+                        width = gr.Slider(label="Width", minimum=256, maximum=1536, value=512)
+                        height = gr.Slider(label="Height", minimum=256, maximum=1536, value=768)
+                    
+                    with gr.Row():
+                        randomize_seed = gr.Checkbox(True, label="Randomize seed")
+                        seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, value=42)
+
+                # 예제 탭
+                with gr.Tabs():
+                    with gr.TabItem("Model Examples"):
+                        gr.Examples(examples=example_model_prompts, inputs=prompt)
+                    with gr.TabItem("Clothes Examples"):
+                        gr.Examples(examples=example_clothes_prompts, inputs=prompt)
 
     # 이벤트 핸들러
+    tryon_btn.click(
+        fn=generate_image,
+        inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
+        outputs=[output_image]
+    )
+    
     generate_button.click(
         generate_fashion,
         inputs=[prompt, mode, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale],
         outputs=[result, seed]
     )
 
-app.queue()
-app.launch()
+if __name__ == "__main__":
+    app.launch(share=True)