Update app.py

app.py

@@ -80,33 +80,27 @@ def inference(model_path):
     shoplifting_prediction = ShopliftingPrediction(model_path, 64, 64, 30)
     shoplifting_prediction.load_model()
     
-    def process_video(video):
+    def process_video(video_frame):
         frames_queue = []
-        if isinstance(video, str):  # If input is a video file path
-            cap = cv2.VideoCapture(video)
-        else:  # If input is a webcam stream
-            cap = cv2.VideoCapture(0)  # Change to 1, 2, etc. for different cameras
-        
-        if not cap.isOpened():
-            raise ValueError("Unable to open video source")
-        
-        ret, previous = cap.read()
-        while ret:
-            ret, frame = cap.read()
-            if not ret:
-                break
-            frame, frames_queue, message = shoplifting_prediction.process_frame(frame, frames_queue, previous)
-            previous = frame
-            yield frame
+        previous = video_frame
+        frame, frames_queue, message = shoplifting_prediction.process_frame(video_frame, frames_queue, previous)
+        return frame, message
 
     return process_video
 
 model_path = 'lrcn_160S_90_90Q.h5'
 process_video = inference(model_path)
 
-iface = gr.Interface(fn=process_video, 
-                     inputs=gr.Video(), 
-                     outputs="video", 
-                     live=True)
+def process_output(video):
+    for frame in video:
+        frame, message = process_video(frame)
+        yield frame, message
+
+iface = gr.Interface(
+    fn=process_output,
+    inputs=gr.Video(),
+    outputs=["video", "text"],
+    live=True
+)
 
 iface.launch()