Update app.py

app.py

@@ -6,6 +6,7 @@ from collections import Counter
 from ultralytics import YOLO
 import plotly.express as px
 import plotly.graph_objects as go
+import os
 
 # Load the model
 model = YOLO("best.pt")
@@ -91,6 +92,13 @@ def process_image(image, conf_threshold=0.25):
             None
         )
 
+def load_example_image():
+    """Load the example test image"""
+    example_image = cv2.imread("test.png")
+    if example_image is None:
+        return None
+    return example_image
+
 # Create Gradio interface with improved layout
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     gr.Markdown("""
@@ -101,6 +109,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     with gr.Row():
         with gr.Column(scale=1):
             input_image = gr.Image(type="numpy", label="Upload Image")
+            with gr.Row():
+                example_btn = gr.Button("Try Example Image")
+                analyze_btn = gr.Button("Analyze Image", variant="primary")
             conf_slider = gr.Slider(
                 minimum=0.1,
                 maximum=1.0,
@@ -109,7 +120,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
                 label="Confidence Threshold",
                 info="Adjust detection sensitivity"
             )
-            analyze_btn = gr.Button("Analyze Image", variant="primary")
         
         with gr.Column(scale=1):
             output_image = gr.Image(type="numpy", label="Detected Cells")
@@ -128,7 +138,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         with gr.Column(scale=1):
             density_plot = gr.Plot(label="Cell Density Heatmap")
     
-    # Handle button click
+    # Handle button clicks
+    example_btn.click(
+        load_example_image,
+        outputs=input_image
+    )
+    
     analyze_btn.click(
         process_image,
         inputs=[input_image, conf_slider],
@@ -137,7 +152,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     
     gr.Markdown("""
     ### Instructions:
-    1. Upload a microscopy image containing cells
+    1. Upload a microscopy image containing cells or click 'Try Example Image'
     2. Adjust the confidence threshold if needed (higher values = stricter detection)
     3. Click 'Analyze Image' to process
     4. View results in the various panels: