Update app.py

app.py

@@ -70,7 +70,7 @@ SAMPLER_MAP = {
     "DEIS": lambda config: DEISMultistepScheduler.from_config(config),
 }
 
-def scan_qr_code(image):
+def scan_qr_code(image, sensitivity=0.5):
     # Convert gradio image to PIL Image if necessary
     if isinstance(image, np.ndarray):
         image = Image.fromarray(image)
@@ -82,7 +82,7 @@ def scan_qr_code(image):
     np_image = np.array(gray_image)
     
     # Preprocess the image
-    _, binary_image = cv2.threshold(np_image, 128, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU)
+    _, binary_image = cv2.threshold(np_image, int(255 * sensitivity), 255, cv2.THRESH_BINARY)
     
     # List to store successful scans
     successful_scans = []
@@ -92,7 +92,7 @@ def scan_qr_code(image):
         qr = qrcode.QRCode()
         qr.add_data('')
         qr.decode(gray_image)
-        successful_scans.append(qr.data.decode('utf-8'))
+        successful_scans.append(("qrcode library", qr.data.decode('utf-8')))
     except Exception as e:
         print(f"qrcode library failed: {str(e)}")
     
@@ -101,7 +101,7 @@ def scan_qr_code(image):
         qr_detector = cv2.QRCodeDetector()
         retval, decoded_info, points, straight_qrcode = qr_detector.detectAndDecodeMulti(np_image)
         if retval:
-            successful_scans.extend(decoded_info)
+            successful_scans.extend([("OpenCV", info) for info in decoded_info if info])
     except Exception as e:
         print(f"OpenCV failed: {str(e)}")
     
@@ -110,13 +110,13 @@ def scan_qr_code(image):
         reader = BarCodeReader()
         results = reader.decode(np_image)
         if results:
-            successful_scans.extend([result.parsed for result in results])
+            successful_scans.extend([("zxing-cpp", result.parsed) for result in results])
         
         # If no results, try with the binary image
         if not results:
             results = reader.decode(binary_image)
             if results:
-                successful_scans.extend([result.parsed for result in results])
+                successful_scans.extend([("zxing-cpp (binary)", result.parsed) for result in results])
     except Exception as e:
         print(f"zxing-cpp failed: {str(e)}")
     
@@ -128,12 +128,12 @@ def scan_qr_code(image):
                 reader = BarCodeReader()
                 results = reader.decode(temp_file.name)
                 if results:
-                    successful_scans.extend([result.parsed for result in results])
+                    successful_scans.extend([("zxing-cpp (file)", result.parsed) for result in results])
             os.unlink(temp_file.name)
         except Exception as e:
             print(f"Temporary file scanning failed: {str(e)}")
     
-    return successful_scans[0] if successful_scans else None
+    return successful_scans
 
 @spaces.GPU()
 def inference(
@@ -148,6 +148,7 @@ def inference(
     qrcode_image: Image.Image | None = None,
     use_qr_code_as_init_image = True,
     sampler = "DPM++ Karras SDE",
+    verify_sensitivity: float = 0.5,
 ):
     try:
         if prompt is None or prompt == "":
@@ -193,11 +194,15 @@ def inference(
             strength=float(strength),
             num_inference_steps=50,
         )
-        return out.images[0]  # type: ignore
+        
+        # Verify the generated QR code
+        verify_result = scan_qr_code(out.images[0], sensitivity=verify_sensitivity)
+        
+        return out.images[0], verify_result
     except Exception as e:
         print(f"Error in inference: {str(e)}")
         # Return a blank image in case of an error
-        return Image.new('RGB', (768, 768), color='white')
+        return Image.new('RGB', (768, 768), color='white'), []
 
 
 
@@ -205,35 +210,25 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as blocks:
     gr.Markdown(
         """
         ![Yamamoto Logo](https://cdn-uploads.huggingface.co/production/uploads/64740cf7485a7c8e1bd51ac9/_VyYxp5qE_nRZ_LJqBxmL.webp)
-
         # 🎨 Yamamoto QR Code Art Generator
-
         ## Transform Your QR Codes into Brand Masterpieces
-
         Welcome to Yamamoto's innovative QR Code Art Generator! This cutting-edge tool empowers our creative team to craft 
         visually stunning, on-brand QR codes that perfectly blend functionality with artistic expression.
-
         ## 🚀 How It Works:
-
         1. **Enter Your QR Code Content**: Start by inputting the URL or text for your QR code.
         2. **Craft Your Prompt**: Describe the artistic style or theme you envision for your QR code.
         3. **Fine-tune with Advanced Settings**: Adjust parameters to perfect your creation (see tips below).
         4. **Generate and Iterate**: Click 'Run' to create your art, then refine as needed.
-
         ## 🌟 Tips for Spectacular Results:
-
         - **Artistic Freedom**: Set between 0.8 and 0.95 for a balance of creativity and scannability.
         - **QR Code Visibility**: Aim for 0.6 to 2.0 to ensure your code is both artistic and functional.
         - **Prompt Crafting**: Use vivid, specific descriptions that align with your brand identity.
         - **Experimentation**: Don't hesitate to try different settings and prompts to find your perfect style!
-
         ## 🎭 Prompt Ideas to Spark Your Creativity:
-
         - "A serene Japanese garden with cherry blossoms and a koi pond"
         - "A futuristic cityscape with neon lights and flying cars"
         - "An abstract painting with swirling colors and geometric shapes"
         - "A vintage-style travel poster featuring iconic landmarks"
-
         Remember, the magic lies in the details of your prompt and the fine-tuning of your settings. 
         Happy creating!
         """
@@ -373,6 +368,22 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as blocks:
                     Try **-1** if you want to explore and generate different designs. If you find something you really love, write down the seed number and use it again to recreate the same design.
                     """
                 )
+            
+                verify_sensitivity = gr.Slider(
+                    minimum=0.1,
+                    maximum=1.0,
+                    step=0.05,
+                    value=0.5,
+                    visible=True
+                    label="Verify Sensitivity",
+                )
+                gr.Markdown(
+                    """
+                    **Verify Sensitivity** controls how sensitive the QR code scanner is when verifying the generated image. 
+                    A lower value (closer to 0.1) makes the scanner more sensitive and able to detect less clear QR codes, 
+                    while a higher value (closer to 1.0) requires clearer, more distinct QR codes to be detected.
+                    """
+                )
             with gr.Row():
                 run_btn = gr.Button("🎨 Create Your QR Art", variant="primary")
                 
@@ -428,10 +439,11 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as blocks:
             qr_code_image,
             use_qr_code_as_init_image,
             sampler,
+            verify_sensitivity,
         ],
-        outputs=[result_image],
+        outputs=[result_image, scan_result],
         concurrency_limit=20
     )
 
 blocks.queue(max_size=20,api_open=False)
-blocks.launch(share=bool(os.environ.get("SHARE", False)), show_api=True)
+blocks.launch(share=bool(os.environ.get("SHARE", False)), show_api=True)