Update app.py

app.py

@@ -94,14 +94,13 @@ class ShopliftingPrediction:
         probability = math.floor(max(probabilities[0], probabilities[1]) * 100)
         return [probability, predicted_label]
 
-    def Process_Stream(self, stream_url, output_file_path=None, max_duration=30, buffer_size=None):
+    def Process_Stream(self, stream_url, output_file_path=None, buffer_size=None):
         """
         Process a live video stream for shoplifting detection
         
         Args:
             stream_url: URL to the HTTP live stream or path to local video file
             output_file_path: Where to save the processed video (if None, a temp file is created)
-            max_duration: Maximum duration to process in seconds (for streams)
             buffer_size: Size of frames to buffer before processing (if None, use sequence_length)
             
         Returns:
@@ -149,11 +148,6 @@ class ShopliftingPrediction:
         frame_count = 0
         
         while self.video_reader.isOpened():
-            # Check if we've exceeded the max duration for streams
-            if is_url and (time.time() - start_time) > max_duration:
-                logging.info(f"Reached maximum stream capture duration of {max_duration} seconds")
-                break
-                
             # Read the next frame
             ok, frame = self.video_reader.read()
             if not ok:
@@ -210,13 +204,12 @@ class ShopliftingPrediction:
 def inference(model_path):
     shoplifting_prediction = ShopliftingPrediction(model_path, 90, 90, sequence_length=160)
     
-    def process_input(input_source, max_duration=30):
+    def process_input(input_source):
         """
         Process either a video file upload or a streaming URL
         
         Args:
             input_source: Either a URL string or a path to an uploaded video file
-            max_duration: Maximum duration to process for streams (in seconds)
             
         Returns:
             Path to the processed video file
@@ -227,7 +220,7 @@ def inference(model_path):
         if isinstance(input_source, str):
             # Input is likely a URL
             logging.info(f"Processing input as URL: {input_source}")
-            return shoplifting_prediction.Process_Stream(input_source, output_file_path, max_duration)
+            return shoplifting_prediction.Process_Stream(input_source, output_file_path)
         else:
             # Input is likely an uploaded file
             logging.info(f"Processing input as uploaded file: {input_source}")
@@ -258,18 +251,11 @@ with gr.Blocks(title="Shoplifting Detection System") as iface:
                 label="Enter HTTP Live Stream URL", 
                 placeholder="https://example.com/stream.m3u8"
             )
-            max_duration = gr.Slider(
-                minimum=5, 
-                maximum=120, 
-                value=30, 
-                step=5, 
-                label="Max Stream Duration (seconds)"
-            )
             stream_submit = gr.Button("Process Stream")
             stream_output = gr.Video()
             stream_submit.click(
                 fn=process_input, 
-                inputs=[stream_url, max_duration], 
+                inputs=[stream_url], 
                 outputs=stream_output
             )