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 |
|