Update app.py

app.py

@@ -33,15 +33,19 @@ def compare_images(image1, image2, blur_value, technique, threshold_value):
     filtered_contours = [cnt for cnt in contours if cv2.contourArea(cnt) > 500]
     
     # Create a mask to isolate only the significant added object
-    mask = np.zeros_like(image1)
+    mask = np.zeros_like(image1, dtype=np.uint8)
     cv2.drawContours(mask, filtered_contours, -1, (255, 255, 255), thickness=cv2.FILLED)
     
     # Apply the mask to highlight the object added in the second image
     highlighted = cv2.bitwise_and(image2, mask)
     
-    # Overlay the difference in magenta over the first image
-    diff_colored = cv2.cvtColor(diff, cv2.COLOR_GRAY2BGR)
-    diff_colored[:, :, 1:] = 0  # Keep only red channel for magenta effect
+    # Create a magenta overlay where changes occurred
+    diff_colored = np.zeros_like(image1, dtype=np.uint8)
+    diff_colored[:, :, 0] = thresh  # Set blue channel to zero
+    diff_colored[:, :, 1] = 0  # Set green channel to zero
+    diff_colored[:, :, 2] = thresh  # Set red channel to highlight in magenta
+    
+    # Combine the original image with the magenta overlay
     overlayed = cv2.addWeighted(image1, 0.7, diff_colored, 0.3, 0)
     
     return highlighted, overlayed
@@ -68,4 +72,4 @@ with gr.Blocks() as demo:
     btn = gr.Button("Process")
     btn.click(compare_images, inputs=[img1, img2, blur_slider, technique_dropdown, threshold_slider], outputs=[output1, output2])
 
-demo.launch()
+demo.launch()