Update app.py

app.py

@@ -45,6 +45,9 @@ if video_file is not None:
     # Collect frames in a list
     frames = []
 
+    # Set the desired size as a percentage of the original size
+    target_size_percentage = 0.65
+
     while cap.isOpened() and not processing_interrupted:
         alert_flag = False
         alert_reason = []
@@ -56,6 +59,13 @@ if video_file is not None:
             # st.warning("Can't receive frame (stream end?). Exiting ...")
             break
 
+        # Calculate the target size
+        target_width = int(frame.shape[1] * target_size_percentage)
+        target_height = int(frame.shape[0] * target_size_percentage)
+
+        # Resize the frame
+        frame = cv2.resize(frame, (target_width, target_height))
+
         if success:
             # Check if the stop button is clicked
             if stop_button:
@@ -124,3 +134,12 @@ if video_file is not None:
         progress_bar_display = st.progress(0)
         fps_delay = 1 / 24  # Delay to achieve 24 FPS
 
+        for i, frame in enumerate(frames):
+            video_placeholder.image(frame, channels="BGR", caption="YOLOv8 Inference")
+            # Update display progress bar
+            progress_bar_display.progress((i + 1) / len(frames))
+            # Introduce a delay to achieve 24 FPS
+            time.sleep(fps_delay)
+
+        # Display completion message
+        progress_bar_processing_slot.text("Video Playback Finished!")