Update app.py

app.py

@@ -59,7 +59,7 @@ def process_video(video_path):
     while cap.isOpened():
         ret, frame = cap.read()
         if not ret:
-            break  # Break if no more frames
+            break  
         
         start_time = time.time()
         
@@ -109,15 +109,6 @@ 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;
@@ -155,11 +146,6 @@ css = """
     border: none;
 }
 
-#example-btn {
-    background-color: #27AE60;
-    border: none;
-}
-
 .output-container {
     margin-top: 1.5rem;
     border: 2px dashed #3498DB;
@@ -207,68 +193,47 @@ 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", 
-                    label="Upload Image", 
-                    interactive=True
-                )
-            
-            # 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"
-                )
-            
-            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, 
-                outputs=image_output
-            )
+    with gr.Row():
+        image_input = gr.Image(
+            type="pil", 
+            label="Upload Image", 
+            interactive=True
+        )
+    
+    with gr.Row(elem_classes="button-row"):  # Move button row here just below image_input
+        clear_button = gr.Button(
+            "Clear", 
+            variant="secondary", 
+            elem_id="clear-btn"
+        )
+        submit_button = gr.Button(
+            "Detect Objects", 
+            variant="primary", 
+            elem_id="submit-btn"
+        )
+    
+    # Define image_output below input and buttons
+    image_output = gr.Image(
+        label="Detected Objects", 
+        elem_id="image-output"
+    )
+    
+    clear_button.click(
+        fn=lambda: None, 
+        inputs=None, 
+        outputs=image_output
+    )
+    
+    submit_button.click(
+        fn=process_image, 
+        inputs=image_input, 
+        outputs=image_output
+    )
+
     
     gr.Markdown("""
     ### Powered by YOLOv5.
     This application allows real-time object detection using the YOLOv5 model.
     """, elem_classes="footer")
 
-demo.launch()
+demo.launch()