Update app.py

app.py

@@ -1,22 +1,14 @@
 import gradio as gr
-import requests
+import torch
+import io
 import smtplib
 from email.message import EmailMessage
-from PIL import Image
-import io
-
-import torch
-from diffusers import StableDiffusionPipeline
-
-device = "cuda" if torch.cuda.is_available() else "cpu"
-model_id = "nitrosocke/Ghibli-Diffusion"
+from diffusers import StableDiffusionImg2ImgPipeline
 
-# Load the model once and keep it in memory
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16 if device == "cuda" else torch.float32)
-pipe.to(device)
-pipe.enable_attention_slicing()  # Optimize memory usage
-
-BACKEND_URL = "https://your-hf-space-url.hf.space/ghibli"
+# Load SD3.5 pipeline
+pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-3.5-large", torch_dtype=torch.bfloat16
+).to("cuda")
 
 style_prompts = {
     "Single Photo With Dhaka Topi - Ghibli Style": "Create a Ghibli-style portrait using the uploaded photo as the person’s face to get a magical Ghibli-style portrait, wearing a traditional Nepali Dhaka topi. The dreamy background features Mount Everest, blooming laliguras flowers, Kathmandu’s Ghanta Ghar tower, elegant Newari window patterns, and colorful Nepali prayer flags waving in the wind. The art should blend Studio Ghibli’s enchanting style with authentic Nepali culture.",
@@ -26,72 +18,76 @@ style_prompts = {
     "Princess Mononoke Wild": "Forest warrior aesthetic, powerful gaze, nature spirit background, strong composition, Ghibli anime style",
     "Ghibli Vintage Portrait": "Classic Ghibli portrait, soft facial features, minimal background, vintage anime color palette, watercolor ink"
 }
+sample_keys = list(style_prompts.keys())[:5]
 
-def send_to_backend(image, prompt):
-    buffered = io.BytesIO()
-    image.save(buffered, format="PNG")
-    response = requests.post(
-        BACKEND_URL,
-        files={"file": ("upload.png", buffered.getvalue(), "image/png")},
-        data={"prompt": prompt}
-    )
-    return Image.open(io.BytesIO(response.content))
-
-def generate_image(image, prompt):
-    return f"Generated Ghibli style image"
+def generate_ghibli_style(image, prompt, strength=0.6):
+    with torch.inference_mode():
+        out = pipe(prompt=prompt,
+                   image=image,
+                   strength=strength,
+                   guidance_scale=6.5,
+                   num_inference_steps=25).images[0]
+    return out
 
-def generate_ghibli_style(image, prompt):
-    print(f"Prompt: {prompt}")
-    with torch.inference_mode():  # Disables gradient calculations for faster inference
-        result = pipe(prompt, image=image, strength=0.6, guidance_scale=6.5, num_inference_steps=25).images[0]  # Reduced steps & optimized scale
-    return result
+def process(image, style, email):
+    prompt = style_prompts.get(style, "")
+    result = generate_ghibli_style(image, prompt)
+    email_status = ""
+    if email:
+        email_status = send_email(email, result)
+    return result, "✅ Generated", email_status
 
 def send_email(email, image):
     msg = EmailMessage()
     msg["Subject"] = "Your Ghibli-style Portrait"
     msg["From"] = "[email protected]"
     msg["To"] = email
-    buf = io.BytesIO()
-    image.save(buf, format="PNG")
-    buf.seek(0)
-    msg.add_attachment(buf.read(), maintype='image', subtype='png', filename='portrait.png')
-
-    with smtplib.SMTP_SSL("smtp.example.com", 465) as server:
-        server.login("[email protected]", "yourpassword")
-        server.send_message(msg)
+    buf = io.BytesIO(); image.save(buf, format="PNG"); buf.seek(0)
+    msg.add_attachment(buf.read(), maintype="image", subtype="png", filename="portrait.png")
+    with smtplib.SMTP_SSL("smtp.example.com", 465) as s:
+        s.login("your@email.com", "password")
+        s.send_message(msg)
     return "✅ Sent to email!"
 
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple")) as demo:
-    gr.Markdown(
-        """
-        # Nepali Roots Ghibli Art Booth
-        Welcome to the magical world of Studio Ghibli!  
-        Take a photo or upload one, select your art style, and transform into a dreamy anime character ✨
-        """
-    )
+with gr.Blocks(
+    theme=gr.themes.Soft(primary_hue="emerald", secondary_hue="amber", neutral_hue="stone"),
+    css="""
+        body { font-family: 'Merriweather', serif; background: #f9f5f0; }
+        h1 { color: #2a6f48; font-family: 'Karma', serif; }
+        .sample-btn { margin: 4px; background: #a4c639; color: white; border-radius:8px; }
+        .sample-btn:hover { background: #7fa32c; }
+    """
+) as demo:
+    gr.Markdown("## 🌸 Nepali Roots Ghibli Art Booth")
 
     with gr.Row():
-        with gr.Column():
-            style_choice = gr.Dropdown(
-                choices=list(style_prompts.keys()),
-                value="Totoro Forest Dream",
-                label="🎨 Choose a Ghibli Style"
-            )
-            # image_input = gr.Image(sources=["webcam"], type="pil", label="📸 Take or Upload Image")
-            image_input = gr.Image(sources=["webcam","upload"], type="pil", label="📸 Take or Upload Image")
-            email_input = gr.Textbox(label="📧 Enter your email to receive your portrait")
-            send_button = gr.Button("🎆 Generate")
-        with gr.Column():
-            image_output = gr.Image(label="🌠 Ghibli-Styled Result")
+        with gr.Column(scale=1):
+            gr.Markdown("### Choose a Ghibli prompt style:")
+            sample_buttons = []
+            for key in sample_keys:
+                btn = gr.Button(key, elem_classes="sample-btn")
+                sample_buttons.append(btn)
+            style_choice = gr.Dropdown(list(style_prompts.keys()), value=sample_keys[0], label="🎨 Style")
+            image_input = gr.Image(label="📸 Upload or take photo", type="pil", source="upload")
+            email_input = gr.Textbox(label="📧 Email (optional)")
+            generate_button = gr.Button("✨ Generate & Project")
+
+        with gr.Column(scale=1):
+            image_output = gr.Image(label="🌠 Result", show_fullscreen_button=True)
             status = gr.Textbox(label="Status")
+            email_status = gr.Textbox(label="Email Status")
+            projector = gr.HTML("", elem_id="projector")
+
+    # sample button callback
+    for btn, key in zip(sample_buttons, sample_keys):
+        btn.click(lambda k=key: gr.update(value=k), inputs=None, outputs=style_choice)
 
-    def process(image, style, email):
-        prompt = style_prompts.get(style, list(style_prompts.values())[0])
-        # result = send_to_backend(image, prompt)
-        result = generate_ghibli_style(image, prompt)
-        # send_email(email, result)
-        return result, "✅ Portrait generated"
+    generate_button.click(process, [image_input, style_choice, email_input], [image_output, status, email_status])
 
-    send_button.click(fn=process, inputs=[image_input, style_choice, email_input], outputs=[image_output, status])
+    # projector link update
+    def projector_link(img):
+        if img is None: return ""
+        return '<a href="#" onclick="window.open(window.images[0].src,\'_blank\')">🔍 Open full-screen on projector</a>'
+    image_output.change(projector_link, inputs=[image_output], outputs=[projector])
 
 demo.launch()