Update app.py

app.py

@@ -3,17 +3,16 @@ import tempfile
 import os
 import ffmpeg  # Make sure to install ffmpeg-python
 
-def save_uploaded_file(uploaded_file, progress_bar, progress_increment):
+def save_uploaded_file(uploaded_file):
     try:
         with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(uploaded_file.name)[1]) as tmp_file:
             tmp_file.write(uploaded_file.read())
-            current_progress = progress_bar.progress(progress_bar.progress_value + progress_increment)
             return tmp_file.name
     except Exception as e:
         st.error(f"Error saving uploaded file: {e}")
         return None
 
-def add_subtitle_to_video(input_video_path, subtitle_file_path, subtitle_language, soft_subtitle, progress_bar, progress_increment):
+def add_subtitle_to_video(input_video_path, subtitle_file_path, subtitle_language, soft_subtitle):
     output_video_path = os.path.join(tempfile.gettempdir(), f"output-{os.path.basename(input_video_path)}")
     try:
         if soft_subtitle:
@@ -31,30 +30,24 @@ def add_subtitle_to_video(input_video_path, subtitle_file_path, subtitle_languag
                 **{'vf': f'subtitles={subtitle_file_path}'}
             )
         ffmpeg.run(stream, overwrite_output=True)
-        progress_bar.progress(progress_bar.progress_value + progress_increment)
     except ffmpeg.Error as e:
         st.error(f"FFmpeg error: {e.stderr}")
         return None
     return output_video_path
 
 def process_video(video_file, subtitle_file, subtitle_type, subtitle_language):
-    progress_bar_text = "Operation in progress. Please Wait."
-    progress_bar = st.progress(0)
-    progress_bar.progress_value = 0  # Initialize progress value
-    
     if video_file is not None and subtitle_file is not None:
-        video_path = save_uploaded_file(video_file, progress_bar, 30)  # Assuming 30% progress for video upload
-        subtitle_path = save_uploaded_file(subtitle_file, progress_bar, 30)  # Another 30% for subtitle upload
+        video_path = save_uploaded_file(video_file)
+        subtitle_path = save_uploaded_file(subtitle_file)
         if video_path and subtitle_path:
             soft_subtitle = subtitle_type == "Soft"
-            processed_video_path = add_subtitle_to_video(video_path, subtitle_path, subtitle_language, soft_subtitle, progress_bar, 40)  # Remaining 40% for processing
-            progress_bar.progress(100)  # Complete the progress bar
+            processed_video_path = add_subtitle_to_video(video_path, subtitle_path, subtitle_language, soft_subtitle)
             return processed_video_path
     return None
 
 # Streamlit UI
 st.title("Video Subtitler")
-st.markdown("For API use, please visit [this space](https://huggingface.co/spaces/Lenylvt/SRT_to_Video-API).")
+st.markdown("The process can be long. For API use, please visit [this space](https://huggingface.co/spaces/Lenylvt/SRT_to_Video-API).")
 
 video_file = st.file_uploader("📹 Upload a video", type=["mp4", "avi", "mov"])
 subtitle_file = st.file_uploader("📜 Upload subtitle file", type=["srt", "vtt"])
@@ -66,4 +59,4 @@ if st.button("📁 Process Video"):
     if processed_video_path:
         st.video(processed_video_path)
     else:
-        st.error("🔴 Please upload both a video and a subtitle file.")
+        st.error("🔴 Please upload both a video and a subtitle file.")