Actualización de app.py para guardar videos en carpeta outputs

app.py

@@ -2,11 +2,12 @@ import gradio as gr
 from tts_module import get_voices, text_to_speech  # Usamos el tts_module.py actualizado
 from pexels_api import search_pexels
 from moviepy.editor import AudioFileClip, VideoFileClip, CompositeAudioClip, concatenate_audioclips, concatenate_videoclips
-
 import asyncio
 import os
 import requests
-import tempfile
+import time
+import threading
+from datetime import datetime, timedelta
 
 # Forzar instalación de moviepy si no está disponible
 def install(package):
@@ -27,7 +28,11 @@ except ImportError:
     except ImportError:
         raise ImportError("Error crítico: No se pudo instalar moviepy.editor. Verifica las dependencias.")
 
-# Ajustar música de fondo (repetición automática)
+# Define la carpeta de salida
+output_folder = "outputs"
+os.makedirs(output_folder, exist_ok=True)  # Crea la carpeta si no existe
+
+# Función para ajustar música de fondo (repetición automática)
 def adjust_background_music(video_duration, music_file):
     music = AudioFileClip(music_file)
     if music.duration < video_duration:
@@ -39,11 +44,10 @@ def adjust_background_music(video_duration, music_file):
     music = music.volumex(0.2)  # Reducir volumen al 20%
     return music
 
-# Concatenar múltiples videos de Pexels
+# Función para concatenar múltiples videos de Pexels
 def concatenate_pexels_videos(text, num_videos=5):
     sentences = [sentence.strip() for sentence in text.split(".") if sentence.strip()]
     video_links = []
-
     for sentence in sentences:
         try:
             links = search_pexels(sentence, num_results=num_videos)
@@ -52,56 +56,43 @@ def concatenate_pexels_videos(text, num_videos=5):
         except Exception as e:
             print(f"Error al buscar video para la frase '{sentence}': {e}")
             continue
-
     if not video_links:
         raise Exception("No se encontraron videos relevantes para el texto proporcionado.")
-
     video_clips = []
     for link in video_links:
         video_response = requests.get(link)
         with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmp_video:
             tmp_video.write(video_response.content)
             video_clips.append(VideoFileClip(tmp_video.name))
-
     # Concatenar videos
     final_clip = concatenate_videoclips(video_clips, method="compose")
     return final_clip
 
-# Combinar audio, video y música con fade out solo en el video y la música
+# Función para combinar audio, video y música
 def combine_audio_video(audio_file, video_clip, music_clip=None):
     audio_clip = AudioFileClip(audio_file)
-
-    # Duración total: speech + 5 segundos para fade out
     total_duration = audio_clip.duration + 5
 
-    # Extender la duración del video si es más corto que el audio + fade out
     if video_clip.duration < total_duration:
         video_clip = video_clip.loop(duration=total_duration)  # Repetir el video si es necesario
+    video_clip = video_clip.set_duration(total_duration).fadeout(5)  # Aplicar fade out al video
+    final_clip = video_clip.set_audio(audio_clip)  # Combinar audio y video
 
-    # Aplicar fade out solo al video
-    video_clip = video_clip.set_duration(total_duration).fadeout(5)
-
-    # Combinar audio y video
-    final_clip = video_clip.set_audio(audio_clip)
-
-    # Añadir música de fondo si aplica
     if music_clip:
-        # Extender la música para que coincida con la duración total
         if music_clip.duration < total_duration:
             repetitions = int(total_duration / music_clip.duration) + 1
             music_clips = [music_clip] * repetitions
             music_clip = concatenate_audioclips(music_clips)
         if music_clip.duration > total_duration:
             music_clip = music_clip.subclip(0, total_duration)
+        music_clip = music_clip.audio_fadeout(5)  # Aplicar fade out a la música
+        final_clip = final_clip.set_audio(CompositeAudioClip([audio_clip, music_clip]))  # Combinar audio y música
 
-        # Aplicar fade out a la música
-        music_clip = music_clip.audio_fadeout(5)
-
-        # Combinar audio principal, música y video
-        final_clip = final_clip.set_audio(CompositeAudioClip([audio_clip, music_clip]))
+    # Generar un nombre único para el video
+    output_filename = f"final_video_{int(time.time())}.mp4"
+    output_path = os.path.join(output_folder, output_filename)
 
     # Exportar el video final
-    output_path = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4").name
     final_clip.write_videofile(output_path, codec="libx264", audio_codec="aac", fps=24)
     return output_path
 
@@ -138,13 +129,44 @@ def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch):
         else:
             music_clip = None
 
-        # Combinar audio, video y música con fade out solo en el video y la música
-        final_video = combine_audio_video(audio_file, video_clip, music_clip)
-        return final_video
+        # Combinar audio, video y música
+        final_video_path = combine_audio_video(audio_file, video_clip, music_clip)
+
+        # Devolver la ruta del video generado
+        return final_video_path
 
     except Exception as e:
         return f"Error durante el procesamiento: {e}", None
 
+# Función para limpiar archivos antiguos
+def cleanup_old_files(folder_path, max_age_hours=1):
+    now = datetime.now()
+    threshold = now - timedelta(hours=max_age_hours)
+
+    for filename in os.listdir(folder_path):
+        file_path = os.path.join(folder_path, filename)
+        if os.path.isfile(file_path):
+            modification_time = datetime.fromtimestamp(os.path.getmtime(file_path))
+            if modification_time < threshold:
+                try:
+                    os.remove(file_path)
+                    print(f"Eliminado archivo antiguo: {file_path}")
+                except Exception as e:
+                    print(f"Error al eliminar {file_path}: {e}")
+
+# Iniciar hilo para limpieza automática
+def start_cleanup_thread(folder_path, interval_seconds=300, max_age_hours=1):
+    def cleanup_loop():
+        while True:
+            cleanup_old_files(folder_path, max_age_hours)
+            time.sleep(interval_seconds)
+
+    cleanup_thread = threading.Thread(target=cleanup_loop, daemon=True)
+    cleanup_thread.start()
+
+# Iniciar el hilo de limpieza al inicio del script
+start_cleanup_thread(output_folder, interval_seconds=300, max_age_hours=1)
+
 # Interfaz Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Text-to-Video Generator")
@@ -157,9 +179,8 @@ with gr.Blocks() as demo:
             voice_dropdown = gr.Dropdown(choices=list(voices.keys()), label="Select Voice")
             rate_slider = gr.Slider(minimum=-50, maximum=50, value=0, label="Speech Rate Adjustment (%)", step=1)
             pitch_slider = gr.Slider(minimum=-20, maximum=20, value=0, label="Pitch Adjustment (Hz)", step=1)
-
         with gr.Column():
-            output_video = gr.Video(label="Generated Video")
+            output_video = gr.File(label="Download Generated Video")  # Usar gr.File para permitir descargas
 
     btn = gr.Button("Generate Video")
     btn.click(
@@ -168,4 +189,5 @@ with gr.Blocks() as demo:
         outputs=output_video
     )
 
+# Lanzar la aplicación
 demo.launch(server_name="0.0.0.0", server_port=7860, share=True)