Update app.py

app.py

@@ -213,30 +213,32 @@ def inference(
                 back_color=qr_back_color,
             )
             qrcode_image = resize_for_condition_image(qrcode_image, 768)
+        else:
+            print("Using QR Code Image")
+            qrcode_image = resize_for_condition_image(qrcode_image, 768)
 
         # Generate base64 string of the initial QR code
         initial_qr_base64 = image_to_base64(qrcode_image)
 
-        # Use QR code as init image if specified
-        if use_qr_code_as_init_image:
-            init_image = qrcode_image
+        # hack due to gradio examples
+        init_image = qrcode_image
 
         out = pipe(
             prompt=prompt,
             negative_prompt=negative_prompt,
-            image=init_image,
-            control_image=qrcode_image,
-            width=768,
-            height=768,
+            image=qrcode_image,
+            control_image=qrcode_image,  # type: ignore
+            width=768,  # type: ignore
+            height=768,  # type: ignore
             guidance_scale=float(guidance_scale),
-            controlnet_conditioning_scale=float(controlnet_conditioning_scale),
+            controlnet_conditioning_scale=float(controlnet_conditioning_scale),  # type: ignore
             generator=generator,
             strength=float(strength),
             num_inference_steps=50,
         )
         
         # Verify the generated QR code
-        verify_result = scan_qr_code(out.images[0].convert('RGB'), sensitivity=verify_sensitivity)
+        verify_result = scan_qr_code(out.images[0], sensitivity=verify_sensitivity)
         
         return out.images[0], verify_result, initial_qr_base64
     except Exception as e:
@@ -446,14 +448,6 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as blocks:
                 """
             )
 
-        with gr.Accordion("QR Code Customization", open=True):
-            qr_version = gr.Slider(minimum=1, maximum=40, step=1, value=1, label="QR Code Version")
-            qr_error_correction = gr.Dropdown(choices=["L", "M", "Q", "H"], value="H", label="Error Correction Level")
-            qr_box_size = gr.Slider(minimum=1, maximum=50, step=1, value=10, label="Box Size")
-            qr_border = gr.Slider(minimum=0, maximum=10, step=1, value=4, label="Border Size")
-            qr_fill_color = gr.ColorPicker(label="QR Code Color", value="#000000")
-            qr_back_color = gr.ColorPicker(label="Background Color", value="#FFFFFF")
-
     def scan_and_display(image):
         if image is None:
             return "No image to scan"
@@ -491,12 +485,6 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as blocks:
             use_qr_code_as_init_image,
             sampler,
             verify_sensitivity,
-            qr_version,
-            qr_error_correction,
-            qr_box_size,
-            qr_border,
-            qr_fill_color,
-            qr_back_color,
         ],
         outputs=[result_image, scan_result, initial_qr_image],
         concurrency_limit=20
@@ -504,6 +492,5 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as blocks:
 
     download_initial_qr.click(lambda x: x, inputs=[initial_qr_image], outputs=[initial_qr_image])
 
-
 blocks.queue(max_size=20,api_open=False)
 blocks.launch(share=bool(os.environ.get("SHARE", False)), show_api=True)