TheChesireCat commited on
Commit
9d02fe5
·
1 Parent(s): 4ca3799
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -74,13 +74,17 @@ def streamlit_video_detect(uploaded_file, centerface, threshold, replacewith, ma
74
 
75
  # Empty the progress bar after completion
76
  my_bar.empty()
 
77
  os.system(f"ffmpeg -y -i {processed_video_temp_path} -c:v libx264 {processed_video_temp_path.split('.')[0]}_processed_.mp4")
78
  # If keep_audio is checked, use FFmpeg to overlay the original audio on the processed video
79
  if keep_audio:
80
- st.write("Overlaying audio. Please wait...")
81
- command = f"ffmpeg -i {processed_video_temp_path.split('.')[0]}_processed_.mp4 -i {temp_video_path} -c:v copy -c:a aac -strict experimental {output_with_audio_temp_path}"
82
- os.system(command)
83
- return output_with_audio_temp_path
 
 
 
84
  else:
85
  return f"{processed_video_temp_path.split('.')[0]}_processed_.mp4"
86
 
 
74
 
75
  # Empty the progress bar after completion
76
  my_bar.empty()
77
+ st.write("Re-encoding video. Please wait..."")
78
  os.system(f"ffmpeg -y -i {processed_video_temp_path} -c:v libx264 {processed_video_temp_path.split('.')[0]}_processed_.mp4")
79
  # If keep_audio is checked, use FFmpeg to overlay the original audio on the processed video
80
  if keep_audio:
81
+ try:
82
+ st.write("Overlaying audio. Please wait...")
83
+ command = f"ffmpeg -y -i {processed_video_temp_path.split('.')[0]}_processed_.mp4 -i {temp_video_path} -c:v copy -c:a aac -strict experimental {output_with_audio_temp_path}"
84
+ os.system(command)
85
+ return output_with_audio_temp_path
86
+ except:
87
+ return f"{processed_video_temp_path.split('.')[0]}_processed_.mp4"
88
  else:
89
  return f"{processed_video_temp_path.split('.')[0]}_processed_.mp4"
90