Update app.py

app.py

@@ -40,12 +40,15 @@ if uploaded_video is not None:
     frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
     fps = cap.get(cv2.CAP_PROP_FPS)
+    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
 
     # Create a VideoWriter object to save the colorized video
     fourcc = cv2.VideoWriter_fourcc(*"mp4v")
     out = cv2.VideoWriter(output_path, fourcc, fps, (frame_width, frame_height))
 
+    # Initialize progress bar and frame counter
     frame_count = 0
+    progress_bar = st.progress(0)
     stframe = st.empty()  # Placeholder for showing frames in Streamlit
 
     # Process each frame
@@ -55,7 +58,8 @@ if uploaded_video is not None:
             break
 
         frame_count += 1
-        print(f"Frame: {frame_count}")
+        st.write(f"Processing frame {frame_count} of {total_frames}")
+
         # Convert frame to LAB color space and preprocess
         normalized = frame.astype("float32") / 255.0
         lab = cv2.cvtColor(normalized, cv2.COLOR_BGR2LAB)
@@ -74,7 +78,9 @@ if uploaded_video is not None:
         colorized = cv2.cvtColor(colorized, cv2.COLOR_LAB2BGR)
         colorized = (255 * colorized).astype("uint8")
 
-              
+        # Update the progress bar and display the current frame in Streamlit
+        progress_bar.progress(frame_count / total_frames)
+        stframe.image(colorized, channels="BGR", caption=f"Colorized Frame {frame_count}")
 
         # Write colorized frame to output
         out.write(colorized)
@@ -86,4 +92,4 @@ if uploaded_video is not None:
     # Provide a download link for the colorized video
     st.success("Video colorization completed!")
     with open(output_path, "rb") as file:
-        btn = st.download_button(label="Download Colorized Video", data=file, file_name="colorized_video.mp4", mime="video/mp4")
+        st.download_button(label="Download Colorized Video", data=file, file_name="colorized_video.mp4", mime="video/mp4")