Update app.py

app.py

@@ -1,51 +1,60 @@
 import gradio as gr
 import qrcode
-from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
-from diffusers.utils import load_image
-import torch
 from PIL import Image
-import io
+import torch
+from diffusers import StableDiffusionControlNetImg2ImgPipeline, ControlNetModel
 
-# Load ControlNet model khusus QR
+# Load ControlNet QR model (publik)
 controlnet = ControlNetModel.from_pretrained(
-    "monster-labs/controlnet-qr-code", torch_dtype=torch.float16
+    "DionTimmer/controlnet_qrcode-control_v1p_sd15",
+    torch_dtype=torch.float16
 )
-pipe = StableDiffusionControlNetPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
+
+pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
+    "runwayml/stable-diffusion-v1-5",
+    controlnet=controlnet,
+    safety_checker=None,
+    torch_dtype=torch.float16
 ).to("cuda")
 
-def generate_qr_with_logo(text, image):
-    # 1. Buat QR code dasar
+def generate_qr_art(text, base_image):
+    # Generate QR dasar
     qr = qrcode.QRCode(error_correction=qrcode.constants.ERROR_CORRECT_H)
     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("RGB").resize((512, 512))
 
-    # 2. Resize agar sama dengan gambar input
-    if image is not None:
-        image = image.resize((512, 512))
-    qr_img = qr_img.resize((512, 512))
+    # Gunakan gambar dasar jika ada
+    if base_image:
+        base = base_image.resize((512, 512))
+    else:
+        base = qr_img
 
-    # 3. Gabungkan: ControlNet akan memandu agar QR mengikuti gambar dasar
-    prompt = "artistic QR code blended with uploaded image, still scannable"
+    # Generate artistic QR
     result = pipe(
-        prompt,
-        image=qr_img,
+        prompt="artistic QR code blended with uploaded image, still scannable",
+        image=base,
         control_image=qr_img,
-        num_inference_steps=30,
-        guidance_scale=9.0
+        width=512,
+        height=512,
+        guidance_scale=7.5,
+        num_inference_steps=40,
+        controlnet_conditioning_scale=1.2
     ).images[0]
 
     return result
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Artistic QR Code Generator (seperti Starbucks QR)")
+    gr.Markdown("## 🎨 Artistic QR Code Generator")
+    gr.Markdown("Masukkan teks/URL lalu upload logo atau gambar dasar → hasilnya QR artistik seperti contoh Starbucks.")
+
     with gr.Row():
-        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")
+        text_input = gr.Textbox(label="Teks atau URL", placeholder="contoh: https://example.com")
+        img_input = gr.Image(label="Upload Gambar Dasar (opsional)", type="pil")
+
+    btn = gr.Button("🚀 Generate QR")
+    output = gr.Image(label="Hasil QR Artistik")
 
-    btn.click(fn=generate_qr_with_logo, inputs=[text, img], outputs=output)
+    btn.click(generate_qr_art, inputs=[text_input, img_input], outputs=output)
 
 demo.launch()