Update app.py

app.py

@@ -49,24 +49,23 @@ def compare_images(image1, image2, blur_value, technique, threshold_value):
 def update_threshold_visibility(technique):
     return gr.update(visible=(technique == "Simple Binary"))
 
-demo = gr.Interface(
-    fn=compare_images,
-    inputs=[
-        gr.Image(type="numpy", label="Image Without Object"),
-        gr.Image(type="numpy", label="Image With Object"),
-        gr.Slider(minimum=1, maximum=15, step=2, value=5, label="Gaussian Blur"),
-        gr.Dropdown(["Adaptive Threshold", "Otsu's Threshold", "Simple Binary"], label="Thresholding Technique", value="Adaptive Threshold", interactive=True),
-        gr.Slider(minimum=0, maximum=255, step=1, value=50, label="Threshold Value", visible=False)
-    ],
-    outputs=[
-        gr.Image(type="numpy", label="Highlighted Differences"),
-        gr.Image(type="numpy", label="Raw Difference Overlay (Magenta)")
-    ],
-    title="Object Difference Highlighter",
-    description="Upload two images: one without an object and one with an object. The app will highlight only the newly added object and show the real differences in magenta overlayed on the original image.",
-    live=True
-)
-
-demo.load(update_threshold_visibility, inputs=["Thresholding Technique"], outputs=["Threshold Value"])
+with gr.Blocks() as demo:
+    gr.Markdown("# Object Difference Highlighter\nUpload two images: one without an object and one with an object. The app will highlight only the newly added object and show the real differences in magenta overlayed on the original image.")
+    
+    with gr.Row():
+        img1 = gr.Image(type="numpy", label="Image Without Object")
+        img2 = gr.Image(type="numpy", label="Image With Object")
+    
+    blur_slider = gr.Slider(minimum=1, maximum=15, step=2, value=5, label="Gaussian Blur")
+    technique_dropdown = gr.Dropdown(["Adaptive Threshold", "Otsu's Threshold", "Simple Binary"], label="Thresholding Technique", value="Adaptive Threshold", interactive=True)
+    threshold_slider = gr.Slider(minimum=0, maximum=255, step=1, value=50, label="Threshold Value", visible=False)
+    
+    technique_dropdown.change(update_threshold_visibility, inputs=[technique_dropdown], outputs=[threshold_slider])
+    
+    output1 = gr.Image(type="numpy", label="Highlighted Differences")
+    output2 = gr.Image(type="numpy", label="Raw Difference Overlay (Magenta)")
+    
+    btn = gr.Button("Process")
+    btn.click(compare_images, inputs=[img1, img2, blur_slider, technique_dropdown, threshold_slider], outputs=[output1, output2])
 
-demo.launch()
+demo.launch()