Update 01_📼_Upload_Video_File.py

01_📼_Upload_Video_File.py

@@ -1,32 +1,22 @@
-import streamlit as st
-from streamlit_lottie import st_lottie
-from models.deep_colorization.colorizers import *
-from utils import load_lottieurl, format_time, colorize_frame
 import os
 import tempfile
+import time
+
 import cv2
 import moviepy.editor as mp
-import time
+import numpy as np
+import streamlit as st
+from streamlit_lottie import st_lottie
 from tqdm import tqdm
 
-st.set_page_config(page_title="Image & Video Colorizer", page_icon="🎨", layout="wide")
+from models.deep_colorization.colorizers import eccv16
+from utils import load_lottieurl, format_time, colorize_frame, change_model
 
+st.set_page_config(page_title="Image & Video Colorizer", page_icon="🎨", layout="wide")
 
 loaded_model = eccv16(pretrained=True).eval()
 current_model = "None"
 
-
-def change_model(current_model, model):
-    if current_model != model:
-        if model == "ECCV16":
-            loaded_model = eccv16(pretrained=True).eval()
-        elif model == "SIGGRAPH17":
-            loaded_model = siggraph17(pretrained=True).eval()
-        return loaded_model
-    else:
-        raise Exception("Model is the same as the current one.")
-
-
 col1, col2 = st.columns([1, 3])
 with col1:
     lottie = load_lottieurl("https://assets5.lottiefiles.com/packages/lf20_RHdEuzVfEL.json")
@@ -36,7 +26,7 @@ with col2:
     st.write("""
     ## B&W Videos Colorizer
     ##### Upload a black and white video and get a colorized version of it.
-    ###### ➠ If you want to colorize multiple videos just upload them one by one.
+    ###### ➠ If you want to colorize multiple videos just upload them all at once.
     ###### ➠ This space is using CPU Basic so it might take a while to colorize a video.
     ###### ➠ If you want more models and GPU available please support this space by donating.""")
 
@@ -79,10 +69,12 @@ def main():
                         # Colorize video frames and store in a list
                         output_frames = []
                         total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
-                        progress_bar = st.empty()
+                        progress_bar = st.progress(0)  # Create a progress bar
 
                         start_time = time.time()
-                        for i in tqdm(range(total_frames), unit='frame', desc="Progress"):
+                        time_text = st.text("Time Remaining: ")  # Initialize text value
+
+                        for _ in tqdm(range(total_frames), unit='frame', desc="Progress"):
                             ret, frame = video.read()
                             if not ret:
                                 break
@@ -95,11 +87,12 @@ def main():
                             frames_remaining = total_frames - frames_completed
                             time_remaining = (frames_remaining / frames_completed) * elapsed_time
 
-                            progress_bar.progress(frames_completed / total_frames)
+                            progress_bar.progress(frames_completed / total_frames)  # Update progress bar
 
                             if frames_completed < total_frames:
-                                progress_bar.text(f"Time Remaining: {format_time(time_remaining)}")
+                                time_text.text(f"Time Remaining: {format_time(time_remaining)}")  # Update text value
                             else:
+                                time_text.empty()  # Remove text value
                                 progress_bar.empty()
 
                     with st.spinner("Merging frames to video..."):