Update app.py

app.py

@@ -251,24 +251,37 @@ with gr.Blocks(title=title) as demo:
         outputs=[pred_out, pb_out, pv_out, masked_img_out, overlay_out]
     )
 
-    gr.Markdown("## Test Samples")
-
-    with gr.Row(scale=1):
 
-        # NORMAL
-        with gr.Column(scale=1):
-            gr.Markdown("### NORMAL")
-            gr.Image("images/NORMAL.jpeg", show_label=False, height=220)
+    test_samples = [
+        ("images/NORMAL.jpeg", "NORMAL"),
+        ("images/VIRAL.jpeg", "VIRAL"),
+        ("images/BACT.jpeg", "BACTERIAL"),
+    ]
+    
+    gr.Markdown("## Test Samples")
     
-        # VIRAL
-        with gr.Column(scale=1):
-            gr.Markdown("### VIRAL")
-            gr.Image("images/VIRAL.jpeg", show_label=False, height=220)
+    sample_buttons = []
     
-        # BACTERIAL
-        with gr.Column(scale=1):
-            gr.Markdown("### BACTERIAL")
-            gr.Image("images/BACT.jpeg", show_label=False, height=220)
+    with gr.Row():
+        for img_path, label in test_samples:
+            with gr.Column(scale=1):
+                gr.Markdown(f"### {label}")
+                btn = gr.Image(
+                    img_path,
+                    show_label=False,
+                    height=220,
+                    interactive=True
+                )
+                sample_buttons.append(btn)
+    
+    # Auto-run inference when any sample is clicked
+    for btn in sample_buttons:
+        btn.change(
+            inference_pipeline,
+            inputs=[btn, thresh_b, thresh_v, seg_thresh],
+            outputs=[pred_out, pb_out, pv_out, masked_img_out, overlay_out]
+        )
+
 
 if __name__ == "__main__":
     demo.launch()