Update app.py

app.py

@@ -337,54 +337,59 @@ examples = [
 
 output_image = gr.Image(label="Virtual Try-On Result")
 
-with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange") as app:
-    gr.Markdown("# Style Generator")
-    gr.Markdown("Upload your photo and try on different clothing items virtually using AI. The system will generate an image of you wearing the selected clothing while maintaining your pose and appearance.")
+# Gradio 인터페이스 생성
+demo = gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange")
+
+with demo:
+    gr.Markdown("# 🎭 StyleGen : Fashion AI Studio")
+    gr.Markdown("Generate fashion images and try on virtual clothing using AI")
     
-    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.Tabs():
+        # Virtual Try-On 탭
+        with gr.TabItem("👔 Virtual Try-On"):
             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.Column():
+                    prompt_input = gr.Textbox(
+                        label="Style Description", 
+                        placeholder="Describe the desired style (e.g., 'person wearing elegant dress')"
                     )
-                with gr.Group():
-                    style_image = gr.Image(
-                        label="Clothing Item", 
-                        type="filepath"
+                    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=False
                     )
-                    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"
-                    )
-            generate_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,
-                cache_mode="lazy"
-            )
-        
-        with gr.Column():
-            output_image.render()
+                
+                with gr.Column():
+                    output_image = gr.Image(label="Virtual Try-On Result")
     
     
     generate_btn.click(