Update app.py

app.py

@@ -111,7 +111,9 @@ def scan_qr_code(image):
     return None
 
 def invert_image(image):
-    return ImageOps.invert(image)
+    if image is None:
+        return None
+    return ImageOps.invert(image.convert('RGB'))
 
 @spaces.GPU()
 def inference(
@@ -382,7 +384,9 @@ with gr.Blocks(theme='Hev832/Applio') as blocks:
                 
         with gr.Column():
             result_image = gr.Image(label="Your Artistic QR Code")
-            scan_button = gr.Button("Scan QR Code")
+            with gr.Row():
+                scan_button = gr.Button("Scan QR Code")
+                invert_button = gr.Button("Invert Image")
             scan_result = gr.Textbox(label="Scan Result", interactive=False)
             
             gr.Markdown(
@@ -408,13 +412,23 @@ with gr.Blocks(theme='Hev832/Applio') as blocks:
         else:
             return "Failed to scan QR code. Try adjusting the settings for better visibility."
 
+    def invert_displayed_image(image):
+        if image is None:
+            return None
+        return invert_image(image)
+
     scan_button.click(
         scan_and_display,
         inputs=[result_image],
         outputs=[scan_result]
     )
 
-            
+    invert_button.click(
+        invert_displayed_image,
+        inputs=[result_image],
+        outputs=[result_image]
+    )
+
     run_btn.click(
         inference,
         inputs=[