Update app.py

app.py

@@ -9,6 +9,7 @@ import io
 import os
 from PIL import Image
 import spaces
+import numpy as np
 
 from diffusers import (
     StableDiffusionPipeline,
@@ -30,12 +31,14 @@ qrcode_generator = qrcode.QRCode(
 )
 
 controlnet = ControlNetModel.from_pretrained(
-    "DionTimmer/controlnet_qrcode-control_v1p_sd15", torch_dtype=torch.float16
+    "monster-labs/control_v1p_sd15_qrcode_monster",
+    #"DionTimmer/controlnet_qrcode-control_v1p_sd15", 
+    torch_dtype=torch.float16
 ).to("cuda")
 
 pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
-    "digiplay/GhostMixV1.2VAE",
-    #"benjamin-paine/stable-diffusion-v1-5",
+    #"digiplay/GhostMixV1.2VAE",
+    "benjamin-paine/stable-diffusion-v1-5",
     controlnet = controlnet,
     torch_dtype = torch.float16,
     safety_checker = None,
@@ -64,6 +67,26 @@ SAMPLER_MAP = {
     "DEIS": lambda config: DEISMultistepScheduler.from_config(config),
 }
 
+def scan_qr_code(image):
+    # Convert gradio image to PIL Image if necessary
+    if isinstance(image, np.ndarray):
+        image = Image.fromarray(image)
+    
+    # Convert to grayscale
+    image = image.convert('L')
+    
+    # Increase contrast
+    image = Image.eval(image, lambda x: 255 if x > 128 else 0)
+    
+    # Try to decode the QR code
+    try:
+        qr = qrcode.QRCode()
+        qr.add_data('')
+        qr.decode(image)
+        return qr.data.decode('utf-8')
+    except Exception as e:
+        return None
+
 @spaces.GPU()
 def inference(
     qr_code_content: str,
@@ -313,6 +336,24 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as blocks:
                 Enjoy the process of bringing your unique vision to life!
                 """
             )
+
+    def scan_and_display(image):
+        if image is None:
+            return "No image to scan"
+        
+        scanned_text = scan_qr_code(image)
+        if scanned_text:
+            return f"Scanned successfully: {scanned_text}"
+        else:
+            return "Failed to scan QR code. Try adjusting the settings for better visibility."
+
+    scan_button.click(
+        scan_and_display,
+        inputs=[result_image],
+        outputs=[scan_result]
+    )
+
+            
     run_btn.click(
         inference,
         inputs=[