Update app.py

app.py

@@ -9,7 +9,7 @@ import gradio as gr
 import torch
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
 from keybert import KeyBERT
-# CORRECCIÓN CRÍTICA DEFINITIVA: Eliminar 'concatenate_videoclip' (singular) de la importación
+# Importación correcta: Solo 'concatenate_videoclips'
 from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip, CompositeAudioClip, concatenate_audioclips, AudioClip
 import re
 import math
@@ -35,7 +35,7 @@ logger.info("="*80)
 PEXELS_API_KEY = os.environ.get("PEXELS_API_KEY")
 if not PEXELS_API_KEY:
     logger.critical("NO SE ENCONTRÓ PEXELS_API_KEY EN VARIABLES DE ENTORNO")
-    # raise ValueError("API key de Pexels no configurada") # Descomentar para forzar fallo si no está
+    # raise ValueError("API key de Pexels no configurada")
 
 # Inicialización de modelos
 MODEL_NAME = "datificate/gpt2-small-spanish"
@@ -387,21 +387,19 @@ def crear_video(prompt_type, input_text, musica_file=None):
         primary_voice = "es-ES-JuanNeural"
         fallback_voice = "es-ES-ElviraNeural" # Otra voz en español
         tts_success = False
-        retries = 3 # Número de intentos
+        retries = 3
 
         for attempt in range(retries):
             current_voice = primary_voice if attempt == 0 else fallback_voice
             if attempt > 0: logger.warning(f"Reintentando TTS ({attempt+1}/{retries})...")
             logger.info(f"Intentando TTS con voz: {current_voice}")
             try:
-                # Llamar a la función async text_to_speech
                 tts_success = asyncio.run(text_to_speech(guion, voz_path, voice=current_voice))
                 if tts_success:
                     logger.info(f"TTS exitoso en intento {attempt + 1} con voz {current_voice}.")
-                    break # Salir del bucle de reintentos si tiene éxito
+                    break
             except Exception as e:
-                 # La excepción ya se registra dentro de text_to_speech
-                 pass # Continuar al siguiente intento
+                 pass
 
             if not tts_success and attempt == 0 and primary_voice != fallback_voice:
                  logger.warning(f"Fallo con voz {primary_voice}, intentando voz de respaldo: {fallback_voice}")
@@ -409,14 +407,12 @@ def crear_video(prompt_type, input_text, musica_file=None):
                  logger.warning(f"Fallo con voz {current_voice}, reintentando...")
 
 
-        # Verificar si el archivo fue creado después de todos los intentos
         if not tts_success or not os.path.exists(voz_path) or os.path.getsize(voz_path) <= 100:
              logger.error(f"Fallo en la generación de voz después de {retries} intentos. Archivo de audio no creado o es muy pequeño.")
              raise ValueError("Error generando voz a partir del guion (fallo de TTS).")
 
-        temp_intermediate_files.append(voz_path) # Añadir a la lista de limpieza si se creó
+        temp_intermediate_files.append(voz_path)
 
-        # Continuar cargando el archivo de audio generado
         audio_tts_original = AudioFileClip(voz_path)
 
         if audio_tts_original.reader is None or audio_tts_original.duration is None or audio_tts_original.duration <= 0:
@@ -709,8 +705,8 @@ def crear_video(prompt_type, input_text, musica_file=None):
 
                 if musica_audio_looped:
                     composite_audio = CompositeAudioClip([
-                        musica_audio_looped.volumex(0.2),
-                        audio_tts_original.volumex(1.0)
+                        musica_audio_looped.volumex(0.2), # Volumen 20% para música
+                        audio_tts_original.volumex(1.0) # Volumen 100% para voz
                     ])
 
                     if composite_audio.duration is None or composite_audio.duration <= 0:
@@ -721,7 +717,7 @@ def crear_video(prompt_type, input_text, musica_file=None):
                     else:
                          logger.info("Mezcla de audio completada (voz + música).")
                          final_audio = composite_audio
-                         musica_audio = musica_audio_looped
+                         musica_audio = musica_audio_looped # Asignar para limpieza
 
             except Exception as e:
                 logger.warning(f"Error procesando música de fondo: {str(e)}", exc_info=True)
@@ -847,12 +843,12 @@ def crear_video(prompt_type, input_text, musica_file=None):
              logger.info(f"Directorio temporal intermedio {temp_dir_intermediate} persistirá para que Gradio lea el video final.")
 
 
-# CAMBIO CRÍTICO: run_app define 4 argumentos
+# La función run_app ahora recibe todos los inputs de texto y el archivo de música
 def run_app(prompt_type, prompt_ia, prompt_manual, musica_file):
     logger.info("="*80)
     logger.info("SOLICITUD RECIBIDA EN INTERFAZ")
 
-    # La lógica para elegir el texto de entrada YA ESTÁ AQUÍ y funciona con ambos inputs pasados
+    # Elegir el texto de entrada basado en el prompt_type
     input_text = prompt_ia if prompt_type == "Generar Guion con IA" else prompt_manual
 
     output_video = None
@@ -861,6 +857,7 @@ def run_app(prompt_type, prompt_ia, prompt_manual, musica_file):
 
     if not input_text or not input_text.strip():
         logger.warning("Texto de entrada vacío.")
+        # Retornar None para video y archivo, actualizar estado con mensaje de error
         return None, None, gr.update(value="⚠️ Por favor, ingresa texto para el guion o el tema.", interactive=False)
 
     logger.info(f"Tipo de entrada: {prompt_type}")
@@ -872,14 +869,14 @@ def run_app(prompt_type, prompt_ia, prompt_manual, musica_file):
 
     try:
         logger.info("Llamando a crear_video...")
-        # Pasar el input_text elegido y el archivo de música
+        # Pasar el input_text elegido y el archivo de música a crear_video
         video_path = crear_video(prompt_type, input_text, musica_file)
 
         if video_path and os.path.exists(video_path):
             logger.info(f"crear_video retornó path: {video_path}")
             logger.info(f"Tamaño del archivo de video retornado: {os.path.getsize(video_path)} bytes")
-            output_video = video_path
-            output_file = video_path
+            output_video = video_path # Establecer valor del componente de video
+            output_file = video_path   # Establecer valor del componente de archivo para descarga
             status_msg = gr.update(value="✅ Video generado exitosamente.", interactive=False)
         else:
             logger.error(f"crear_video no retornó un path válido o el archivo no existe: {video_path}")
@@ -893,6 +890,7 @@ def run_app(prompt_type, prompt_ia, prompt_manual, musica_file):
         status_msg = gr.update(value=f"❌ Error inesperado: {str(e)}", interactive=False)
     finally:
         logger.info("Fin del handler run_app.")
+        # Retornar las tres salidas esperadas por el evento click
         return output_video, output_file, status_msg
 
 
@@ -914,11 +912,12 @@ with gr.Blocks(title="Generador de Videos con IA", theme=gr.themes.Soft(), css="
             )
 
             # Contenedores para los campos de texto para controlar la visibilidad
+            # Nombrados para que coincidan con los outputs del evento change
             with gr.Column(visible=True) as ia_guion_column:
                 prompt_ia = gr.Textbox(
                     label="Tema para IA",
                     lines=2,
-                    placeholder="Ej: Un paisaje natural con montañas y ríos al amanecer...",
+                    placeholder="Ej: Un paisaje natural con montañas y ríos al amanecer, mostrando la belleza de la naturaleza...",
                     max_lines=4,
                     value=""
                 )
@@ -927,7 +926,7 @@ with gr.Blocks(title="Generador de Videos con IA", theme=gr.themes.Soft(), css="
                 prompt_manual = gr.Textbox(
                     label="Tu Guion Completo",
                     lines=5,
-                    placeholder="Ej: En este video exploraremos los misterios del océano...",
+                    placeholder="Ej: En este video exploraremos los misterios del océano. Veremos la vida marina fascinante y los arrecifes de coral vibrantes. ¡Acompáñanos en esta aventura subacuática!",
                     max_lines=10,
                     value=""
                 )
@@ -960,31 +959,38 @@ with gr.Blocks(title="Generador de Videos con IA", theme=gr.themes.Soft(), css="
                 value="Esperando entrada..."
             )
 
-    # CAMBIO CRÍTICO: Controlar la visibilidad de las COLUMNAS contenedoras
+    # Evento para mostrar/ocultar los campos de texto según el tipo de prompt
+    # Apuntar a los componentes Column padre para controlar la visibilidad
     prompt_type.change(
         lambda x: (gr.update(visible=x == "Generar Guion con IA"),
                    gr.update(visible=x == "Usar Mi Guion")),
         inputs=prompt_type,
-        # Apuntar a los componentes Column padre
+        # Pasar los componentes Column
         outputs=[ia_guion_column, manual_guion_column]
     )
 
     # Evento click del botón de generar video
     generate_btn.click(
         # Acción 1 (síncrona): Resetear salidas y establecer estado a procesando
-        lambda: (None, None, gr.update(value="⏳ Procesando... Esto puede tomar 2-5 minutos o más para videos largos.", interactive=False)),
+        # Retorna None para los 3 outputs iniciales
+        lambda: (None, None, gr.update(value="⏳ Procesando... Esto puede tomar varios minutos.", interactive=False)),
         outputs=[video_output, file_output, status_output],
-        queue=True,
+        queue=True, # Usar la cola de Gradio para tareas largas
     ).then(
         # Acción 2 (asíncrona): Llamar a la función principal de procesamiento
         run_app,
-        # CAMBIO CRÍTICO: Pasar los 4 argumentos definidos por run_app
+        # PASAR TODOS LOS INPUTS DE LA INTERFAZ que run_app espera
         inputs=[prompt_type, prompt_ia, prompt_manual, musica_input],
+        # run_app retornará los 3 outputs esperados aquí
         outputs=[video_output, file_output, status_output]
     ).then(
         # Acción 3 (síncrona): Hacer visible el enlace de descarga si se retornó un archivo
-        lambda video_path, file_path: gr.update(visible=file_path is not None),
-        inputs=[video_output, file_output],
+        # Esta función recibe las salidas de la Acción 2 (video_path, file_path, status_msg)
+        # Solo necesitamos video_path o file_path para decidir si mostrar el enlace
+        lambda video_path, file_path, status_msg: gr.update(visible=file_path is not None),
+        # Inputs son las salidas de la función .then() anterior
+        inputs=[video_output, file_output, status_output],
+        # Actualizamos la visibilidad del componente file_output
         outputs=[file_output]
     )