Update app.py

app.py

@@ -33,40 +33,25 @@ def detect_flash(image: Image.Image):
 
     return output_image, message
 
-def extract_first_frame(video):
-    cap = cv2.VideoCapture(video)
-    success, frame = cap.read()
-    cap.release()
-    if success:
-        # Convert BGR to RGB
-        frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-        return Image.fromarray(frame_rgb)
-    return None
-
-def process_webcam_video(video):
-    image = extract_first_frame(video)
-    if image:
-        return detect_flash(image)
-    else:
-        return None, "Could not read frame from video."
+def clear_all():
+    return None, None, ""
 
 # Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("Flash Detection with YOLOv8")
-    gr.Markdown("Upload an image or record from your webcam to detect flash.")
+    gr.Markdown("Upload an image to detect flash.")
 
-    with gr.Tab("Upload Image"):
+    with gr.Row():
         img_input = gr.Image(type="pil", label="Upload Image")
-        detect_btn = gr.Button("Detect Flash")
         output_img = gr.Image(type="pil", label="Detected Image")
-        output_text = gr.Textbox(label="Detection Info")
-        detect_btn.click(fn=detect_flash, inputs=img_input, outputs=[output_img, output_text])
 
-    with gr.Tab("Use Webcam (Record Video)"):
-        webcam_video = gr.Video(label="Record from Webcam")
-        webcam_btn = gr.Button("Detect Flash")
-        webcam_img = gr.Image(type="pil", label="Detected Image")
-        webcam_text = gr.Textbox(label="Detection Info")
-        webcam_btn.click(fn=process_webcam_video, inputs=webcam_video, outputs=[webcam_img, webcam_text])
+    output_text = gr.Textbox(label="Detection Info", lines=6)
+
+    with gr.Row():
+        detect_btn = gr.Button("Detect Flash")
+        clear_btn = gr.Button("Clear")
+
+    detect_btn.click(fn=detect_flash, inputs=img_input, outputs=[output_img, output_text])
+    clear_btn.click(fn=clear_all, inputs=[], outputs=[img_input, output_img, output_text])
 
 demo.launch()