xuanzz commited on
Commit
b3b3cc0
·
1 Parent(s): 0047cbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -30,8 +30,8 @@ def download_video(vid_link):
30
  video_stream = yt.streams.filter(
31
  progressive=True, file_extension="mp4"
32
  ).first()
33
- filename = "UIFiles/" + yt.title + ".mp4"
34
- video_stream.download("UIFiles/", yt.title + ".mp4")
35
 
36
  isEngCaptions = False
37
 
@@ -131,13 +131,15 @@ if submitted:
131
  else:
132
  model = load_whisper_model()
133
  tts = load_audio_model()
134
-
 
135
  audio_fname, audio_text, segments_df = vid_audio_to_text(vid_fname, model)
136
-
137
  # subtitles with timestamp
138
  segments_df['Translated Text'] = segments_df['Original Text'].apply((lambda x: translate(x)))
139
  translated_audio_text = translate(audio_text)
140
-
 
141
  sub_vid_fname, audio_fr_name, new_audio_fr = add_subtitles_and_translation_to_movie(segments_df[['Timestamp', 'End_Timestamp', 'Translated Text']], translated_audio_text, vid_fname)
142
 
143
  st.video(sub_vid_fname)
 
30
  video_stream = yt.streams.filter(
31
  progressive=True, file_extension="mp4"
32
  ).first()
33
+ filename = yt.title + ".mp4"
34
+ video_stream.download("", yt.title + ".mp4")
35
 
36
  isEngCaptions = False
37
 
 
131
  else:
132
  model = load_whisper_model()
133
  tts = load_audio_model()
134
+
135
+ st.write("Processing video... This might take more than 10 minutes.")
136
  audio_fname, audio_text, segments_df = vid_audio_to_text(vid_fname, model)
137
+ st.write("Processing audio...")
138
  # subtitles with timestamp
139
  segments_df['Translated Text'] = segments_df['Original Text'].apply((lambda x: translate(x)))
140
  translated_audio_text = translate(audio_text)
141
+
142
+ st.write("Adding subtitles and translated audio...")
143
  sub_vid_fname, audio_fr_name, new_audio_fr = add_subtitles_and_translation_to_movie(segments_df[['Timestamp', 'End_Timestamp', 'Translated Text']], translated_audio_text, vid_fname)
144
 
145
  st.video(sub_vid_fname)