Create app.py

app.py

@@ -0,0 +1,48 @@
+import gradio as gr
+from PIL import Image
+import torch
+from diffusers import StableDiffusionPipeline
+
+# cartoon 캐리커쳐 특화 모델(대표 예, 바꿔도 됨)
+SD_MODEL_ID = "Fictiverse/Stable-Diffusion-Comic"
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+
+pipe = StableDiffusionPipeline.from_pretrained(
+    SD_MODEL_ID,
+    torch_dtype=torch.float16 if DEVICE=="cuda" else torch.float32
+).to(DEVICE)
+pipe.safety_checker = None
+
+def to_cartoon(image):
+    prompt = "caricature, cartoon, highly detailed, cute, colorful, big head, digital illustration, clean background, outline"
+    # img2img 방식: strength(변화 세기) 조절
+    cartoon = pipe(prompt=prompt, image=image, strength=0.7, num_inference_steps=20).images[0]
+    return cartoon
+
+with gr.Blocks(css="""
+body { background: linear-gradient(135deg, #fceabb 0%, #f8b500 100%);}
+.gradio-container { background-color: #fffde7 !important;}
+.gr-button { background: #a7ffeb; border-radius: 18px; }
+#preview-img img { border-radius: 28px; border: 3px solid #ffc107; box-shadow:0 8px 32px #ffa00044;}
+""") as app:
+    gr.Markdown("""
+    <div style="text-align:center;font-size:2em;margin-bottom:1em;">🎨 <b>카메라 캐리커쳐 앱</b> 🖨️</div>
+    """)
+    with gr.Row():
+        # 카메라/앨범 업로드 모두 허용
+        inp = gr.Image(sources=["upload", "camera"], label="사진을 찍거나 선택하세요", type="pil", height=270)
+        btn = gr.Button("👉 캐리커쳐 만들기!", elem_id="make-btn", variant="primary")
+        out = gr.Image(label="캐리커쳐 미리보기", elem_id="preview-img", height=320)
+    btn.click(to_cartoon, inputs=inp, outputs=out)
+
+    gr.Markdown("""
+    <div style="text-align:center;margin-top:1em; background:#fffbea; border-radius:16px;line-height:1.7em;">
+    <b>💡 사용법 & 인쇄 안내</b><br>
+    <ul style="text-align:left;margin-left:2em;">
+        <li>카메라 촬영 또는 파일 선택 후 [👉 캐리커쳐 만들기!] 클릭</li>
+        <li>미리보기 이미지를 길게 누르거나 우클릭해 저장 또는 인쇄(Ctrl+P)!</li>
+    </ul>
+    </div>
+    """)
+
+app.launch()