Update app.py

app.py

@@ -20,19 +20,21 @@ except ImportError:
     logger = logging.getLogger(__name__)
     logger.info("Intentando instalar moviepy e imageio-ffmpeg...")
     try:
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-cache-dir", "--force-reinstall", "moviepy>=1.0.3", "imageio-ffmpeg>=0.5.1"])
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-cache-dir", "moviepy>=1.0.3", "imageio-ffmpeg>=0.5.1"])
         from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip, CompositeAudioClip, concatenate_audioclips, AudioClip
         logger.info("MoviePy instalado tras reintento")
     except Exception as e:
         logger.critical(f"Fallo al instalar moviepy: {str(e)}")
-        logger.info("Continuando con placeholder. Video no se generará, solo archivo vacío.")
-        moviepy = None
+        logger.info("Continuando con placeholder para pruebas...")
+        moviepy = None  # Placeholder para evitar errores
 import re
 import math
 import shutil
 import json
 from collections import Counter
+import time
 
+# Configuración de logging
 logging.basicConfig(
     level=logging.DEBUG,
     format='%(asctime)s - %(levelname)s - %(message)s',
@@ -45,6 +47,7 @@ logger.info("="*80)
 logger.info("INICIO DE EJECUCIÓN - GENERADOR DE VIDEOS")
 logger.info("="*80)
 
+# Diccionario de voces TTS disponibles organizadas por idioma
 VOCES_DISPONIBLES = {
     "Español (España)": {
         "es-ES-JuanNeural": "Juan (España) - Masculino",
@@ -107,6 +110,7 @@ VOCES_DISPONIBLES = {
     }
 }
 
+# Función para obtener lista plana de voces para el dropdown
 def get_voice_choices():
     choices = []
     for region, voices in VOCES_DISPONIBLES.items():
@@ -114,8 +118,9 @@ def get_voice_choices():
             choices.append((f"{voice_name} ({region})", voice_id))
     return choices
 
+# Obtener las voces al inicio del script
 AVAILABLE_VOICES = get_voice_choices()
-DEFAULT_VOICE_ID = "es-MX-DaliaNeural"
+DEFAULT_VOICE_ID = "es-MX-DaliaNeural"  # Voz más estable
 DEFAULT_VOICE_NAME = DEFAULT_VOICE_ID
 for text, voice_id in AVAILABLE_VOICES:
     if voice_id == DEFAULT_VOICE_ID:
@@ -126,10 +131,12 @@ if DEFAULT_VOICE_ID not in [v[1] for v in AVAILABLE_VOICES]:
     DEFAULT_VOICE_NAME = AVAILABLE_VOICES[0][0] if AVAILABLE_VOICES else "Dalia (México) - Femenino"
 logger.info(f"Voz por defecto seleccionada (ID): {DEFAULT_VOICE_ID}")
 
+# Clave API de Pexels
 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")
 
+# Inicialización de modelos
 MODEL_NAME = "datificate/gpt2-small-spanish"
 logger.info(f"Inicializando modelo GPT-2: {MODEL_NAME}")
 tokenizer = None
@@ -157,6 +164,7 @@ def buscar_videos_pexels(query, api_key, per_page=5):
     if not api_key:
         logger.warning("No se puede buscar en Pexels: API Key no configurada.")
         return []
+
     logger.debug(f"Buscando en Pexels: '{query}' | Resultados: {per_page}")
     headers = {"Authorization": api_key}
     try:
@@ -192,8 +200,10 @@ def generate_script(prompt, max_length=150):
     if not tokenizer or not model:
         logger.warning("Modelos GPT-2 no disponibles - Usando prompt original como guion.")
         return prompt.strip()
+
     instruction_phrase_start = "Escribe un guion corto, interesante y coherente sobre:"
     ai_prompt = f"{instruction_phrase_start} {prompt}"
+
     try:
         inputs = tokenizer(ai_prompt, return_tensors="pt", truncation=True, max_length=512)
         device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -254,6 +264,7 @@ async def text_to_speech(text, output_path, voice):
         logger.warning(f"edge_tts falló, intentando gTTS...")
     except Exception as e:
         logger.error(f"Error en edge_tts con voz '{voice}': {str(e)}")
+
     try:
         tts = gTTS(text=text, lang='es')
         tts.save(output_path)
@@ -372,14 +383,15 @@ async def crear_video_async(prompt_type, input_text, selected_voice, musica_file
     video_final = None
     source_clips = []
     clips_to_concatenate = []
-    output_path = None
 
     try:
+        # 1. Generar o usar guion
         guion = generate_script(input_text) if prompt_type == "Generar Guion con IA" else input_text.strip()
         logger.info(f"Guion final ({len(guion)} chars): '{guion[:100]}...'")
         if not guion.strip():
             raise ValueError("El guion está vacío.")
 
+        # 2. Generar audio de voz
         voz_path = os.path.join(temp_dir_intermediate, "voz.mp3")
         tts_voices_to_try = [selected_voice, "es-MX-DaliaNeural"]
         tts_success = False
@@ -433,16 +445,18 @@ async def crear_video_async(prompt_type, input_text, selected_voice, musica_file
             audio_duration = audio_tts_original.duration
         else:
             logger.warning("MoviePy no disponible, asumiendo duración mínima para audio")
-            audio_duration = 1.0
+            audio_duration = 1.0  # Valor placeholder
         logger.info(f"Duración audio voz: {audio_duration:.2f} segundos")
         if audio_duration < 1.0:
             raise ValueError("Audio de voz demasiado corto.")
 
+        # 3. Extraer palabras clave
         keywords = extract_visual_keywords_from_script(guion)
         if not keywords:
             keywords = ["video", "background"]
         logger.info(f"Palabras clave: {keywords}")
 
+        # 4. Buscar y descargar videos
         videos_data = []
         total_desired_videos = 10
         per_page_per_keyword = max(1, total_desired_videos // len(keywords))
@@ -477,13 +491,15 @@ async def crear_video_async(prompt_type, input_text, selected_voice, musica_file
         if not video_paths:
             raise ValueError("No se descargaron videos utilizables.")
 
+        # 5. Procesar y concatenar clips de video
         if not moviepy:
             logger.warning("MoviePy no disponible, retornando placeholder...")
             output_filename = f"video_{int(datetime.now().timestamp())}.mp4"
-            output_path = os.path.join(temp_dir_intermediate, output_filename)
-            open(output_path, 'a').close()
-            download_url = f"https://gnosticdev-invideo-basic.hf.space/file={output_path}"
-            return output_path, download_url
+            persistent_path = os.path.join("/data", output_filename)
+            os.makedirs("/data", exist_ok=True)
+            open(persistent_path, 'a').close()  # Crea archivo vacío como placeholder
+            download_url = f"https://gnosticdev-invideo-basic.hf.space/file={persistent_path}"
+            return persistent_path, download_url
 
         current_duration = 0
         min_clip_duration = 0.5
@@ -512,6 +528,7 @@ async def crear_video_async(prompt_type, input_text, selected_voice, musica_file
         if video_base.duration is None or video_base.duration <= 0:
             raise ValueError("Video base inválido.")
 
+        # Ajustar duración del video
         if video_base.duration < audio_duration:
             num_full_repeats = int(audio_duration // video_base.duration)
             remaining_duration = audio_duration % video_base.duration
@@ -523,6 +540,7 @@ async def crear_video_async(prompt_type, input_text, selected_voice, musica_file
         elif video_base.duration > audio_duration:
             video_base = video_base.subclip(0, audio_duration)
 
+        # 6. Manejar música de fondo
         final_audio = audio_tts
         if musica_file:
             try:
@@ -543,9 +561,13 @@ async def crear_video_async(prompt_type, input_text, selected_voice, musica_file
         if abs(final_audio.duration - video_base.duration) > 0.2:
             final_audio = final_audio.subclip(0, video_base.duration)
 
+        # 7. Combinar audio y video
         video_final = video_base.set_audio(final_audio)
         output_filename = f"video_{int(datetime.now().timestamp())}.mp4"
         output_path = os.path.join(temp_dir_intermediate, output_filename)
+        persistent_dir = "/data"
+        os.makedirs(persistent_dir, exist_ok=True)
+        persistent_path = os.path.join(persistent_dir, output_filename)
 
         video_final.write_videofile(
             output_path,
@@ -558,12 +580,13 @@ async def crear_video_async(prompt_type, input_text, selected_voice, musica_file
             logger='bar'
         )
 
-        download_url = f"https://gnosticdev-invideo-basic.hf.space/file={output_path}"
-        logger.info(f"Video guardado en: {output_path}")
+        shutil.move(output_path, persistent_path)
+        download_url = f"https://gnosticdev-invideo-basic.hf.space/file={persistent_path}"
+        logger.info(f"Video guardado en: {persistent_path}")
         logger.info(f"URL de descarga: {download_url}")
         total_time = (datetime.now() - start_time).total_seconds()
         logger.info(f"Video generado en {total_time:.2f}s")
-        return output_path, download_url
+        return persistent_path, download_url
 
     except ValueError as ve:
         logger.error(f"Error controlado: {str(ve)}")
@@ -603,13 +626,13 @@ async def crear_video_async(prompt_type, input_text, selected_voice, musica_file
             except:
                 pass
         for path in temp_intermediate_files:
-            if os.path.isfile(path) and (output_path is None or path != output_path):
+            if os.path.isfile(path) and path != persistent_path:
                 try:
                     os.remove(path)
                 except:
                     logger.warning(f"No se pudo eliminar {path}")
         try:
-            if os.path.exists(temp_dir_intermediate) and (output_path is None or not output_path.startswith(temp_dir_intermediate)):
+            if os.path.exists(temp_dir_intermediate):
                 shutil.rmtree(temp_dir_intermediate)
         except:
             logger.warning(f"No se pudo eliminar directorio temporal {temp_dir_intermediate}")
@@ -620,7 +643,7 @@ async def run_app_async(prompt_type, prompt_ia, prompt_manual, musica_file, sele
     input_text = prompt_ia if prompt_type == "Generar Guion con IA" else prompt_manual
     output_video = None
     output_file = None
-    status_msg = gr.update(value="⏳ Iniciando generación de video...")
+    status_msg = gr.update(value="⏳ Procesando... Esto puede tomar hasta 1 hora.")
 
     if not input_text or not input_text.strip():
         logger.warning("Texto de entrada vacío.")
@@ -632,53 +655,34 @@ async def run_app_async(prompt_type, prompt_ia, prompt_manual, musica_file, sele
         selected_voice = DEFAULT_VOICE_ID
 
     try:
-        task = asyncio.create_task(crear_video_async(prompt_type, input_text, selected_voice, musica_file))
-        timeout = 600
-        interval = 5
-        elapsed = 0
-        while elapsed < timeout:
-            if task.done():
-                video_path, download_url = await task
-                if video_path and os.path.exists(video_path):
-                    output_video = video_path
-                    output_file = video_path
-                    if not moviepy and os.path.getsize(video_path) <= 100:
-                        status_msg = gr.update(value=f"⚠️ Video no generado (MoviePy no disponible). Archivo vacío: {download_url} (descárgalo pronto, /tmp es temporal). Revisa video_generator_full.log.")
-                    else:
-                        status_msg = gr.update(value=f"✅ Video generado exitosamente. Descarga: {download_url} (Nota: Archivos en /tmp son temporales, descárgalo pronto)")
-                    logger.info(f"Retornando video_path: {video_path}, URL: {download_url}")
-                    return output_video, gr.File(value=output_file, label="Descargar Video"), status_msg
-                else:
-                    status_msg = gr.update(value="❌ Error: Falló la generación del video.")
-                    logger.error("No se generó video_path válido.")
-                    return None, None, status_msg
-            await asyncio.sleep(interval)
-            elapsed += interval
-            status_msg = gr.update(value=f"⏳ Procesando... Tiempo transcurrido: {elapsed}s")
-            logger.debug(f"Esperando video, tiempo transcurrido: {elapsed}s")
-        logger.error("Tiempo de espera excedido para la generación del video.")
-        status_msg = gr.update(value="❌ Error: Tiempo de espera excedido (10 minutos).")
-        task.cancel()
-        try:
-            await task
-        except asyncio.CancelledError:
-            pass
-        return None, None, status_msg
+        logger.info("Iniciando generación de video...")
+        video_path, download_url = await crear_video_async(prompt_type, input_text, selected_voice, musica_file)
+        if video_path and os.path.exists(video_path):
+            output_video = video_path
+            output_file = video_path
+            status_msg = gr.update(value=f"✅ Video generado exitosamente. Descarga: {download_url}")
+            logger.info(f"Retornando video_path: {video_path}, URL: {download_url}")
+        else:
+            status_msg = gr.update(value="❌ Error: Falló la generación del video.")
+            logger.error("No se generó video_path válido.")
     except ValueError as ve:
         logger.warning(f"Error de validación: {str(ve)}")
         status_msg = gr.update(value=f"⚠️ Error: {str(ve)}")
-        return None, None, status_msg
     except Exception as e:
         logger.critical(f"Error crítico: {str(e)}")
         status_msg = gr.update(value=f"❌ Error inesperado: {str(e)}")
-        return None, None, status_msg
+    finally:
+        logger.info("Finalizando run_app_async")
+        return output_video, gr.File(value=output_file, label="Descargar Video"), status_msg
 
 def run_app(prompt_type, prompt_ia, prompt_manual, musica_file, selected_voice):
     return asyncio.run(run_app_async(prompt_type, prompt_ia, prompt_manual, musica_file, selected_voice))
 
+# Interfaz de Gradio
 with gr.Blocks(title="Generador de Videos con IA", theme=gr.themes.Soft()) as app:
     gr.Markdown("# 🎬 Generador Automático de Videos con IA")
     gr.Markdown("Genera videos cortos a partir de un tema o guion, usando imágenes de archivo de Pexels y voz generada.")
+
     with gr.Row():
         with gr.Column():
             prompt_type = gr.Radio(
@@ -729,13 +733,15 @@ with gr.Blocks(title="Generador de Videos con IA", theme=gr.themes.Soft()) as ap
                 placeholder="Esperando acción...",
                 value="Esperando entrada..."
             )
+
     prompt_type.change(
         fn=lambda x: (gr.update(visible=x == "Generar Guion con IA"), gr.update(visible=x == "Usar Mi Guion")),
         inputs=prompt_type,
         outputs=[ia_guion_column, manual_guion_column]
     )
+
     generate_btn.click(
-        fn=lambda: (None, None, gr.update(value="⏳ Iniciando generación de video...")),
+        fn=lambda: (None, None, gr.update(value="⏳ Procesando... Esto puede tomar hasta 1 hora.")),
         outputs=[video_output, file_output, status_output]
     ).then(
         fn=run_app,
@@ -743,10 +749,11 @@ with gr.Blocks(title="Generador de Videos con IA", theme=gr.themes.Soft()) as ap
         outputs=[video_output, file_output, status_output],
         queue=True
     ).then(
-        fn=lambda video_path, file_output, status_msg: gr.update(visible=bool(file_output)),
+        fn=lambda video_path, file_output, status_msg: gr.update(visible=file_output.value is not None),
         inputs=[video_output, file_output, status_output],
         outputs=[file_output]
     )
+
     gr.Markdown("### Instrucciones:")
     gr.Markdown("""
     1. Configura la variable de entorno `PEXELS_API_KEY`.
@@ -754,8 +761,8 @@ with gr.Blocks(title="Generador de Videos con IA", theme=gr.themes.Soft()) as ap
     3. Sube música (opcional).
     4. Selecciona la voz.
     5. Haz clic en "✨ Generar Video".
-    6. Revisa el estado. Si el video se genera, estará disponible en /tmp (descárgalo pronto, es temporal).
-    7. Si el video es 0 KB, revisa video_generator_full.log (probable fallo de MoviePy).
+    6. Revisa el estado. Si el video se genera, estará disponible en /data.
+    7. Consulta `video_generator_full.log` para detalles.
     """)
 
 if __name__ == "__main__":