Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,13 +7,16 @@ import edge_tts
|
|
| 7 |
import gradio as gr
|
| 8 |
from pydub import AudioSegment
|
| 9 |
|
|
|
|
| 10 |
logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
|
| 11 |
|
|
|
|
| 12 |
INTRO_VIDEO = "introvideo.mp4"
|
| 13 |
OUTRO_VIDEO = "outrovideo.mp4"
|
| 14 |
MUSIC_BG = "musicafondo.mp3"
|
| 15 |
EJEMPLO_VIDEO = "ejemplo.mp4"
|
| 16 |
|
|
|
|
| 17 |
for file in [INTRO_VIDEO, OUTRO_VIDEO, MUSIC_BG, EJEMPLO_VIDEO]:
|
| 18 |
if not os.path.exists(file):
|
| 19 |
logging.error(f"Falta archivo necesario: {file}")
|
|
@@ -33,15 +36,39 @@ def eliminar_archivo_tiempo(ruta, delay=1800):
|
|
| 33 |
async def procesar_audio(texto, voz, duracion_video, audio_original):
|
| 34 |
temp_files = []
|
| 35 |
try:
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
tts_audio = AudioFileClip(tmp_tts.name)
|
| 40 |
-
temp_files.append(tmp_tts.name)
|
| 41 |
-
|
| 42 |
-
if tts_audio.duration > duracion_video:
|
| 43 |
-
tts_audio = tts_audio.subclip(0, duracion_video)
|
| 44 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 45 |
bg_music = AudioSegment.from_mp3(MUSIC_BG)
|
| 46 |
needed_ms = int(duracion_video * 1000)
|
| 47 |
repeticiones = needed_ms // len(bg_music) + 1
|
|
@@ -53,10 +80,11 @@ async def procesar_audio(texto, voz, duracion_video, audio_original):
|
|
| 53 |
bg_audio = AudioFileClip(tmp_bg.name).volumex(0.15)
|
| 54 |
temp_files.append(tmp_bg.name)
|
| 55 |
|
|
|
|
| 56 |
audios = [bg_audio.set_duration(duracion_video)]
|
| 57 |
if audio_original:
|
| 58 |
-
audios.append(audio_original.volumex(0.7))
|
| 59 |
-
audios.append(
|
| 60 |
|
| 61 |
audio_final = CompositeAudioClip(audios).set_duration(duracion_video)
|
| 62 |
return audio_final
|
|
@@ -73,13 +101,16 @@ async def procesar_audio(texto, voz, duracion_video, audio_original):
|
|
| 73 |
|
| 74 |
async def procesar_video(video_input, texto_tts, voz_seleccionada):
|
| 75 |
try:
|
|
|
|
| 76 |
intro = VideoFileClip(INTRO_VIDEO)
|
| 77 |
outro = VideoFileClip(OUTRO_VIDEO)
|
| 78 |
video_original = VideoFileClip(video_input)
|
| 79 |
audio_original = video_original.audio
|
| 80 |
|
|
|
|
| 81 |
duracion_video = video_original.duration
|
| 82 |
|
|
|
|
| 83 |
audio_final = await procesar_audio(
|
| 84 |
texto_tts,
|
| 85 |
voz_seleccionada,
|
|
@@ -87,28 +118,54 @@ async def procesar_video(video_input, texto_tts, voz_seleccionada):
|
|
| 87 |
audio_original
|
| 88 |
)
|
| 89 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 90 |
target_width = 1920
|
| 91 |
target_height = 1080
|
| 92 |
-
video_resized =
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 93 |
|
| 94 |
-
|
|
|
|
| 95 |
|
|
|
|
| 96 |
video_final = concatenate_videoclips(
|
| 97 |
[intro, video_con_audio, outro],
|
| 98 |
-
method="compose",
|
| 99 |
-
padding=0
|
| 100 |
)
|
| 101 |
|
|
|
|
| 102 |
with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmp:
|
| 103 |
video_final.write_videofile(
|
| 104 |
tmp.name,
|
| 105 |
codec="libx264",
|
| 106 |
audio_codec="aac",
|
| 107 |
-
fps=video_original.fps,
|
| 108 |
threads=4,
|
| 109 |
ffmpeg_params=[
|
| 110 |
-
"-aspect", "16:9",
|
| 111 |
-
"-vf", "scale=1920:1080"
|
| 112 |
],
|
| 113 |
verbose=False
|
| 114 |
)
|
|
@@ -118,6 +175,7 @@ async def procesar_video(video_input, texto_tts, voz_seleccionada):
|
|
| 118 |
logging.error(f" fallo general: {str(e)}")
|
| 119 |
raise
|
| 120 |
|
|
|
|
| 121 |
with gr.Blocks() as demo:
|
| 122 |
gr.Markdown("# Editor de Video con IA")
|
| 123 |
|
|
|
|
| 7 |
import gradio as gr
|
| 8 |
from pydub import AudioSegment
|
| 9 |
|
| 10 |
+
# Configuraci贸n de Logs
|
| 11 |
logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
|
| 12 |
|
| 13 |
+
# CONSTANTES DE ARCHIVOS
|
| 14 |
INTRO_VIDEO = "introvideo.mp4"
|
| 15 |
OUTRO_VIDEO = "outrovideo.mp4"
|
| 16 |
MUSIC_BG = "musicafondo.mp3"
|
| 17 |
EJEMPLO_VIDEO = "ejemplo.mp4"
|
| 18 |
|
| 19 |
+
# Validar existencia de archivos
|
| 20 |
for file in [INTRO_VIDEO, OUTRO_VIDEO, MUSIC_BG, EJEMPLO_VIDEO]:
|
| 21 |
if not os.path.exists(file):
|
| 22 |
logging.error(f"Falta archivo necesario: {file}")
|
|
|
|
| 36 |
async def procesar_audio(texto, voz, duracion_video, audio_original):
|
| 37 |
temp_files = []
|
| 38 |
try:
|
| 39 |
+
# Validar texto
|
| 40 |
+
if not texto.strip():
|
| 41 |
+
raise ValueError("El texto para TTS no puede estar vac铆o.")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
|
| 43 |
+
# Dividir el texto en fragmentos si es demasiado largo
|
| 44 |
+
def dividir_texto(texto, max_length=3000):
|
| 45 |
+
return [texto[i:i + max_length] for i in range(0, len(texto), max_length)]
|
| 46 |
+
|
| 47 |
+
fragmentos = dividir_texto(texto)
|
| 48 |
+
audios_tts = []
|
| 49 |
+
|
| 50 |
+
for fragmento in fragmentos:
|
| 51 |
+
# Generar TTS
|
| 52 |
+
communicate = edge_tts.Communicate(fragmento, voz)
|
| 53 |
+
with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_tts:
|
| 54 |
+
try:
|
| 55 |
+
await communicate.save(tmp_tts.name)
|
| 56 |
+
except edge_tts.exceptions.NoAudioReceived as e:
|
| 57 |
+
logging.error(f"Error en TTS: {str(e)}")
|
| 58 |
+
raise ValueError("No se pudo generar el audio. Verifica tu conexi贸n o los par谩metros del TTS.")
|
| 59 |
+
|
| 60 |
+
tts_audio = AudioFileClip(tmp_tts.name)
|
| 61 |
+
temp_files.append(tmp_tts.name)
|
| 62 |
+
audios_tts.append(tts_audio)
|
| 63 |
+
|
| 64 |
+
# Combinar todos los fragmentos de TTS
|
| 65 |
+
tts_audio_final = concatenate_audioclips(audios_tts)
|
| 66 |
+
|
| 67 |
+
# Limitar TTS al video
|
| 68 |
+
if tts_audio_final.duration > duracion_video:
|
| 69 |
+
tts_audio_final = tts_audio_final.subclip(0, duracion_video)
|
| 70 |
+
|
| 71 |
+
# Preparar m煤sica de fondo en loop
|
| 72 |
bg_music = AudioSegment.from_mp3(MUSIC_BG)
|
| 73 |
needed_ms = int(duracion_video * 1000)
|
| 74 |
repeticiones = needed_ms // len(bg_music) + 1
|
|
|
|
| 80 |
bg_audio = AudioFileClip(tmp_bg.name).volumex(0.15)
|
| 81 |
temp_files.append(tmp_bg.name)
|
| 82 |
|
| 83 |
+
# Combinar audios
|
| 84 |
audios = [bg_audio.set_duration(duracion_video)]
|
| 85 |
if audio_original:
|
| 86 |
+
audios.append(audio_original.volumex(0.7)) # Audio original al 70%
|
| 87 |
+
audios.append(tts_audio_final.volumex(0.85).set_start(0)) # TTS al 85%
|
| 88 |
|
| 89 |
audio_final = CompositeAudioClip(audios).set_duration(duracion_video)
|
| 90 |
return audio_final
|
|
|
|
| 101 |
|
| 102 |
async def procesar_video(video_input, texto_tts, voz_seleccionada):
|
| 103 |
try:
|
| 104 |
+
# Cargar componentes
|
| 105 |
intro = VideoFileClip(INTRO_VIDEO)
|
| 106 |
outro = VideoFileClip(OUTRO_VIDEO)
|
| 107 |
video_original = VideoFileClip(video_input)
|
| 108 |
audio_original = video_original.audio
|
| 109 |
|
| 110 |
+
# Duraci贸n del video editado (sin intro/outro)
|
| 111 |
duracion_video = video_original.duration
|
| 112 |
|
| 113 |
+
# Procesar audio
|
| 114 |
audio_final = await procesar_audio(
|
| 115 |
texto_tts,
|
| 116 |
voz_seleccionada,
|
|
|
|
| 118 |
audio_original
|
| 119 |
)
|
| 120 |
|
| 121 |
+
# Redimensionar temporalmente el video a un tama帽o muy grande (4000px de ancho)
|
| 122 |
+
video_temporal = video_original.resize(width=4000)
|
| 123 |
+
|
| 124 |
+
# Redimensionar nuevamente al tama帽o final deseado (1920x1080)
|
| 125 |
target_width = 1920
|
| 126 |
target_height = 1080
|
| 127 |
+
video_resized = video_temporal.resize(height=target_height)
|
| 128 |
+
|
| 129 |
+
# Manejar m谩rgenes si es necesario
|
| 130 |
+
current_width = video_resized.w
|
| 131 |
+
current_height = video_resized.h
|
| 132 |
+
if current_width < target_width:
|
| 133 |
+
# Agregar m谩rgenes laterales
|
| 134 |
+
margin_left = (target_width - current_width) // 2
|
| 135 |
+
margin_right = target_width - current_width - margin_left
|
| 136 |
+
video_final = video_resized.margin(
|
| 137 |
+
left=margin_left,
|
| 138 |
+
right=margin_right,
|
| 139 |
+
color=(0, 0, 0) # Negro
|
| 140 |
+
)
|
| 141 |
+
else:
|
| 142 |
+
# Recortar horizontalmente si es m谩s ancho que 1920
|
| 143 |
+
video_final = video_resized.crop(
|
| 144 |
+
x1=(current_width - target_width) // 2,
|
| 145 |
+
x2=current_width - (current_width - target_width) // 2
|
| 146 |
+
)
|
| 147 |
|
| 148 |
+
# Combinar video con audio
|
| 149 |
+
video_con_audio = video_final.set_audio(audio_final)
|
| 150 |
|
| 151 |
+
# Concatenar intro + video + outro SIN alteraciones
|
| 152 |
video_final = concatenate_videoclips(
|
| 153 |
[intro, video_con_audio, outro],
|
| 154 |
+
method="compose", # Evitar problemas de grid
|
| 155 |
+
padding=0 # Sin espacio entre clips
|
| 156 |
)
|
| 157 |
|
| 158 |
+
# Renderizar video final con metadatos correctos
|
| 159 |
with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmp:
|
| 160 |
video_final.write_videofile(
|
| 161 |
tmp.name,
|
| 162 |
codec="libx264",
|
| 163 |
audio_codec="aac",
|
| 164 |
+
fps=video_original.fps, # Mantener FPS original
|
| 165 |
threads=4,
|
| 166 |
ffmpeg_params=[
|
| 167 |
+
"-aspect", "16:9", # Forzar relaci贸n de aspecto
|
| 168 |
+
"-vf", "scale=1920:1080" # Forzar escalado expl铆cito
|
| 169 |
],
|
| 170 |
verbose=False
|
| 171 |
)
|
|
|
|
| 175 |
logging.error(f" fallo general: {str(e)}")
|
| 176 |
raise
|
| 177 |
|
| 178 |
+
# Interfaz Gradio
|
| 179 |
with gr.Blocks() as demo:
|
| 180 |
gr.Markdown("# Editor de Video con IA")
|
| 181 |
|