Update app.py

app.py

@@ -5,26 +5,12 @@ from gtts import gTTS
 import os
 import tempfile
 import speech_recognition as sr
-from googletrans import Translator
 
 def upload_video():
     st.subheader("Paso 1: Cargar archivo MP4")
     uploaded_file = st.file_uploader("Seleccione un archivo MP4", type=["mp4"])
     return uploaded_file
 
-def select_language():
-    st.subheader("Seleccione el idioma:")
-    language_options = ["español", "inglés"]
-    selected_language = st.selectbox("Idioma", language_options)
-    return selected_language
-
-def translate_to_english(text, language):
-    if language == "español":
-        translator = Translator()
-        return translator.translate(text, dest='en').text
-    else:
-        return text
-
 def extract_and_convert_audio(video_file):
     st.subheader("Paso 2: Extraer audio y convertir a MP3")
     with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_video:
@@ -40,7 +26,7 @@ def extract_and_convert_audio(video_file):
     return text_temp_file, audio_path, video.duration, temp_video_path
 
 def audio_to_text(audio_path):
-    st.subheader("Paso 3: Convertir audio a texto temporal")
+    st.subheader("Paso intermedio: Convertir audio a texto temporal")
     audio = AudioSegment.from_file(audio_path)
     temp_wav_file = "temp_audio.wav"
     audio.export(temp_wav_file, format="wav")
@@ -63,19 +49,18 @@ def audio_to_text(audio_path):
                 os.remove(temp_wav_file)
 
 def save_text_to_temp_file(text):
-    st.subheader("Paso 4: Guardar texto en archivo temporal")
+    st.subheader("Paso intermedio: Guardar texto en archivo temporal")
     with tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode="w", encoding="utf-8") as temp_text_file:
         temp_text_file.write(text)
         temp_text_file_path = temp_text_file.name
     return temp_text_file_path
 
-def generate_gtts(text_temp_file, duration, language):
-    st.subheader("Paso 5: Generar GTTS desde el texto")
+def generate_gtts(text_temp_file, duration):
+    st.subheader("Paso 3: Generar GTTS desde el texto")
     with open(text_temp_file, "r", encoding="utf-8") as text_file:
         text_for_gtts = text_file.read()
-
-    lang_code = "es" if language == "español" else "en"
-    tts = gTTS(text_for_gtts, lang=lang_code)
+    
+    tts = gTTS(text_for_gtts, lang="es")
     tts_duration = len(AudioSegment.from_mp3(audio_path)) / 1000
     tts.speed = duration / tts_duration
     tts.save("gtts_audio.mp3")
@@ -86,15 +71,7 @@ def merge_video_with_audio(video_file, audio_file, result_path):
     video = VideoFileClip(video_file)
     audio = AudioFileClip(audio_file)
     video = video.set_audio(audio)
-    result_directory = os.path.dirname(result_path)
-    if not os.path.exists(result_directory):
-        os.makedirs(result_directory)
-
-    try:
-        video.write_videofile(result_path, codec='libx264', audio_codec='aac')
-        st.success("Archivo de video creado con éxito en: {}".format(result_path))
-    except Exception as e:
-        st.error("Error al escribir el archivo de video: {}".format(e))
+    video.write_videofile(result_path, codec='libx264', audio_codec='aac')
     
     return result_path
 
@@ -102,10 +79,8 @@ st.title("Doblado GTTS de Video")
 uploaded_file = upload_video()
 
 if uploaded_file:
-    selected_language = select_language()
     text_temp_file, audio_path, duration, temp_video_path = extract_and_convert_audio(uploaded_file)
-    translated_text = translate_to_english(text_temp_file, selected_language)
-    gtts_audio_path = generate_gtts(translated_text, duration, selected_language)
+    gtts_audio_path = generate_gtts(text_temp_file, duration)
     result_path = merge_video_with_audio(temp_video_path, gtts_audio_path, "video_con_audio.mp4")
     
     st.success("Proceso completado.")
@@ -113,4 +88,4 @@ if uploaded_file:
     video_file = open(result_path, 'rb')
     video_bytes = video_file.read()
     st.video(video_bytes)
-    video_file.close()
+    video_file.close()