Update app.py

app.py

@@ -37,8 +37,14 @@ def process_video(uploaded_file):
     
     # Read the video file
     video = cv2.VideoCapture(temp_file_path)
+    total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))  # Get the total number of frames
     frames = []
     
+    # Create a Streamlit progress bar
+    progress_bar = st.progress(0)
+    
+    current_frame = 0
+    
     while True:
         ret, frame = video.read()
         if not ret:
@@ -50,6 +56,11 @@ def process_video(uploaded_file):
         # Process the results and draw boxes on the current frame
         processed_frame = process_results(results, frame)
         frames.append(processed_frame)  # Save the processed frame
+        
+        current_frame += 1
+        # Update the progress bar
+        progress_percentage = current_frame / total_frames
+        progress_bar.progress(progress_percentage)
 
     video.release()
     
@@ -61,6 +72,11 @@ def process_video(uploaded_file):
         out.write(frame)  # Write each processed frame to the video
 
     out.release()
+    
+    # Complete the progress bar
+    progress_bar.progress(1)
+    st.success('Video processing complete!')
+
 
 
 def main():