Update app.py

app.py

@@ -1,4 +1,10 @@
 import gradio as gr
+import requests
+import smtplib
+from email.message import EmailMessage
+from PIL import Image
+import io
+
 import torch
 from diffusers import StableDiffusionPipeline
 
@@ -10,18 +16,82 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float
 pipe.to(device)
 pipe.enable_attention_slicing()  # Optimize memory usage
 
-def generate_ghibli_style(image):
-    prompt = "ghibli style portrait"
+BACKEND_URL = "https://your-hf-space-url.hf.space/ghibli"
+
+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.",
+    "Two Person Picture With Nepali Style Ghibli Art": "Create a Ghibli-style portrait using the uploaded photo as the person’s face. Background features Mount Everest, laliguras flowers, Kathmandu’s Ghanta Ghar tower, Newari window patterns, colorful Nepali prayer flags, the Nepali flag, Pashupatinath temple, and a flying Himalayan monal (Danfe). Make it vibrant, enchanting, and full of Nepali cultural details.",
+    "Spirited Away Fantasy": "Studio Ghibli style portrait with fantasy elements, mysterious background, dreamy lighting, spirited away aesthetic",
+    "Kiki's Cozy Town": "A warm, cozy background with bakery and rooftops, Studio Ghibli anime style, charming and colorful",
+    "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"
+}
+
+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):
+    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
 
-iface = gr.Interface(
-    fn=generate_ghibli_style,
-    inputs=gr.Image(type="pil"),
-    outputs=gr.Image(),
-    title="Studio Ghibli Portrait Generator",
-    description="Upload a photo to generate a Ghibli-style portrait!"
-)
+def send_email(email, image):
+    msg = EmailMessage()
+    msg["Subject"] = "Your Ghibli-style Portrait"
+    msg["From"] = "your@email.com"
+    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)
+    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.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=["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")
+            status = gr.Textbox(label="Status")
+
+    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"
+
+    send_button.click(fn=process, inputs=[image_input, style_choice, email_input], outputs=[image_output, status])
 
-iface.launch()
+demo.launch()