Update app.py

app.py

@@ -3,13 +3,17 @@ import gradio as gr
 import cv2 as cv
 from ultralytics import YOLO
 
+# Global variables to control the process
+process = False
+model = None
+cap = None
 
 def load_yolo_model():
     return YOLO('yolov8n-seg.pt')
 
-
-def process_video(model, cap):
-    while cap.isOpened():
+def process_video():
+    global process, model, cap
+    while process and cap.isOpened():
         ret, image = cap.read()
         if not ret:
             break
@@ -26,26 +30,28 @@ def process_video(model, cap):
     cap.release()
     cv.destroyAllWindows()
 
-
 def segment_video(uploaded_video):
+    global model, cap
     model = load_yolo_model()
     cap = cv.VideoCapture(uploaded_video.name)
-    process_video(model, cap)
-
+    process_video()
 
 def segment_webcam():
+    global model, cap
     model = load_yolo_model()
     cap = cv.VideoCapture(0)
-    process_video(model, cap)
-
+    process_video()
 
 def process_inputs(start_button, stop_button, mode_selection, uploaded_video):
+    global process
     if start_button:
+        process = True
         if mode_selection == "Video" and uploaded_video:
             segment_video(uploaded_video)
         elif mode_selection == "Webcam":
             segment_webcam()
-
+    elif stop_button:
+        process = False
 
 iface = gr.Interface(
     fn=process_inputs,