Update app.py

app.py

@@ -108,6 +108,16 @@ def process_image(image):
 
     return Image.fromarray(img)
 
+
+def use_example_image():
+    # Function to load the example image
+    example_path = "pexels-hikaique-109919.jpg"
+    if os.path.exists(example_path):
+        return Image.open(example_path)
+    else:
+        print(f"Warning: Example image not found at {example_path}")
+        return None
+
 css = """
 #title {
     text-align: center;
@@ -145,6 +155,11 @@ css = """
     border: none;
 }
 
+#example-btn {
+    background-color: #27AE60;
+    border: none;
+}
+
 .output-container {
     margin-top: 1.5rem;
     border: 2px dashed #3498DB;
@@ -192,6 +207,12 @@ with gr.Blocks(css=css, title="Real-Time YOLOv5 Video & Image Object Detection")
             )
             
         with gr.TabItem("Image Detection", elem_classes="tab-item"):
+            # Define image_output before it's referenced
+            image_output = gr.Image(
+                label="Detected Objects", 
+                elem_id="image-output"
+            )
+            
             with gr.Row():
                 image_input = gr.Image(
                     type="pil", 
@@ -199,31 +220,46 @@ with gr.Blocks(css=css, title="Real-Time YOLOv5 Video & Image Object Detection")
                     interactive=True
                 )
             
-            # Define image_output before it's referenced
-            image_output = gr.Image(
-                label="Detected Objects", 
-                elem_id="image-output"
-            )
-            
+            # Place buttons immediately below the upload section
             with gr.Row(elem_classes="button-row"):
                 clear_button = gr.Button(
                     "Clear", 
                     variant="secondary", 
                     elem_id="clear-btn"
                 )
+                example_button = gr.Button(
+                    "Use Example Image", 
+                    variant="secondary", 
+                    elem_id="example-btn"
+                )
                 submit_button = gr.Button(
                     "Detect Objects", 
                     variant="primary", 
                     elem_id="submit-btn"
                 )
             
-            # Now the clear_button can reference image_output
+            with gr.Row(elem_classes="output-container"):
+                pass
+            
+            # Button click functions
+            clear_button.click(
+                fn=lambda: None, 
+                inputs=None, 
+                outputs=image_input
+            )
+            
             clear_button.click(
                 fn=lambda: None, 
                 inputs=None, 
                 outputs=image_output
             )
             
+            example_button.click(
+                fn=use_example_image, 
+                inputs=None, 
+                outputs=image_input
+            )
+            
             submit_button.click(
                 fn=process_image, 
                 inputs=image_input, 
@@ -235,4 +271,4 @@ with gr.Blocks(css=css, title="Real-Time YOLOv5 Video & Image Object Detection")
     This application allows real-time object detection using the YOLOv5 model.
     """, elem_classes="footer")
 
-demo.launch()
+demo.launch()