Update app.py

app.py

@@ -3,7 +3,8 @@ from tts_module import get_voices, text_to_speech
 from pixabay_api import search_pixabay
 from moviepy.editor import (
     AudioFileClip, VideoFileClip, CompositeAudioClip,
-    concatenate_audioclips, concatenate_videoclips, vfx, CompositeVideoClip
+    concatenate_audioclips, concatenate_videoclips, vfx, CompositeVideoClip,
+    ColorClip
 )
 import asyncio
 import os
@@ -12,7 +13,8 @@ import requests
 import tempfile
 import re
 import random
-from google_drive_upload import upload_to_google_drive  # Importa la nueva función
+import numpy as np
+from google_drive_upload import upload_to_google_drive
 
 # Define la carpeta de salida
 output_folder = "outputs"
@@ -20,9 +22,9 @@ os.makedirs(output_folder, exist_ok=True)
 
 # Crear un video predeterminado si no existe
 if not os.path.exists(os.path.join(output_folder, "blank.mp4")):
-    from moviepy.editor import VideoClip
-    blank_clip = VideoClip(lambda t: None, duration=1)  # Video vacío de 1 segundo
-    blank_clip.write_videofile(os.path.join(output_folder, "blank.mp4"), fps=24)
+    # Create a black frame
+    color_clip = ColorClip(size=(1920, 1080), color=[0, 0, 0], duration=1)
+    color_clip.write_videofile(os.path.join(output_folder, "blank.mp4"), fps=24)
 
 def clean_text_for_search(text):
     """Limpia el texto para hacer búsquedas válidas en Pixabay"""
@@ -65,7 +67,9 @@ 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:
-        raise Exception("No se proporcionaron palabras clave válidas.")
+        # Si no hay palabras clave, crear un video negro
+        color_clip = ColorClip(size=(1920, 1080), color=[0, 0, 0], duration=5)
+        return color_clip
 
     video_clips = []
 
@@ -75,11 +79,11 @@ def concatenate_pixabay_videos(keywords, num_videos_per_keyword=1):
             links = search_pixabay(keyword, num_results=num_videos_per_keyword)
             if not links:
                 print(f"No se encontraron videos para la palabra clave '{keyword}'. Usando palabra clave predeterminada 'nature'.")
-                links = search_pixabay("nature", num_results=num_videos_per_keyword)  # Palabra clave predeterminada
+                links = search_pixabay("nature", num_results=num_videos_per_keyword)
                 if not links:
                     continue
 
-            link = links[0]  # Usamos solo el primer video encontrado
+            link = links[0]
             video_response = requests.get(link)
             if video_response.status_code != 200:
                 print(f"Error al descargar video desde {link}: Código de estado {video_response.status_code}")
@@ -96,7 +100,9 @@ def concatenate_pixabay_videos(keywords, num_videos_per_keyword=1):
             continue
 
     if not video_clips:
-        raise Exception("No se pudieron obtener videos válidos.")
+        # Si no se encontraron videos, crear un video negro
+        color_clip = ColorClip(size=(1920, 1080), color=[0, 0, 0], duration=5)
+        return color_clip
 
     random.shuffle(video_clips)
     return concatenate_videoclips(video_clips, method="compose")
@@ -144,7 +150,7 @@ def combine_audio_video(audio_file, video_clip, music_clip=None):
 
     except Exception as e:
         print(f"Error combinando audio y video: {e}")
-        return os.path.join(output_folder, "blank.mp4")  # Devuelve un video predeterminado en caso de error
+        return os.path.join(output_folder, "blank.mp4")
 
 def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch, keywords):
     """Procesa la entrada del usuario y genera el video final."""
@@ -154,23 +160,23 @@ def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch, keyword
         elif txt_file is not None:
             final_text = txt_file.decode("utf-8")
         else:
-            return os.path.join(output_folder, "blank.mp4")  # Devuelve un video predeterminado si no hay entrada
+            return os.path.join(output_folder, "blank.mp4")
 
         voices = asyncio.run(get_voices())
         if selected_voice not in voices:
-            return os.path.join(output_folder, "blank.mp4")  # Devuelve un video predeterminado si la voz no es válida
+            return os.path.join(output_folder, "blank.mp4")
 
         try:
             audio_file = asyncio.run(text_to_speech(final_text, selected_voice, rate, pitch))
         except Exception as e:
             print(f"Error generando audio: {e}")
-            return os.path.join(output_folder, "blank.mp4")  # Devuelve un video predeterminado en caso de error
+            return os.path.join(output_folder, "blank.mp4")
 
         try:
             video_clip = concatenate_pixabay_videos(keywords, num_videos_per_keyword=1)
         except Exception as e:
             print(f"Error concatenando videos: {e}")
-            return os.path.join(output_folder, "blank.mp4")  # Devuelve un video predeterminado en caso de error
+            return os.path.join(output_folder, "blank.mp4")
 
         if mp3_file is not None:
             music_clip = adjust_background_music(video_clip.duration, mp3_file.name)
@@ -178,12 +184,12 @@ def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch, keyword
             music_clip = None
 
         final_video_path = combine_audio_video(audio_file, video_clip, music_clip)
-        upload_to_google_drive(final_video_path)  # Sube a Google Drive
+        upload_to_google_drive(final_video_path)
         return final_video_path
 
     except Exception as e:
         print(f"Error durante el procesamiento: {e}")
-        return os.path.join(output_folder, "blank.mp4")  # Devuelve un video predeterminado en caso de error
+        return os.path.join(output_folder, "blank.mp4")
 
 # Interfaz Gradio
 with gr.Blocks() as demo:
@@ -193,7 +199,10 @@ with gr.Blocks() as demo:
             text_input = gr.Textbox(label="Write your text here", lines=5)
             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)")
+            keyword_input = gr.Textbox(
+                label="Enter keywords separated by commas (e.g., universe, galaxy, forest, cat)",
+                value="nature"  # Valor predeterminado
+            )
             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)
@@ -208,8 +217,8 @@ with gr.Blocks() as demo:
         outputs=output_video
     )
 
-# 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)
+if __name__ == "__main__":
+    # 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)