feat: 애니메이션 모델 지원하는 Stable Diffusion WebUI 추가

.dockerignore

@@ -0,0 +1,24 @@
+# Git
+.git
+.gitattributes
+
+# Python
+__pycache__
+*.pyc
+*.pyo
+*.pyd
+.Python
+env
+venv
+.venv
+
+# IDE
+.vscode
+.idea
+*.swp
+*.swo
+
+# 기타
+*.log
+.DS_Store
+Thumbs.db

Dockerfile

@@ -0,0 +1,40 @@
+# 허깅페이스 스페이스용 스테이블 디퓨전 WebUI Dockerfile
+FROM python:3.10-slim
+
+# 시스템 의존성 설치
+RUN apt-get update && apt-get install -y \
+    git \
+    wget \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    libgomp1 \
+    && rm -rf /var/lib/apt/lists/*
+
+# 작업 디렉토리 설정
+WORKDIR /app
+
+# 비root 사용자 생성 (허깅페이스 요구사항)
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+
+# 작업 디렉토리 설정
+WORKDIR $HOME/app
+
+# requirements 복사 및 설치
+COPY --chown=user requirements.txt .
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt
+
+# 애플리케이션 파일 복사
+COPY --chown=user . .
+
+# 허깅페이스 스페이스는 포트 7860 사용
+EXPOSE 7860
+
+# 애플리케이션 실행
+CMD ["python", "app.py"]

README.md

@@ -1,12 +1,63 @@
 ---
 title: Stable Diffusion WebUI
-emoji: 👁
+emoji: 🌸
 colorFrom: pink
-colorTo: yellow
-sdk: gradio
-sdk_version: 6.3.0
-app_file: app.py
+colorTo: purple
+sdk: docker
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# 🌸 Anime Diffusion WebUI
+
+허깅페이스 스페이스에서 실행되는 애니메이션 스타일 이미지 생성기입니다.
+
+## ✨ 기능
+
+- **텍스트-이미지 생성**: 프롬프트를 입력하여 이미지 생성
+- **다양한 애니메이션 모델**: 7가지 인기 모델 지원
+- **네거티브 프롬프트**: 원하지 않는 요소 제외
+- **이미지 크기 조절**: 256x256 ~ 768x768
+- **스텝 수 조절**: 생성 품질 조절
+- **가이던스 스케일**: 프롬프트 준수 정도 조절
+- **시드 설정**: 재현 가능한 결과
+
+## 🤖 지원 모델
+
+| 모델 | 특징 |
+|------|------|
+| **Anything V5** | 가장 인기 있는 애니메이션 모델 |
+| **Counterfeit V3** | 고품질 애니메이션/일러스트 |
+| **DreamShaper V8** | 다양한 스타일 지원 |
+| **OpenJourney V4** | Midjourney 스타일 |
+| **Stable Diffusion v1.5** | 기본 모델 |
+| **MeinaMix V11** | 고퀄리티 애니메이션 |
+| **ReV Animated** | 애니메이션 특화 |
+
+## 🚀 사용법
+
+1. 드롭다운에서 원하는 모델을 선택합니다
+2. 프롬프트 입력창에 원하는 이미지를 설명합니다
+3. (선택) 네거티브 프롬프트로 원하지 않는 요소를 지정합니다
+4. 이미지 크기와 생성 옵션을 조절합니다
+5. "🚀 이미지 생성" 버튼을 클릭합니다
+
+## 💡 프롬프트 예시
+
+```
+1girl, solo, long blue hair, blue eyes, school uniform, 
+cherry blossoms, masterpiece, best quality, detailed
+```
+
+## ⚠️ 주의사항
+
+- CPU 모드에서는 이미지 생성에 2-5분 정도 소요됩니다
+- 첫 실행 시 모델 다운로드로 추가 시간이 필요합니다
+- GPU 스페이스를 사용하면 훨씬 빠른 생성이 가능합니다
+
+## 🛠️ 기술 스택
+
+- **Diffusers** - 허깅페이스 디퓨전 라이브러리
+- **Gradio** - 웹 인터페이스
+- **Docker** - 컨테이너 환경
+- **PyTorch** - 딥러닝 프레임워크
+

app.py

@@ -0,0 +1,388 @@
+"""
+스테이블 디퓨전 WebUI - 허깅페이스 스페이스용
+Gradio 인터페이스를 통한 이미지 생성 (애니메이션 모델 지원)
+"""
+
+import gradio as gr
+import torch
+from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
+from PIL import Image
+import os
+import gc
+
+# 사용 가능한 모델 목록
+MODELS = {
+    "🎨 Anything V5 (애니메이션)": "stablediffusionapi/anything-v5",
+    "🌸 Counterfeit V3 (고품질 애니메이션)": "gsdf/Counterfeit-V3.0",
+    "✨ DreamShaper V8 (다목적)": "Lykon/DreamShaper",
+    "🎭 OpenJourney (Midjourney 스타일)": "prompthero/openjourney-v4",
+    "🖼️ Stable Diffusion v1.5 (기본)": "runwayml/stable-diffusion-v1-5",
+    "🌟 MeinaMix (애니메이션)": "Meina/MeinaMix_V11",
+    "💫 ReV Animated (애니메이션)": "stablediffusionapi/rev-animated",
+}
+
+# 디바이스 설정
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+DTYPE = torch.float16 if torch.cuda.is_available() else torch.float32
+
+print(f"🚀 디바이스: {DEVICE}, 데이터 타입: {DTYPE}")
+
+# 현재 로드된 모델 정보
+current_model_id = None
+pipe = None
+
+def clear_memory():
+    """메모리 정리"""
+    global pipe
+    if pipe is not None:
+        del pipe
+        pipe = None
+    gc.collect()
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
+
+def load_model(model_name: str):
+    """모델 로드 함수"""
+    global pipe, current_model_id
+    
+    model_id = MODELS.get(model_name)
+    if model_id is None:
+        return None, f"❌ 알 수 없는 모델: {model_name}"
+    
+    # 이미 같은 모델이 로드되어 있으면 스킵
+    if current_model_id == model_id and pipe is not None:
+        return pipe, f"✅ {model_name} 이미 로드됨"
+    
+    # 기존 모델 정리
+    clear_memory()
+    
+    print(f"📥 모델 로딩 중: {model_name}...")
+    
+    try:
+        pipe = StableDiffusionPipeline.from_pretrained(
+            model_id,
+            torch_dtype=DTYPE,
+            safety_checker=None,
+            requires_safety_checker=False,
+            use_safetensors=True
+        )
+        
+        # 빠른 스케줄러 사용
+        pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+        
+        # 디바이스로 이동
+        pipe = pipe.to(DEVICE)
+        
+        # 메모리 최적화
+        pipe.enable_attention_slicing()
+        if hasattr(pipe, 'enable_vae_slicing'):
+            pipe.enable_vae_slicing()
+        
+        current_model_id = model_id
+        print(f"✅ 모델 로딩 완료: {model_name}")
+        return pipe, f"✅ {model_name} 로딩 완료!"
+    
+    except Exception as e:
+        print(f"❌ 모델 로딩 실패: {e}")
+        return None, f"❌ 모델 로딩 실패: {str(e)}"
+
+def generate_image(
+    model_name: str,
+    prompt: str,
+    negative_prompt: str = "",
+    num_inference_steps: int = 25,
+    guidance_scale: float = 7.5,
+    width: int = 512,
+    height: int = 512,
+    seed: int = -1,
+    progress=gr.Progress()
+):
+    """
+    이미지 생성 함수
+    
+    Args:
+        model_name: 사용할 모델 이름
+        prompt: 이미지 생성 프롬프트
+        negative_prompt: 네거티브 프롬프트
+        num_inference_steps: 추론 스텝 수
+        guidance_scale: 가이던스 스케일
+        width: 이미지 너비
+        height: 이미지 높이
+        seed: 시드 값 (-1이면 랜덤)
+    
+    Returns:
+        생성된 이미지, 상태 메시지
+    """
+    
+    global pipe
+    
+    if not prompt.strip():
+        return None, "⚠️ 프롬프트를 입력해주세요!"
+    
+    # 모델 로드
+    progress(0.1, desc="모델 로딩 중...")
+    pipe, status = load_model(model_name)
+    if pipe is None:
+        return None, status
+    
+    # 애니메이션 모델용 기본 네거티브 프롬프트
+    default_negative = "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry"
+    
+    if negative_prompt.strip():
+        full_negative = f"{negative_prompt}, {default_negative}"
+    else:
+        full_negative = default_negative
+    
+    # 시드 설정
+    if seed == -1:
+        seed = torch.randint(0, 2**32 - 1, (1,)).item()
+    
+    generator = torch.Generator(device=DEVICE).manual_seed(int(seed))
+    
+    try:
+        progress(0.3, desc="이미지 생성 중...")
+        print(f"🎨 이미지 생성 중... 프롬프트: {prompt[:50]}...")
+        
+        # 이미지 생성
+        result = pipe(
+            prompt=prompt,
+            negative_prompt=full_negative,
+            num_inference_steps=num_inference_steps,
+            guidance_scale=guidance_scale,
+            width=width,
+            height=height,
+            generator=generator
+        )
+        
+        image = result.images[0]
+        progress(1.0, desc="완료!")
+        print("✅ 이미지 생성 완료!")
+        
+        return image, f"✅ 생성 완료! (시드: {seed})"
+    
+    except Exception as e:
+        print(f"❌ 이미지 생성 실패: {e}")
+        return None, f"❌ 이미지 생성 실패: {str(e)}"
+
+# Gradio 인터페이스 생성
+def create_interface():
+    """Gradio 웹 인터페이스 생성"""
+    
+    # 커스텀 CSS
+    custom_css = """
+    .gradio-container {
+        font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+        max-width: 1200px !important;
+    }
+    
+    .generate-btn {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
+        border: none !important;
+        color: white !important;
+        font-weight: bold !important;
+        font-size: 1.2em !important;
+        padding: 15px 30px !important;
+        border-radius: 10px !important;
+        transition: all 0.3s ease !important;
+        width: 100% !important;
+    }
+    
+    .generate-btn:hover {
+        transform: translateY(-2px) !important;
+        box-shadow: 0 6px 20px rgba(102, 126, 234, 0.5) !important;
+    }
+    
+    .title {
+        text-align: center;
+        background: linear-gradient(135deg, #ff6b9d 0%, #c44569 50%, #764ba2 100%);
+        -webkit-background-clip: text;
+        -webkit-text-fill-color: transparent;
+        font-size: 2.8em;
+        font-weight: bold;
+        margin-bottom: 5px;
+    }
+    
+    .subtitle {
+        text-align: center;
+        color: #888;
+        font-size: 1.1em;
+        margin-bottom: 25px;
+    }
+    
+    .model-dropdown {
+        border: 2px solid #764ba2 !important;
+        border-radius: 8px !important;
+    }
+    
+    .output-image {
+        border-radius: 12px !important;
+        box-shadow: 0 4px 15px rgba(0,0,0,0.1) !important;
+    }
+    
+    .status-box {
+        background: linear-gradient(135deg, #f5f7fa 0%, #e4e8eb 100%);
+        border-radius: 8px;
+        padding: 10px;
+        text-align: center;
+    }
+    """
+    
+    with gr.Blocks(css=custom_css, title="Stable Diffusion WebUI - Anime") as demo:
+        # 헤더
+        gr.HTML("""
+            <div class="title">🌸 Anime Diffusion WebUI</div>
+            <div class="subtitle">애니메이션 스타일 이미지 생성기 | AI Art Generator</div>
+        """)
+        
+        with gr.Row():
+            # 왼쪽: 입력 패널
+            with gr.Column(scale=1):
+                # 모델 선택
+                model_dropdown = gr.Dropdown(
+                    label="🤖 모델 선택 (Model)",
+                    choices=list(MODELS.keys()),
+                    value="🎨 Anything V5 (애니메이션)",
+                    elem_classes=["model-dropdown"]
+                )
+                
+                prompt = gr.Textbox(
+                    label="📝 프롬프트 (Prompt)",
+                    placeholder="1girl, solo, long hair, blue eyes, school uniform, cherry blossoms, detailed, masterpiece, best quality",
+                    lines=3
+                )
+                
+                negative_prompt = gr.Textbox(
+                    label="🚫 네거티브 프롬프트 (Negative Prompt)",
+                    placeholder="추가할 네거티브 프롬프트 (기본값이 자동 적용됩니다)",
+                    lines=2
+                )
+                
+                with gr.Row():
+                    width = gr.Slider(
+                        label="📐 너비",
+                        minimum=256,
+                        maximum=768,
+                        value=512,
+                        step=64
+                    )
+                    height = gr.Slider(
+                        label="📐 높이",
+                        minimum=256,
+                        maximum=768,
+                        value=768,
+                        step=64
+                    )
+                
+                with gr.Row():
+                    num_steps = gr.Slider(
+                        label="🔄 스텝 수",
+                        minimum=10,
+                        maximum=50,
+                        value=25,
+                        step=1
+                    )
+                    guidance = gr.Slider(
+                        label="🎯 CFG 스케일",
+                        minimum=1.0,
+                        maximum=15.0,
+                        value=7.0,
+                        step=0.5
+                    )
+                
+                seed = gr.Number(
+                    label="🎲 시드 (-1 = 랜덤)",
+                    value=-1,
+                    precision=0
+                )
+                
+                generate_btn = gr.Button(
+                    "🚀 이미지 생성",
+                    elem_classes=["generate-btn"]
+                )
+                
+                status_text = gr.Textbox(
+                    label="📊 상태",
+                    value="모델을 선택하고 프롬프트를 입력해주세요",
+                    interactive=False,
+                    elem_classes=["status-box"]
+                )
+            
+            # 오른쪽: 출력 패널
+            with gr.Column(scale=1):
+                output_image = gr.Image(
+                    label="🖼️ 생성된 이미지",
+                    type="pil",
+                    elem_classes=["output-image"]
+                )
+        
+        # 예제 - 애니메이션 스타일
+        gr.Examples(
+            examples=[
+                ["🎨 Anything V5 (애니메이션)", "1girl, solo, long blue hair, heterochromia, school uniform, cherry blossoms, sitting, detailed background, masterpiece, best quality", ""],
+                ["🎨 Anything V5 (애니메이션)", "1boy, solo, white hair, red eyes, dark cape, sword, dark fantasy, dramatic lighting, masterpiece", ""],
+                ["🌸 Counterfeit V3 (고품질 애니메이션)", "1girl, kimono, japanese garden, autumn leaves, serene expression, ultra detailed, 8k", ""],
+                ["✨ DreamShaper V8 (다목적)", "beautiful anime girl, studio ghibli style, forest spirit, magical atmosphere, soft lighting", ""],
+                ["🎭 OpenJourney (Midjourney 스타일)", "mdjrny-v4 style, cute robot character, pastel colors, kawaii, digital art", ""],
+                ["🌟 MeinaMix (애니메이션)", "1girl, maid outfit, silver hair, purple eyes, cafe interior, warm lighting, masterpiece", ""],
+            ],
+            inputs=[model_dropdown, prompt, negative_prompt],
+            label="💡 예제 프롬프트 (클릭하여 적용)"
+        )
+        
+        # 프롬프트 가이드
+        with gr.Accordion("📖 프롬프트 작성 가이드", open=False):
+            gr.Markdown("""
+            ### 🎨 애니메이션 스타일 프롬프트 팁
+            
+            **품질 태그** (권장):
+            - `masterpiece, best quality, high quality` - 고품질 출력
+            - `detailed, ultra detailed` - 디테일 강조
+            - `8k, highres` - 고해상도 느낌
+            
+            **캐릭터 묘사**:
+            - `1girl, 1boy, solo` - 캐릭터 수
+            - `long hair, short hair, blue eyes` - 외모
+            - `school uniform, dress, armor` - 복장
+            
+            **배경/분위기**:
+            - `cherry blossoms, night sky, fantasy` - 배경
+            - `dramatic lighting, soft lighting` - 조명
+            - `floating, dynamic pose` - 포즈
+            
+            **스타일**:
+            - `anime style, manga style` - 일반 애니메이션
+            - `studio ghibli style` - 지브리 스타일
+            - `chibi, cute` - 귀여운 스타일
+            """)
+        
+        # 이벤트 핸들러
+        generate_btn.click(
+            fn=generate_image,
+            inputs=[model_dropdown, prompt, negative_prompt, num_steps, guidance, width, height, seed],
+            outputs=[output_image, status_text]
+        )
+        
+        # 푸터
+        gr.HTML("""
+            <div style="text-align: center; margin-top: 30px; padding: 20px; color: #888; border-top: 1px solid #eee;">
+                <p>🌸 Powered by Diffusers & Gradio | 🤗 Hugging Face Spaces</p>
+                <p style="font-size: 0.9em;">⚠️ CPU 모드에서는 이미지 생성에 2-5분 정도 소요됩니다.</p>
+                <p style="font-size: 0.85em; color: #aaa;">첫 실행 시 모델 다운로드로 인해 시간이 더 걸릴 수 있습니다.</p>
+            </div>
+        """)
+    
+    return demo
+
+# 메인 실행
+if __name__ == "__main__":
+    print("🌸 Anime Diffusion WebUI 시작...")
+    
+    # Gradio 앱 생성 및 실행
+    demo = create_interface()
+    
+    # 허깅페이스 스페이스에서는 포트 7860 사용
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )

requirements.txt

@@ -0,0 +1,19 @@
+# 스테이블 디퓨전 핵심 라이브러리
+diffusers>=0.25.0
+transformers>=4.36.0
+accelerate>=0.25.0
+safetensors>=0.4.0
+
+# 이미지 처리
+Pillow>=10.0.0
+numpy>=1.24.0
+
+# 웹 인터페이스
+gradio>=4.15.0
+
+# PyTorch (CPU 버전 - 허깅페이스 무료 스페이스용)
+torch>=2.1.0
+torchvision>=0.16.0
+
+# 추가 유틸리티
+invisible-watermark>=0.2.0