Update app.py

app.py

@@ -34,15 +34,19 @@ def upload_to_google_drive(file_path):
 
 # Ajustar música de fondo
 def adjust_background_music(video_duration, music_file):
-    music = AudioFileClip(music_file)
-    if music.duration < video_duration:
-        repetitions = int(video_duration / music.duration) + 1
-        music_clips = [music] * repetitions
-        music = concatenate_audioclips(music_clips)
-    if music.duration > video_duration:
-        music = music.subclip(0, video_duration)
-    music = music.volumex(0.2)
-    return music
+    try:
+        music = AudioFileClip(music_file)
+        if music.duration < video_duration:
+            repetitions = int(video_duration / music.duration) + 1
+            music_clips = [music] * repetitions
+            music = concatenate_audioclips(music_clips)
+        if music.duration > video_duration:
+            music = music.subclip(0, video_duration)
+        music = music.volumex(0.2)
+        return music
+    except Exception as e:
+        print(f"Error al ajustar música de fondo: {e}")
+        return None
 
 # Concatenar múltiples videos de Pexels
 def concatenate_pexels_videos(text, num_videos=5, target_width=1920, target_height=1080):
@@ -61,7 +65,7 @@ def concatenate_pexels_videos(text, num_videos=5, target_width=1920, target_heig
             print(f"Enlace encontrado para '{sentence}': {link}")
 
             # Descargar el video desde Pexels
-            video_response = requests.get(link)
+            video_response = requests.get(link, stream=True)
             if video_response.status_code != 200:
                 print(f"Error al descargar video desde {link}: Código de estado {video_response.status_code}")
                 continue
@@ -100,27 +104,32 @@ def concatenate_pexels_videos(text, num_videos=5, target_width=1920, target_heig
 
 # Combinar audio, video y música
 def combine_audio_video(audio_file, video_clip, music_clip=None):
-    audio_clip = AudioFileClip(audio_file)
-    total_duration = audio_clip.duration + 5
-    if video_clip.duration < total_duration:
-        video_clip = video_clip.loop(duration=total_duration)
-    video_clip = video_clip.set_duration(total_duration).fadeout(5)
-    final_clip = video_clip.set_audio(audio_clip)
-
-    if music_clip:
-        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)
-        final_clip = final_clip.set_audio(CompositeAudioClip([audio_clip, music_clip]))
-
-    output_filename = f"final_video_{int(time.time())}.mp4"
-    output_path = os.path.join(output_folder, output_filename)
-    final_clip.write_videofile(output_path, codec="libx264", audio_codec="aac", fps=24)
-    return output_path
+    try:
+        audio_clip = AudioFileClip(audio_file)
+        total_duration = audio_clip.duration + 5
+        if video_clip.duration < total_duration:
+            video_clip = video_clip.loop(duration=total_duration)
+        video_clip = video_clip.set_duration(total_duration).fadeout(5)
+        final_clip = video_clip.set_audio(audio_clip)
+
+        if music_clip:
+            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)
+            final_clip = final_clip.set_audio(CompositeAudioClip([audio_clip, music_clip]))
+
+        output_filename = f"final_video_{int(time.time())}.mp4"
+        output_path = os.path.join(output_folder, output_filename)
+        final_clip.write_videofile(output_path, codec="libx264", audio_codec="aac", fps=24)
+        return output_path
+
+    except Exception as e:
+        print(f"Error al combinar audio, video y música: {e}")
+        return None
 
 # Función principal
 def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch):
@@ -130,34 +139,35 @@ def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch):
         elif txt_file is not None:
             final_text = txt_file.decode("utf-8")
         else:
-            return "No input provided"
+            return None
 
         voices = asyncio.run(get_voices())
         if selected_voice not in voices:
-            return f"La voz '{selected_voice}' no es válida. Por favor, seleccione una de las siguientes voces: {', '.join(voices.keys())}"
+            return None
 
-        try:
-            audio_file = asyncio.run(text_to_speech(final_text, selected_voice, rate, pitch))
-        except Exception as e:
-            return f"Error con la voz seleccionada: {e}"
+        # Generar audio con TTS
+        audio_file = asyncio.run(text_to_speech(final_text, selected_voice, rate, pitch))
 
-        try:
-            video_clip = concatenate_pexels_videos(final_text, num_videos=5)
-        except Exception as e:
-            return f"Error al buscar videos en Pexels: {e}"
+        # Concatenar videos de Pexels
+        video_clip = concatenate_pexels_videos(final_text, num_videos=5)
 
+        # Ajustar música de fondo
         if mp3_file is not None:
             music_clip = adjust_background_music(video_clip.duration, mp3_file.name)
         else:
             music_clip = None
 
+        # Combinar audio, video y música
         final_video_path = combine_audio_video(audio_file, video_clip, music_clip)
-        upload_to_google_drive(final_video_path)  # Sube el video final a Google Drive
+
+        # Subir el video final 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 f"Error durante el procesamiento: {e}"
+        return None
 
 # Interfaz Gradio
 with gr.Blocks() as demo: