Update app.py

app.py

@@ -13,35 +13,54 @@ torch.backends.cuda.matmul.allow_tf32 = True
 # 번역 모델 초기화
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 
+# 기본 모델 및 LoRA 설정
 base_model = "black-forest-labs/FLUX.1-dev"
-pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
-
-lora_repo = "prithivMLmods/Canopus-Clothing-Flux-LoRA"
-trigger_word = ""
-pipe.load_lora_weights(lora_repo)
+model_lora_repo = "Motas/Flux_Fashion_Photography_Style"  # 패션 모델 LoRA
+clothes_lora_repo = "prithivMLmods/Canopus-Clothing-Flux-LoRA"  # 의류 LoRA
 
+pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
 pipe.to("cuda")
 
 MAX_SEED = 2**32-1
 
+# 예제 설정
+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"
+]
+
 @spaces.GPU()
-def translate_and_generate(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
+def generate_fashion(prompt, mode, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
     # 한글 감지 및 번역
     def contains_korean(text):
         return any(ord('가') <= ord(char) <= ord('힣') for char in text)
     
     if contains_korean(prompt):
-        # 한글을 영어로 번역
         translated = translator(prompt)[0]['translation_text']
         actual_prompt = translated
     else:
         actual_prompt = prompt
 
+    # 모드에 따른 LoRA 및 트리거워드 설정
+    if mode == "Generate Model":
+        pipe.load_lora_weights(model_lora_repo)
+        trigger_word = "fashion photography, professional model"
+    else:
+        pipe.load_lora_weights(clothes_lora_repo)
+        trigger_word = "upper clothing, fashion item"
+
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device="cuda").manual_seed(seed)
 
-    progress(0, "Starting image generation...")
+    progress(0, "Starting fashion generation...")
 
     for i in range(1, steps + 1):
         if i % (steps // 10) == 0:
@@ -60,59 +79,44 @@ def translate_and_generate(prompt, cfg_scale, steps, randomize_seed, seed, width
     progress(100, "Completed!")
     return image, seed
 
-example_image_path = "example0.webp"
-example_prompt = """Cozy winter scene with a Christmas atmosphere: a snow-covered cabin in the forest, warm light glowing from the windows, surrounded by sparkling Christmas decorations and a beautifully adorned Christmas tree. The sky is filled with stars, and soft snowflakes are gently falling, creating a serene and warm ambiance"""
-example_cfg_scale = 3.2
-example_steps = 32
-example_width = 1152
-example_height = 896
-example_seed = 3981632454
-example_lora_scale = 0.85
-
-def load_example():
-    example_image = Image.open(example_image_path)
-    return example_prompt, example_cfg_scale, example_steps, True, example_seed, example_width, example_height, example_lora_scale, example_image
-
-
 # CSS 정의
 custom_css = """
-/* 기본 스타일 */
 body {
-    margin: 0;
-    padding: 0;
-    background: url('file/example0.webp') no-repeat center center fixed;
-    background-size: cover;
-    min-height: 100vh;
+    background-color: #f5f5f5;
+    font-family: 'Arial', sans-serif;
 }
 
-/* 컨테이너 */
 .container {
     max-width: 1200px;
     margin: 0 auto;
     padding: 20px;
-    box-sizing: border-box;
 }
 
-/* 헤더 */
 .header {
     text-align: center;
-    color: white;
-    text-shadow: 2px 2px 4px rgba(0,0,0,0.7);
+    color: #333;
     margin-bottom: 30px;
     font-size: 2.5em;
+    text-transform: uppercase;
+    letter-spacing: 2px;
 }
 
-/* 박스 공통 스타일 */
 .box-common {
-    background-color: rgba(255, 255, 255, 0.85);
-    backdrop-filter: blur(10px);
+    background-color: white;
     border-radius: 15px;
-    padding: 20px;
+    padding: 25px;
     margin: 20px 0;
     box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
 }
 
-/* 결과 이미지 박스 */
+.mode-box {
+    background-color: #fff;
+    padding: 20px;
+    border-radius: 10px;
+    margin-bottom: 20px;
+    border: 2px solid #eee;
+}
+
 .result-box {
     width: 90%;
     max-width: 1000px;
@@ -126,162 +130,99 @@ body {
     display: block;
 }
 
-/* 프롬프트 입력 박스 */
 .prompt-box {
     width: 90%;
     max-width: 1000px;
     margin: 20px auto;
 }
 
-/* 버튼 스타일 */
 .generate-btn {
-    background-color: #2ecc71 !important;
+    background: linear-gradient(45deg, #ff6b6b, #ff8e8e) !important;
     color: white !important;
-    padding: 12px 30px !important;
+    padding: 15px 30px !important;
     border-radius: 8px !important;
     border: none !important;
     font-size: 1.1em !important;
     cursor: pointer !important;
-    transition: background-color 0.3s ease !important;
-    width: 200px !important;
+    transition: all 0.3s ease !important;
+    width: 250px !important;
     margin: 20px auto !important;
     display: block !important;
+    text-transform: uppercase !important;
+    letter-spacing: 1px !important;
 }
 
 .generate-btn:hover {
-    background-color: #27ae60 !important;
+    background: linear-gradient(45deg, #ff8e8e, #ff6b6b) !important;
+    transform: translateY(-2px) !important;
+    box-shadow: 0 5px 15px rgba(255, 107, 107, 0.4) !important;
 }
 
-/* 아코디언 스타일 */
 .accordion {
     width: 90%;
     max-width: 1000px;
     margin: 20px auto;
 }
 
-/* 오디오 컨트롤 */
-.audio-controls {
-    position: fixed;
-    bottom: 20px;
-    right: 20px;
-    z-index: 1000;
-    display: flex;
-    gap: 10px;
-    background-color: rgba(255, 255, 255, 0.9);
+.examples-table {
+    background-color: rgba(255, 255, 255, 0.95);
+    border-radius: 8px;
     padding: 15px;
-    border-radius: 10px;
-    box-shadow: 0 2px 10px rgba(0, 0, 0, 0.1);
-}
-
-.audio-btn {
-    background-color: #3498db;
-    color: white;
-    border: none;
-    padding: 8px 15px;
-    border-radius: 5px;
-    cursor: pointer;
-    transition: background-color 0.3s ease;
-}
-
-.audio-btn:hover {
-    background-color: #2980b9;
-}
-
-/* 눈 내리는 효과 */
-@keyframes snowfall {
-    0% {
-        transform: translateY(-10vh) translateX(0);
-        opacity: 1;
-    }
-    100% {
-        transform: translateY(100vh) translateX(100px);
-        opacity: 0.3;
-    }
+    margin-top: 20px;
 }
 
-.snowflake {
-    position: fixed;
-    color: white;
-    font-size: 1.5em;
-    user-select: none;
-    z-index: 1000;
-    pointer-events: none;
-    animation: snowfall linear infinite;
-}
-"""
-
-# JavaScript 코드
-snow_js = """
-function createSnowflake() {
-    const snowflake = document.createElement('div');
-    snowflake.innerHTML = '❄';
-    snowflake.className = 'snowflake';
-    snowflake.style.left = Math.random() * 100 + 'vw';
-    snowflake.style.animationDuration = Math.random() * 3 + 2 + 's';
-    snowflake.style.opacity = Math.random();
-    document.body.appendChild(snowflake);
-    
-    setTimeout(() => {
-        snowflake.remove();
-    }, 5000);
-}
-
-setInterval(createSnowflake, 200);
-"""
-
-audio_js = """
-let currentlyPlaying = null;
-
-function toggleAudio(num) {
-    const audio = document.getElementById('bgMusic' + num);
-    const otherAudio = document.getElementById('bgMusic' + (num === 1 ? 2 : 1));
-    
-    if (currentlyPlaying === audio) {
-        audio.pause();
-        currentlyPlaying = null;
-    } else {
-        if (currentlyPlaying) {
-            currentlyPlaying.pause();
-        }
-        otherAudio.pause();
-        audio.play();
-        currentlyPlaying = audio;
-    }
-}
-
-function stopAllAudio() {
-    const audios = document.querySelectorAll('audio');
-    audios.forEach(audio => audio.pause());
-    currentlyPlaying = null;
+.parameter-box {
+    background-color: #f8f9fa;
+    padding: 20px;
+    border-radius: 8px;
+    margin: 10px 0;
 }
 """
 
 # Gradio 인터페이스 구성
-app = gr.Blocks(css=custom_css,theme="Yntec/HaleyCH_Theme_Orange")
-
-with app:
-    # JavaScript 초기화
-    gr.HTML(f"<script>{snow_js}</script>")
-    
+with gr.Blocks(css=custom_css, theme="Yntec/HaleyCH_Theme_Orange") as app:
     with gr.Column(elem_classes="container"):
-        gr.Markdown("# 🎄 X-MAS LoRA", elem_classes="header")
+        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 Image")
+            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="✍️ Your Prompt (한글 또는 영어)",
-                placeholder="이미지를 설명하세요...",
+                label="✍️ Fashion Description (한글 또는 영어)",
+                placeholder="패션 모델이나 의류를 설명하세요...",
                 lines=5
             )
             generate_button = gr.Button(
-                "🚀 Generate Image",
+                "���� 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"):
@@ -294,21 +235,21 @@ with app:
                             minimum=1,
                             maximum=20,
                             step=0.5,
-                            value=example_cfg_scale
+                            value=7.0
                         )
                         steps = gr.Slider(
                             label="Steps",
                             minimum=1,
                             maximum=100,
                             step=1,
-                            value=example_steps
+                            value=30
                         )
                         lora_scale = gr.Slider(
                             label="LoRA Scale",
                             minimum=0,
                             maximum=1,
                             step=0.01,
-                            value=example_lora_scale
+                            value=0.85
                         )
             
             with gr.Group(elem_classes="parameter-box"):
@@ -319,14 +260,14 @@ with app:
                         minimum=256,
                         maximum=1536,
                         step=64,
-                        value=example_width
+                        value=512
                     )
                     height = gr.Slider(
                         label="Height",
                         minimum=256,
                         maximum=1536,
                         step=64,
-                        value=example_height
+                        value=768
                     )
             
             with gr.Group(elem_classes="parameter-box"):
@@ -341,35 +282,13 @@ with app:
                         minimum=0,
                         maximum=MAX_SEED,
                         step=1,
-                        value=example_seed
+                        value=42
                     )
 
-    # 오디오 컨트롤
-    gr.HTML(f"""
-        <div class="audio-controls">
-            <button class="audio-btn" onclick="toggleAudio(1)">🎵 Music 1</button>
-            <button class="audio-btn" onclick="toggleAudio(2)">🎵 Music 2</button>
-            <button class="audio-btn" onclick="stopAllAudio()">⏹ Stop</button>
-        </div>
-        <audio id="bgMusic1" loop>
-            <source src="file/1.mp3" type="audio/mp3">
-        </audio>
-        <audio id="bgMusic2" loop>
-            <source src="file/2.mp3" type="audio/mp3">
-        </audio>
-        <script>{audio_js}</script>
-    """)
-
     # 이벤트 핸들러
-    app.load(
-        load_example,
-        inputs=[],
-        outputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale, result]
-    )
-    
     generate_button.click(
-        translate_and_generate,
-        inputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale],
+        generate_fashion,
+        inputs=[prompt, mode, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale],
         outputs=[result, seed]
     )