Update app.py

app.py

@@ -17,23 +17,19 @@ from google.oauth2 import service_account
 from googleapiclient.http import MediaFileUpload
 from io import BytesIO
 
-# Crear el archivo de credenciales de servicio desde los secretos
 service_account_info = json.loads(os.getenv('GOOGLE_SERVICE_ACCOUNT', '{}'))
 if service_account_info:
     with open('service-account.json', 'w') as f:
         json.dump(service_account_info, f)
 
-# Define la carpeta de salida y la carpeta temporal
 output_folder = "outputs"
 temp_dir = "temp_files"
 os.makedirs(output_folder, exist_ok=True)
 os.makedirs(temp_dir, exist_ok=True)
 
-# ID de la carpeta de destino en Google Drive
-FOLDER_ID = "12S6adpanAXjf71pKKGRRPqpzbJa5XEh3"  # Reemplaza con tu ID de carpeta
+FOLDER_ID = "12S6adpanAXjf71pKKGRRPqpzbJa5XEh3"
 
 def cleanup_temp_files():
-    """Elimina todos los archivos temporales de la carpeta temp_files."""
     for filename in os.listdir(temp_dir):
         file_path = os.path.join(temp_dir, filename)
         try:
@@ -43,43 +39,37 @@ def cleanup_temp_files():
             print(f"Error deleting {file_path}: {e}")
 
 def resize_and_blur_video(clip, target_width=1920, target_height=1080):
-    """Redimensiona el video al tamaño 1080p (16:9) y aplica desenfoque si es necesario."""
     try:
         w, h = clip.size
         current_aspect_ratio = w / h
         target_aspect_ratio = target_width / target_height
 
         if abs(current_aspect_ratio - target_aspect_ratio) < 0.1:
-            # Si la relación de aspecto ya es cercana a 16:9, solo redimensionamos
             return clip.resize((target_width, target_height))
 
-        # Crear un fondo borroso con las dimensiones objetivo
         background = ColorClip(size=(target_width, target_height), color=[0, 0, 0]).set_duration(clip.duration)
         try:
             background = background.fx(vfx.blur, sigma=50)
         except Exception as e:
             print(f"Error al aplicar blur: {e}")
 
-        # Redimensionar el video original para mantener su proporción
-        if current_aspect_ratio < target_aspect_ratio:  # Video vertical
+        if current_aspect_ratio < target_aspect_ratio:
             new_height = target_height
             new_width = int(new_height * current_aspect_ratio)
             x_center = (target_width - new_width) / 2
             resized_clip = clip.resize(width=new_width).set_position((x_center, 0))
-        else:  # Video horizontal
+        else:
             new_width = target_width
             new_height = int(new_width / current_aspect_ratio)
             y_center = (target_height - new_height) / 2
             resized_clip = clip.resize(height=new_height).set_position((0, y_center))
 
-        # Combinar el fondo borroso con el video redimensionado
         return CompositeVideoClip([background, resized_clip], size=(target_width, target_height))
     except Exception as e:
         print(f"Error en resize_and_blur_video: {e}")
         return clip
 
 def download_video(link):
-    """Descarga un video desde un enlace y lo guarda en la carpeta temporal."""
     try:
         video_response = requests.get(link)
         if video_response.status_code != 200:
@@ -93,14 +83,12 @@ def download_video(link):
         return None
 
 def concatenate_pixabay_videos(keywords, num_videos_per_keyword=1):
-    """Concatena videos de Pixabay basados en palabras clave."""
     keyword_list = [keyword.strip() for keyword in keywords.split(",") if keyword.strip()]
     if not keyword_list:
-        keyword_list = ["nature"]  # Palabra clave por defecto
+        keyword_list = ["nature"]
     video_clips = []
     for keyword in keyword_list:
         try:
-            print(f"Buscando videos para la palabra clave '{keyword}'...")
             links = search_pixabay(keyword, num_results=num_videos_per_keyword)
             if not links:
                 print(f"No se encontraron videos para '{keyword}', probando con 'nature'")
@@ -112,18 +100,16 @@ def concatenate_pixabay_videos(keywords, num_videos_per_keyword=1):
                 clip = VideoFileClip(temp_video_path)
                 processed_clip = resize_and_blur_video(clip)
                 video_clips.append(processed_clip)
-                os.remove(temp_video_path)  # Eliminar archivo temporal después de usarlo
+                os.remove(temp_video_path)
         except Exception as e:
             print(f"Error procesando palabra clave '{keyword}': {e}")
             continue
     if not video_clips:
-        # Si no hay videos, creamos un clip negro de 5 segundos
         return ColorClip(size=(1920, 1080), color=[0, 0, 0], duration=5)
     random.shuffle(video_clips)
     return concatenate_videoclips(video_clips, method="compose")
 
 def adjust_background_music(video_duration, music_file):
-    """Ajusta la música de fondo para que coincida con la duración del video."""
     try:
         music = AudioFileClip(music_file)
         if music.duration < video_duration:
@@ -137,67 +123,45 @@ def adjust_background_music(video_duration, music_file):
         return None
 
 def combine_audio_video(audio_file, video_clip, music_clip=None):
-    """Combina el audio y el video en un archivo final."""
     try:
         audio_clip = AudioFileClip(audio_file)
-        total_duration = audio_clip.duration + 2  # Añadimos 2 segundos extra
-        
-        # Aseguramos que el video tenga la duración correcta
+        total_duration = audio_clip.duration + 2
         video_clip = video_clip.loop(duration=total_duration)
         video_clip = video_clip.set_duration(total_duration).fadeout(2)
-        
-        # Combinamos el audio principal
         final_clip = video_clip.set_audio(audio_clip)
-        # Añadimos la música de fondo si existe
         if music_clip:
             music_clip = music_clip.set_duration(total_duration).audio_fadeout(2)
             final_clip = final_clip.set_audio(CompositeAudioClip([audio_clip, music_clip]))
-        # Generamos el nombre del archivo y la ruta
         output_filename = f"final_video_{int(time.time())}.mp4"
         output_path = os.path.join(output_folder, output_filename)
-        
-        # Guardamos el video
         final_clip.write_videofile(output_path, codec="libx264", audio_codec="aac", fps=24)
-        
-        # Limpiamos los clips
         final_clip.close()
         video_clip.close()
         audio_clip.close()
         if music_clip:
             music_clip.close()
-            
         return output_path
     except Exception as e:
         print(f"Error combinando audio y video: {e}")
-        if 'final_clip' in locals():
-            final_clip.close()
         return None
 
 def upload_to_google_drive(file_path, folder_id):
-    """Sube un archivo a Google Drive y devuelve el enlace público."""
     try:
-        # Autenticación
         creds = service_account.Credentials.from_service_account_file(
             'service-account.json', scopes=['https://www.googleapis.com/auth/drive']
         )
         service = build('drive', 'v3', credentials=creds)
-
-        # Subir el archivo
         file_metadata = {
             'name': os.path.basename(file_path),
             'parents': [folder_id]
         }
         media = MediaFileUpload(file_path, resumable=True)
         file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
-
-        # Hacer el archivo público
         permission = {
             'type': 'anyone',
             'role': 'reader'
         }
         service.permissions().create(fileId=file['id'], body=permission).execute()
-
-        # Generar el enlace de descarga
         file_id = file['id']
         download_link = f"https://drive.google.com/uc?export=download&id={file_id}"
         return download_link
@@ -206,35 +170,27 @@ def upload_to_google_drive(file_path, folder_id):
         return None
 
 def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch, keywords):
-    """Procesa la entrada del usuario y genera el video final."""
     try:
-        # Determinamos el texto a usar
         if text.strip():
             final_text = text
         elif txt_file is not None:
             final_text = txt_file.decode("utf-8")
         else:
             raise ValueError("No text input provided")
-        # Generamos el audio
         audio_file = asyncio.run(text_to_speech(final_text, selected_voice, rate, pitch))
         if not audio_file:
             raise ValueError("Failed to generate audio")
-        # Generamos el video
         video_clip = concatenate_pixabay_videos(keywords, num_videos_per_keyword=1)
         if not video_clip:
             raise ValueError("Failed to generate video")
-        # Procesamos la música de fondo si existe
         music_clip = None
         if mp3_file is not None:
             music_clip = adjust_background_music(video_clip.duration, mp3_file.name)
-        # Combinamos todo
         final_video_path = combine_audio_video(audio_file, video_clip, music_clip)
         if not final_video_path:
             raise ValueError("Failed to combine audio and video")
-        # Subimos a Google Drive y obtenemos el enlace
         download_link = upload_to_google_drive(final_video_path, folder_id=FOLDER_ID)
         if download_link:
-            print(f"Video subido a Google Drive. Enlace de descarga: {download_link}")
             return f"[Descargar video]({download_link})"
         else:
             raise ValueError("Error subiendo el video a Google Drive")
@@ -242,9 +198,8 @@ def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch, keyword
         print(f"Error durante el procesamiento: {e}")
         return None
     finally:
-        cleanup_temp_files()  # Limpiar archivos temporales al finalizar
+        cleanup_temp_files()
 
-# Interfaz Gradio
 with gr.Blocks() as demo:
     gr.Markdown("# Text-to-Video Generator")
     with gr.Row():
@@ -253,16 +208,15 @@ with gr.Blocks() as demo:
             txt_file_input = gr.File(label="Or upload a .txt file", file_types=[".txt"])
             mp3_file_input = gr.File(label="Upload background music (.mp3)", file_types=[".mp3"])
             keyword_input = gr.Textbox(
-                label="Enter keywords separated by commas (e.g., universe, galaxy, forest, cat)",
-                value="fear, religion, god, demons, aliens, possession, galaxy, mysterious, dystopian, astral, warfare, space, space, galaxy, moon, fear, astral, god, evil, mystery, cosmos, stars, paranormal, inexplicable, hidden, enigma, unknown, unusual, intriguing, curious, strange, supernatural, esoteric, arcane, occultism, supernatural, mystery, phenomenon, rare, unusual, enigmatic, sinister, gloomy, dark, shadowy, macabre, eerie, chilling, cursed, fantastic, unreal, unknown, mysterious, enigmatic, inexplicable, unusual, strange, unusual, arcane, esoteric, hidden, shadowy, dark, gloomy, sinister, macabre, eerie, chilling, cursed, fantastic, unreal, paranormal, supernatural, occultism, phenomenon, rare, intriguing, curious"
+                label="Enter keywords separated by commas",
+                value="fear,religion,god,demons,aliens,possession"
             )
             voices = asyncio.run(get_voices())
             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_link = gr.Markdown("")  # Mostrar el enlace de descarga
-
+            output_link = gr.Markdown("")
     btn = gr.Button("Generate Video")
     btn.click(
         process_input,
@@ -270,8 +224,5 @@ with gr.Blocks() as demo:
         outputs=output_link
     )
 
-# Leer el puerto asignado por Hugging Face
 port = int(os.getenv("PORT", 7860))
-
-# Lanzar la aplicación
 demo.launch(server_name="0.0.0.0", server_port=port, share=True, show_error=True)