Update app.py

app.py

@@ -1,46 +1,51 @@
 import gradio as gr
 import qrcode
+from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
+from diffusers.utils import load_image
+import torch
 from PIL import Image
-
-def generate_qr(data, logo_file, bg_file):
-    # Buat QR dasar
+import io
+
+# Load ControlNet model khusus QR
+controlnet = ControlNetModel.from_pretrained(
+    "monster-labs/controlnet-qr-code", torch_dtype=torch.float16
+)
+pipe = StableDiffusionControlNetPipeline.from_pretrained(
+    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
+).to("cuda")
+
+def generate_qr_with_logo(text, image):
+    # 1. Buat QR code dasar
     qr = qrcode.QRCode(error_correction=qrcode.constants.ERROR_CORRECT_H)
-    qr.add_data(data)
+    qr.add_data(text)
     qr.make(fit=True)
+    qr_img = qr.make_image(fill_color="black", back_color="white").convert("RGB")
 
-    qr_img = qr.make_image(fill_color="black", back_color="white").convert("RGBA")
-
-    # Tambahkan background jika ada
-    if bg_file is not None:
-        bg = Image.open(bg_file).convert("RGBA")
-        bg = bg.resize(qr_img.size)
-        qr_img = Image.alpha_composite(bg, qr_img)
+    # 2. Resize agar sama dengan gambar input
+    if image is not None:
+        image = image.resize((512, 512))
+    qr_img = qr_img.resize((512, 512))
 
-    # Tambahkan logo jika ada
-    if logo_file is not None:
-        logo = Image.open(logo_file).convert("RGBA")
-        # Skala logo agar tidak terlalu besar (misalnya 20% dari lebar QR)
-        logo_size = int(qr_img.size[0] * 0.2)
-        logo = logo.resize((logo_size, logo_size))
-        pos = ((qr_img.size[0] - logo_size) // 2, (qr_img.size[1] - logo_size) // 2)
-        qr_img.paste(logo, pos, mask=logo)
+    # 3. Gabungkan: ControlNet akan memandu agar QR mengikuti gambar dasar
+    prompt = "artistic QR code blended with uploaded image, still scannable"
+    result = pipe(
+        prompt,
+        image=qr_img,
+        control_image=qr_img,
+        num_inference_steps=30,
+        guidance_scale=9.0
+    ).images[0]
 
-    return qr_img
+    return result
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Custom QR Code Generator with Logo & Background")
-
+    gr.Markdown("## Artistic QR Code Generator (seperti Starbucks QR)")
     with gr.Row():
-        data = gr.Textbox(label="Link / Text untuk QR", placeholder="https://contoh.com")
-    
-    with gr.Row():
-        logo_file = gr.File(label="Upload Logo (PNG)" , type="filepath")
-        bg_file = gr.File(label="Upload Background (PNG/JPG)", type="filepath")
-
-    generate_btn = gr.Button("Generate QR")
-    output_img = gr.Image(label="Hasil QR Code")
+        text = gr.Textbox(label="Teks/Link untuk QR code")
+        img = gr.Image(label="Upload gambar dasar (logo/foto)", type="pil")
+    btn = gr.Button("Generate")
+    output = gr.Image(label="QR Code Hasil")
 
-    generate_btn.click(fn=generate_qr, inputs=[data, logo_file, bg_file], outputs=output_img)
+    btn.click(fn=generate_qr_with_logo, inputs=[text, img], outputs=output)
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()