Update app.py

app.py

@@ -1,19 +1,16 @@
 import cv2
 import numpy as np
-from matplotlib import pyplot as plt
+import gradio as gr
 
-# Load the two images provided by the user
-original_image_path = "/mnt/data/artwork done by us_page-0001.jpg"
-adjusted_image_path = "/mnt/data/drawing sent by party (1)_page-0001.jpg"
+def compare_images(original_image_path, adjusted_image_path):
+    # Read images using OpenCV
+    original_image = cv2.imread(original_image_path)
+    adjusted_image = cv2.imread(adjusted_image_path)
 
-# Read images using OpenCV
-original_image = cv2.imread(original_image_path)
-adjusted_image = cv2.imread(adjusted_image_path)
+    # Check if images loaded correctly
+    if original_image is None or adjusted_image is None:
+        return "Error: One or both images could not be loaded."
 
-# Check if images loaded correctly
-if original_image is None or adjusted_image is None:
-    comparison_result = "Error: One or both images could not be loaded."
-else:
     # Resize the adjusted image to match the dimensions of the original for accurate comparison
     original_image_resized = cv2.resize(original_image, (adjusted_image.shape[1], adjusted_image.shape[0]))
 
@@ -39,7 +36,7 @@ else:
     _, threshold_diff = cv2.threshold(difference, 30, 255, cv2.THRESH_BINARY)
     
     # Save the difference image
-    difference_image_path = "/mnt/data/difference_image.jpg"
+    difference_image_path = "difference_image.jpg"
     cv2.imwrite(difference_image_path, threshold_diff)
 
     # Prepare results summary
@@ -54,4 +51,37 @@ else:
         "Difference Image": difference_image_path
     }
 
-comparison_result
+    return comparison_result, threshold_diff
+
+# Gradio interface
+def gradio_compare_images(original_image, adjusted_image):
+    # Save images temporarily
+    original_image_path = "original_image.jpg"
+    adjusted_image_path = "adjusted_image.jpg"
+    
+    cv2.imwrite(original_image_path, original_image)
+    cv2.imwrite(adjusted_image_path, adjusted_image)
+    
+    # Compare images
+    result, diff_image = compare_images(original_image_path, adjusted_image_path)
+    
+    return result, diff_image
+
+# Create Gradio interface
+iface = gr.Interface(
+    fn=gradio_compare_images,
+    inputs=[
+        gr.inputs.Image(type="numpy", label="Original Image"),
+        gr.inputs.Image(type="numpy", label="Adjusted Image")
+    ],
+    outputs=[
+        gr.outputs.JSON(label="Comparison Result"),
+        gr.outputs.Image(type="numpy", label="Difference Image")
+    ],
+    title="Image Comparison Tool",
+    description="Upload two images to compare dimensions, brightness, and highlight differences."
+)
+
+# Launch the Gradio app
+iface.launch()
+