Create app.py

app.py

@@ -1,107 +1,36 @@
-import gradio as gr
-from PIL import Image
 import torch
-from transformers import BlipProcessor, BlipForConditionalGeneration
 from diffusers import StableDiffusionPipeline
-import traceback
-
-# 모델 로딩 (최대한 빠른 처리: BLIP + SD-Turbo)
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
-blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
-blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base").to(device)
+import gradio as gr
 
+# 모델 로드 (로컬에 없을 경우 허깅페이스에서 자동 다운로드됨)
 pipe = StableDiffusionPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-turbo",
-    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
-).to(device)
-pipe.safety_checker = None  # 안전필터 off (실습/학습 목적)
-
-def caricature_from_photo(image):
-    try:
-        raw_image = image.convert('RGB')
-        inputs = blip_processor(raw_image, return_tensors="pt").to(device)
-        out = blip_model.generate(**inputs, max_new_tokens=30)
-        description = blip_processor.decode(out[0], skip_special_tokens=True)
-
-        # 캐리커쳐 스타일 프롬프트 (이모티콘과 직관적 키워드로 꾸밈)
-        caricature_prompt = (
-            f"{description}, cute cartoon, gentle lines, charming and lovely, sweet expression, lovely mood, digital illustration, "
-            "cheerful, simple background, joy, cartoon-style, 😄✨"
-        )
-
-        # turbo SD: 매우 빠른 생성!
-        result = pipe(
-            caricature_prompt, 
-            num_inference_steps=4,         # turbo 권장값: 4
-            guidance_scale=0.0              # turbo 권장값: 0.0 (텍스트 프롬프트 반영)
-        ).images[0]
-
-        info_text = (
-            f"**이미지 설명**: {description}\n\n"
-            f"**캐리커쳐 프롬프트**: `{caricature_prompt}`"
-        )
-        return result, info_text
-    except Exception as e:
-        err = f"❗️ 오류가 발생했어요!\n\n{e}\n\n{traceback.format_exc()}"
-        empty = Image.new("RGB", (512,512), "#FFFDE7")
-        return empty, err
-
-custom_css = """
-body { background: linear-gradient(135deg, #a7ffeb 0%, #fce4ec 100%);}
-.gradio-container { background-color: #fafcff !important;}
-.gr-button { background: #ffd600; color: #252525; border-radius:18px; font-size:1.2em;}
-.gr-button:hover { background:#ffd54f;}
-#caricature-preview img {
-    border-radius: 30px; border: 4px solid #ffff8d;
-    box-shadow: 0px 8px 56px #ffd60044;
-    margin-top: 1em;
-}
-#main-view .block {
-    background: #ffffffb8; border-radius: 22px;
-    box-shadow: 0 4px 32px #b3afec33; margin:18px;
-}
-"""
-
-with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
-    gr.Markdown("""
-    <div style="text-align:center; padding-top:.5em;">
-      <span style="font-size:2.3em;">🎨 <b style="color:#1ec69d">캐리커쳐 매직</b> <span style="font-size:1em;">앱</span> ✨</span>
-      <p style="font-size:1.11em; color:#575d76;">
-        어떤 사진이든 <b>1초 만에 캐리커쳐</b>로!<br>
-        <span style="color:#2196f3">카메라 📷</span> 또는 <span style="color:#ff1696">앨범</span>에서 이미지를 올려보세요!
-      </p>
-    </div>
-    """)
-    with gr.Row(elem_id="main-view"):
-        with gr.Column():
-            photo = gr.Image(
-                label="📷 사진 찍기 또는 선택(앨범)",
-                type="pil",
-                height=300,
-                elem_id="user-photo",
-            )
-            btn = gr.Button("👉 캐리커쳐 만들기!", elem_id="do_caricature")
-        with gr.Column():
-            caricature_img = gr.Image(
-                label="🖼️ 캐리커쳐 미리보기",
-                elem_id="caricature-preview",
-                height=350,
-                show_label=True
-            )
-            info = gr.Markdown(label="생성 내역 및 설명", value="", elem_id="summary-field")
-    btn.click(caricature_from_photo, inputs=photo, outputs=[caricature_img, info])
-
-    # 안내 메시지
-    gr.Markdown("""
-    <div style="background:#fffde7; border-radius:18px; border:1.5px dashed #ffe082; padding:1em; text-align:center; margin-top:18px;">
-      <b>💡 사용법 & 인쇄 안내</b><br/>
-      <ul style="text-align:left; margin-left:1.8em;">
-        <li>휴대폰은 [사진찍기] 선택 시, 카메라와 앨범 둘 다 사용 가능</li>
-        <li>[👉 캐리커쳐 만들기!] 버튼을 누르면 캐리커쳐가 빠르게 생성됩니다.</li>
-        <li>결과 이미지를 길게 누르거나 우클릭하여 저장/공유/인쇄 할 수 있습니다.</li>
-      </ul>
-    </div>
-    """)
-
-demo.launch()
+    "runwayml/stable-diffusion-v1-5",
+    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+    revision="fp16" if torch.cuda.is_available() else None,
+    use_safetensors=True
+)
+
+# GPU 사용 가능하면 이동
+if torch.cuda.is_available():
+    pipe = pipe.to("cuda")
+else:
+    pipe = pipe.to("cpu")
+
+# 생성 함수
+def generate_image(prompt):
+    result = pipe(prompt, guidance_scale=7.5)
+    image = result.images[0]
+    return image
+
+# Gradio 인터페이스
+demo = gr.Interface(
+    fn=generate_image,
+    inputs=gr.Textbox(label="Input Text", placeholder="예: A fantasy landscape with flying whales"),
+    outputs=gr.Image(label="Generated Image"),
+    title="🖼️ Text to Image Generator (Stable Diffusion)",
+    description="AI creates a Image if you give the prompt. 모델: runwayml/stable-diffusion-v1-5"
+)
+
+# 실행
+if __name__ == "__main__":
+    demo.launch()