Update app.py

app.py

@@ -6,7 +6,6 @@ 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")
@@ -92,13 +91,6 @@ 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("""
@@ -109,9 +101,6 @@ 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,
@@ -120,6 +109,7 @@ 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")
@@ -138,12 +128,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         with gr.Column(scale=1):
             density_plot = gr.Plot(label="Cell Density Heatmap")
     
-    # Handle button clicks
-    example_btn.click(
-        load_example_image,
-        outputs=input_image
-    )
-    
+    # Handle button click
     analyze_btn.click(
         process_image,
         inputs=[input_image, conf_slider],
@@ -152,7 +137,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     
     gr.Markdown("""
     ### Instructions:
-    1. Upload a microscopy image containing cells or click 'Try Example Image'
+    1. Upload a microscopy image containing cells
     2. Adjust the confidence threshold if needed (higher values = stricter detection)
     3. Click 'Analyze Image' to process
     4. View results in the various panels: