Update app.py

app.py

@@ -23,6 +23,19 @@ pipe.to("cuda")
 
 MAX_SEED = 2**32-1
 
+# 예시 프롬프트 정의
+model_examples = [
+    "professional fashion model wearing elegant black dress in studio lighting",
+    "fashion model in casual street wear, urban background",
+    "high fashion model in avant-garde outfit on runway"
+]
+
+clothes_examples = [
+    "luxurious red evening gown with detailed embroidery",
+    "casual denim jacket with vintage wash",
+    "modern minimalist white blazer with clean lines"
+]
+
 @spaces.GPU()
 def generate_fashion(prompt, mode, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
     # 한글 감지 및 번역
@@ -36,7 +49,7 @@ def generate_fashion(prompt, mode, cfg_scale, steps, randomize_seed, seed, width
         actual_prompt = prompt
 
     # 모드에 따른 LoRA 및 트리거워드 설정
-    if mode == "Generate Model":
+    if mode == "패션 모델 생성":
         pipe.load_lora_weights(model_lora_repo)
         trigger_word = "fashion photography, professional model"
     else:
@@ -66,96 +79,64 @@ def generate_fashion(prompt, mode, cfg_scale, steps, randomize_seed, seed, width
     progress(100, "Completed!")
     return image, seed
 
-def generate_image(prompt, structure_image, style_image, depth_strength, style_strength):
-    # 실제 이미지 생성 로직을 여기에 구현
-    return Image.new('RGB', (512, 512), 'white')
+def update_examples(mode):
+    if mode == "패션 모델 생성":
+        return gr.Examples(examples=model_examples)
+    else:
+        return gr.Examples(examples=clothes_examples)
 
 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")
+    gr.Markdown("AI를 사용하여 패션 모델과 의류를 생성해보세요")
     
-    with gr.Tabs():
-        # Virtual Try-On 탭
-        with gr.TabItem("👔 Virtual Try-On"):
+    with gr.Column():
+        # 모드 선택
+        mode = gr.Radio(
+            choices=["패션 모델 생성", "패션 의류 생성"],
+            label="생성 모드",
+            value="패션 모델 생성"
+        )
+        
+        # 프롬프트 입력
+        prompt = gr.TextArea(
+            label="✍️ 패션 설명 (한글 또는 영어)",
+            placeholder="패션 모델이나 의류를 설명하세요...",
+            lines=3
+        )
+        
+        # 예시 섹션
+        with gr.Column() as example_container:
+            examples = gr.Examples(
+                examples=model_examples,
+                inputs=prompt,
+                label="예시 프롬프트"
+            )
+        
+        # 결과 이미지
+        result = gr.Image(label="생성된 이미지")
+        
+        generate_button = gr.Button("🚀 이미지 생성")
+        
+        # 고급 설정
+        with gr.Accordion("🎨 고급 설정", 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():
-                with gr.Column():
-                    prompt_input = gr.Textbox(
-                        label="Style Description", 
-                        placeholder="Describe the desired style (e.g., 'person wearing elegant dress')"
-                    )
-                    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")
-                
-                with gr.Column():
-                    output_image = gr.Image(label="Generated Result")
-
-        # 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)
+                randomize_seed = gr.Checkbox(True, label="시드 랜덤화")
+                seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, value=42)
 
     # 이벤트 핸들러
-    tryon_btn.click(
-        fn=generate_image,
-        inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
-        outputs=[output_image]
+    mode.change(
+        fn=update_examples,
+        inputs=[mode],
+        outputs=[examples]
     )
     
     generate_button.click(