Update app.py
Browse files
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 =
|
34 |
-
video_stream.download("
|
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)
|