Update app.py

app.py

@@ -109,6 +109,7 @@ def scan_qr_code(image):
     
     return None
 
+
 @spaces.GPU()
 def inference(
     qr_code_content: str,
@@ -151,23 +152,34 @@ def inference(
             print("Using QR Code Image")
             qrcode_image = resize_for_condition_image(qrcode_image, 768)
 
-        # hack due to gradio examples
-        init_image = qrcode_image
+        # Determine which image to use as init_image and control_image
+        if use_qr_code_as_init_image:
+            init_image = qrcode_image
+            control_image = qrcode_image
+        else:
+            control_image = qrcode_image
+            if init_image is None:
+                # If no init_image provided, set strength to 1.0 to generate a new image
+                strength = 1.0
+
+        # Adjust strength if using an init_image
+        if init_image is not None:
+            strength = min(strength, 0.8)  # Cap strength at 0.8 when using init_image
 
         out = pipe(
             prompt=prompt,
             negative_prompt=negative_prompt,
-            image=qrcode_image,
-            control_image=qrcode_image,  # type: ignore
-            width=768,  # type: ignore
-            height=768,  # type: ignore
+            image=init_image,
+            control_image=control_image,
+            width=768,
+            height=768,
             guidance_scale=float(guidance_scale),
-            controlnet_conditioning_scale=float(controlnet_conditioning_scale),  # type: ignore
+            controlnet_conditioning_scale=float(controlnet_conditioning_scale),
             generator=generator,
             strength=float(strength),
             num_inference_steps=50,
         )
-        return out.images[0]  # type: ignore
+        return out.images[0]
     except Exception as e:
         print(f"Error in inference: {str(e)}")
         # Return a blank image in case of an error
@@ -229,7 +241,7 @@ with gr.Blocks(theme='Hev832/Applio') as blocks:
             use_qr_code_as_init_image = gr.Checkbox(
                 label="Use QR code as init image", 
                 value=True, 
-                interactive=False, 
+                interactive=True,
                 info="Whether init image should be QR code. Unclick to pass init image or generate init image with Stable Diffusion 2.1"
             )
             with gr.Accordion(label="Init Images (Optional)", open=False, visible=True) as init_image_acc: