Update app.py

app.py

@@ -16,7 +16,8 @@ import re
 import random
 from google_drive_upload import authenticate_google_drive, upload_to_google_drive
 from io import BytesIO
-
+from PIL import Image
+import numpy as np
 
 # Crear el archivo de credenciales de servicio desde los secretos
 service_account_info = json.loads(os.getenv('GOOGLE_SERVICE_ACCOUNT', '{}'))
@@ -29,17 +30,12 @@ output_folder = "outputs"
 os.makedirs(output_folder, exist_ok=True)
 
 # ID de la carpeta de destino en Google Drive
-FOLDER_ID = "12S6adpanAXjf71pKKGRRPqpzbJa5XEh3"  # Reemplaza con tu ID de carpeta
+FOLDER_ID = "12S6adpanAXjf71pKKGRRPqpzbJa5XEh3"  # Reemplaza con tu ID real
+
 
 def resize_and_add_background(clip, target_width=1920, target_height=1080, background_url="https://wallpaperaccess.com/full/231401.jpg"):
     """
     Redimensiona el video al tamaño 1080p (16:9) y añade una imagen de fondo descargada desde una URL.
-
-    :param clip: VideoFileClip original.
-    :param target_width: Ancho objetivo del video final (por defecto: 1920).
-    :param target_height: Alto objetivo del video final (por defecto: 1080).
-    :param background_url: URL de la imagen de fondo.
-    :return: VideoFileClip procesado con fondo agregado.
     """
     try:
         w, h = clip.size
@@ -59,9 +55,10 @@ def resize_and_add_background(clip, target_width=1920, target_height=1080, backg
                 .resize((target_width, target_height))
             )
         else:
-            image_data = BytesIO(response.content)
+            image_data = Image.open(BytesIO(response.content))
+            image_array = np.array(image_data)
             background = (
-                ImageClip(image_data)
+                ImageClip(image_array)
                 .set_duration(clip.duration)
                 .resize((target_width, target_height))
             )
@@ -112,7 +109,7 @@ def concatenate_pixabay_videos(keywords, num_videos_per_keyword=1):
             with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmp_video:
                 tmp_video.write(video_response.content)
                 clip = VideoFileClip(tmp_video.name)
-                processed_clip = resize_and_add_background(clip)  # Usa la nueva función
+                processed_clip = resize_and_add_background(clip)
                 video_clips.append(processed_clip)
                 os.unlink(tmp_video.name)  # Limpiamos el archivo temporal
         except Exception as e:
@@ -188,33 +185,40 @@ def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch, keyword
             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
         video_id = upload_to_google_drive(final_video_path, folder_id=FOLDER_ID)
         if video_id:
             print(f"Video subido a Google Drive con ID: {video_id}")
+            download_link = f"https://drive.google.com/file/d/{video_id}/view?usp=sharing"
+            return download_link  # Devuelve el enlace compartido
         else:
             print("Error subiendo el video a Google Drive")
-        
-        return final_video_path
+            return "Error al subir el video"
+
     except Exception as e:
         print(f"Error durante el procesamiento: {e}")
-        return None
+        return f"Error durante el procesamiento: {e}"
 
 
 # Interfaz Gradio
@@ -227,7 +231,7 @@ with gr.Blocks() as demo:
             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="space, espacio,galaxy,moon,fear,astral,god, evil, mistery, cosmos, stars"
+                value="space, espacio,galaxy,moon,fear,astral,god,evil,mistery,cosmos,stars"
             )
             voices = asyncio.run(get_voices())
             voice_dropdown = gr.Dropdown(choices=list(voices.keys()), label="Select Voice")
@@ -235,7 +239,7 @@ with gr.Blocks() as demo:
             pitch_slider = gr.Slider(minimum=-20, maximum=20, value=0, label="Pitch Adjustment (Hz)", step=1)
 
         with gr.Column():
-            output_video = gr.Textbox(label="Download Link")  # Usar gr.Textbox para mostrar el enlace
+            output_video = gr.Textbox(label="Download Link")  # Mostramos el enlace de descarga
 
     btn = gr.Button("Generate Video")
     btn.click(