RXTIME commited on
Commit
133d81e
·
verified ·
1 Parent(s): c7ace10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -6
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
- # Extrair áudio do vídeo
 
76
  audio_path = os.path.join(UPLOAD_FOLDER, "audio.wav")
77
  extract_audio(video_path, audio_path)
78
 
79
- # Transcrever áudio
 
80
  transcribed_text = transcribe_audio(audio_path)
81
  print("Texto transcrito:", transcribed_text)
82
 
83
- # Traduzir texto
 
84
  translated_text = translate_text(transcribed_text, target_language="pt")
85
  print("Texto traduzido:", translated_text)
86
 
87
- # Sintetizar áudio em português
 
88
  synthesized_audio_path = os.path.join(UPLOAD_FOLDER, "synthesized_audio.wav")
89
  synthesize_speech(translated_text, synthesized_audio_path)
90
 
91
- # Substituir áudio no vídeo
 
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: