Update app.py
Browse files
app.py
CHANGED
@@ -67,31 +67,38 @@ def replace_audio_in_video(video_path, audio_path, output_path):
|
|
67 |
ffmpeg.output(video.video, audio.audio, output_path, vcodec="copy", acodec="aac").run(overwrite_output=True)
|
68 |
|
69 |
# Função principal para Gradio
|
70 |
-
def translate_video(video):
|
71 |
try:
|
72 |
# O Gradio passa o caminho do arquivo como uma string
|
73 |
video_path = video
|
74 |
|
75 |
-
#
|
|
|
76 |
audio_path = os.path.join(UPLOAD_FOLDER, "audio.wav")
|
77 |
extract_audio(video_path, audio_path)
|
78 |
|
79 |
-
#
|
|
|
80 |
transcribed_text = transcribe_audio(audio_path)
|
81 |
print("Texto transcrito:", transcribed_text)
|
82 |
|
83 |
-
#
|
|
|
84 |
translated_text = translate_text(transcribed_text, target_language="pt")
|
85 |
print("Texto traduzido:", translated_text)
|
86 |
|
87 |
-
#
|
|
|
88 |
synthesized_audio_path = os.path.join(UPLOAD_FOLDER, "synthesized_audio.wav")
|
89 |
synthesize_speech(translated_text, synthesized_audio_path)
|
90 |
|
91 |
-
#
|
|
|
92 |
output_video_path = os.path.join(OUTPUT_FOLDER, "translated_video.mp4")
|
93 |
replace_audio_in_video(video_path, synthesized_audio_path, output_video_path)
|
94 |
|
|
|
|
|
95 |
return output_video_path
|
96 |
|
97 |
except Exception as e:
|
|
|
67 |
ffmpeg.output(video.video, audio.audio, output_path, vcodec="copy", acodec="aac").run(overwrite_output=True)
|
68 |
|
69 |
# Função principal para Gradio
|
70 |
+
def translate_video(video, progress=gr.Progress()):
|
71 |
try:
|
72 |
# O Gradio passa o caminho do arquivo como uma string
|
73 |
video_path = video
|
74 |
|
75 |
+
# Atualizar progresso
|
76 |
+
progress(0, desc="Extraindo áudio do vídeo...")
|
77 |
audio_path = os.path.join(UPLOAD_FOLDER, "audio.wav")
|
78 |
extract_audio(video_path, audio_path)
|
79 |
|
80 |
+
# Atualizar progresso
|
81 |
+
progress(0.25, desc="Transcrevendo áudio...")
|
82 |
transcribed_text = transcribe_audio(audio_path)
|
83 |
print("Texto transcrito:", transcribed_text)
|
84 |
|
85 |
+
# Atualizar progresso
|
86 |
+
progress(0.5, desc="Traduzindo texto...")
|
87 |
translated_text = translate_text(transcribed_text, target_language="pt")
|
88 |
print("Texto traduzido:", translated_text)
|
89 |
|
90 |
+
# Atualizar progresso
|
91 |
+
progress(0.75, desc="Sintetizando áudio em português...")
|
92 |
synthesized_audio_path = os.path.join(UPLOAD_FOLDER, "synthesized_audio.wav")
|
93 |
synthesize_speech(translated_text, synthesized_audio_path)
|
94 |
|
95 |
+
# Atualizar progresso
|
96 |
+
progress(0.9, desc="Substituindo áudio no vídeo...")
|
97 |
output_video_path = os.path.join(OUTPUT_FOLDER, "translated_video.mp4")
|
98 |
replace_audio_in_video(video_path, synthesized_audio_path, output_video_path)
|
99 |
|
100 |
+
# Finalizar progresso
|
101 |
+
progress(1.0, desc="Concluído!")
|
102 |
return output_video_path
|
103 |
|
104 |
except Exception as e:
|