Rename pages/02_Input_Youtube_Link.py to pages/输入Youtube链接

pages/{02_Input_Youtube_Link.py → 输入Youtube链接}

@@ -18,11 +18,11 @@ from utils import load_lottieurl, change_model
 loaded_model = eccv16(pretrained=True).eval()
 current_model = "None"
 
-st.title("Image & Video Colorizer")
+st.title("图像和视频着色器")
 
 st.write("""
-    ##### Input a YouTube black and white video link and get a colorized version of it.
-    ###### ➠ This space is using CPU Basic so it might take a while to colorize a video.""")
+    ##### 输入 YouTube 黑白视频链接并获取其彩色版本。
+    ###### ➠ 空间使用 CPU Basic，因此可能需要一段时间才能对视频进行着色。""")
 
 @st.cache_data()
 def download_video(link):
@@ -33,25 +33,25 @@ def download_video(link):
 
 def main():
     model = st.selectbox(
-        "Select Model (Both models have their pros and cons, I recommend trying both and keeping the best for you task)",
+        "选择模型（两种模型都有其优点和缺点，建议尝试两种模型并保持最适合您的任务）",
         ["ECCV16", "SIGGRAPH17"], index=0)
 
     loaded_model = change_model(current_model, model)
-    st.write(f"Model is now {model}")
+    st.write(f"现在是 {model}")
 
-    link = st.text_input("YouTube Link (The longer the video, the longer the processing time)")
-    if st.button("Colorize"):
+    link = st.text_input("YouTube链接（视频越长，处理时间越长）")
+    if st.button("着色"):
         if link is not "":
             print(link)
             yt_video = download_video(link)
             print(yt_video)
             col1, col2 = st.columns([0.5, 0.5])
             with col1:
-                st.markdown('<p style="text-align: center;">Before</p>', unsafe_allow_html=True)
+                st.markdown('<p style="text-align: center;">处理前</p>', unsafe_allow_html=True)
                 st.video(yt_video)
             with col2:
-                st.markdown('<p style="text-align: center;">After</p>', unsafe_allow_html=True)
-                with st.spinner("Colorizing frames..."):
+                st.markdown('<p style="text-align: center;">处理后</p>', unsafe_allow_html=True)
+                with st.spinner("上色中..."):
                     # Colorize video frames and store in a list
                     output_frames = []
 
@@ -63,7 +63,7 @@ def main():
 
                     progress_bar = st.progress(0)  # Create a progress bar
                     start_time = time.time()
-                    time_text = st.text("Time Remaining: ")  # Initialize text value
+                    time_text = st.text("剩余时间：")  # Initialize text value
 
                     for _ in tqdm(range(total_frames), unit='frame', desc="Progress"):
                         ret, frame = video.read()
@@ -81,12 +81,12 @@ def main():
                         progress_bar.progress(frames_completed / total_frames)  # Update progress bar
 
                         if frames_completed < total_frames:
-                            time_text.text(f"Time Remaining: {format_time(time_remaining)}")  # Update text value
+                            time_text.text(f"剩余时间： {format_time(time_remaining)}")  # Update text value
                         else:
                             time_text.empty()  # Remove text value
                             progress_bar.empty()
 
-                with st.spinner("Merging frames to video..."):
+                with st.spinner("将帧合并成视频..."):
                     frame_size = output_frames[0].shape[:2]
                     output_filename = "output.mp4"
                     fourcc = cv2.VideoWriter_fourcc(*"mp4v")  # Codec for MP4 video
@@ -121,7 +121,7 @@ def main():
                     # Close and delete the temporary file after processing
                     video.release()
         else:
-            st.warning('Please Type a link', icon="⚠️")
+            st.warning('请输入链接', icon="⚠️")
 
 
 if __name__ == "__main__":