Update app.py

app.py

@@ -10,6 +10,8 @@ import os
 from PIL import Image
 import spaces
 import numpy as np
+import cv2
+from pyzxing import BarCodeReader
 
 from diffusers import (
     StableDiffusionPipeline,
@@ -73,19 +75,39 @@ def scan_qr_code(image):
         image = Image.fromarray(image)
     
     # Convert to grayscale
-    image = image.convert('L')
+    gray_image = image.convert('L')
     
-    # Increase contrast
-    image = Image.eval(image, lambda x: 255 if x > 128 else 0)
+    # Convert to numpy array
+    np_image = np.array(gray_image)
     
-    # Try to decode the QR code
+    # Method 1: Using qrcode library
     try:
         qr = qrcode.QRCode()
         qr.add_data('')
-        qr.decode(image)
+        qr.decode(gray_image)
         return qr.data.decode('utf-8')
-    except Exception as e:
-        return None
+    except Exception:
+        pass
+    
+    # Method 2: Using OpenCV
+    try:
+        qr_detector = cv2.QRCodeDetector()
+        retval, decoded_info, points, straight_qrcode = qr_detector.detectAndDecodeMulti(np_image)
+        if retval:
+            return decoded_info[0]
+    except Exception:
+        pass
+    
+    # Method 3: Fallback to zxing-cpp
+    try:
+        reader = BarCodeReader()
+        results = reader.decode(np_image)
+        if results:
+            return results[0].parsed
+    except Exception:
+        pass
+    
+    return None
 
 @spaces.GPU()
 def inference(
@@ -238,7 +260,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as blocks:
             
                     - **High settings (1.6-5.0)**: If you need to make sure that the QR code is super easy to scan, even if it means the image looks less like art and more like a regular QR code, then choose a higher value. This is ideal when functionality is the main goal, and the artistic side can take a backseat.
                     
-                    Start with **1.3** if you’re unsure, and adjust up or down depending on whether you want the QR code to be more artistic or more functional.
+                    Start with **1.3** if you're unsure, and adjust up or down depending on whether you want the QR code to be more artistic or more functional.
                     """
                 )
             
@@ -251,13 +273,13 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as blocks:
                 )
                 gr.Markdown(
                     """
-                    **Artistic Freedom** controls how much the AI is allowed to change the QR code's look to match your description. It’s like telling the AI how creative it can get with your QR code:
+                    **Artistic Freedom** controls how much the AI is allowed to change the QR code's look to match your description. It's like telling the AI how creative it can get with your QR code:
             
                     - **Low settings (0.0-0.3)**: If you set this low, the AI will make small changes and your QR code will look more like a regular, plain QR code. This is useful if you want something that is still creative but not too wild, keeping it simple and easy to scan.
             
                     - **Medium settings (0.4-0.7)**: Here, the AI will add more artistic touches but keep the QR code recognizable. You get the best of both worlds—your QR code will have some creative flair, but it will still be easy to scan. For most cases, setting it to **0.6** is a great way to keep the code functional and artistic.
             
-                    - **High settings (0.8-1.0)**: If you set this high, the AI will go all-out creative. The QR code will look amazing, but it might be difficult to scan because the art can start to take over the code. This setting is perfect if you’re aiming for a highly creative piece of art and don’t mind if it’s a bit harder to scan. Start at **0.9** to explore creative but functional designs.
+                    - **High settings (0.8-1.0)**: If you set this high, the AI will go all-out creative. The QR code will look amazing, but it might be difficult to scan because the art can start to take over the code. This setting is perfect if you're aiming for a highly creative piece of art and don't mind if it's a bit harder to scan. Start at **0.9** to explore creative but functional designs.
                     """
                 )
             
@@ -276,7 +298,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as blocks:
             
                     - **Medium settings (5-15)**: This is a good balance where the AI will mostly follow your prompt but will also add some of its own creative touches. If you want to see some surprises but still want the design to look like what you described, start at around **7.5**.
             
-                    - **High settings (15+)**: If you choose a higher value, the AI will stick very closely to what you wrote in the description. This is good if you have a very specific idea and don’t want the AI to change much. Just keep in mind that this might limit the AI’s creativity.
+                    - **High settings (15+)**: If you choose a higher value, the AI will stick very closely to what you wrote in the description. This is good if you have a very specific idea and don't want the AI to change much. Just keep in mind that this might limit the AI's creativity.
                     
                     Start at **7.5** for a balanced approach where the AI follows your ideas but still adds some artistic flair.
                     """
@@ -291,13 +313,13 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as blocks:
                     """
                     **Art Style** changes how the AI creates the image, using different methods (or "samplers"). Each method has a different effect on how detailed or artistic the final QR code looks:
             
-                    - **DPM++ Karras SDE**: This is a great all-around option for creating high-quality, detailed images. It’s a good place to start if you want a balance of sharpness and creativity.
+                    - **DPM++ Karras SDE**: This is a great all-around option for creating high-quality, detailed images. It's a good place to start if you want a balance of sharpness and creativity.
             
                     - **Euler**: This method creates very sharp, detailed images, making the QR code look crisp and clear. Choose this if you want a precise, well-defined design.
             
-                    - **DDIM**: This method is better if you want the QR code to have a more artistic, abstract style. It’s great for when you want the QR code to look like a piece of modern art.
+                    - **DDIM**: This method is better if you want the QR code to have a more artistic, abstract style. It's great for when you want the QR code to look like a piece of modern art.
                     
-                    Feel free to experiment with different samplers to see what works best for the look you’re going for!
+                    Feel free to experiment with different samplers to see what works best for the look you're going for!
                     """
                 )