gnosticdev commited on
Commit
00d8b3c
verified
1 Parent(s): f031dba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +75 -17
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
- communicate = edge_tts.Communicate(texto, voz)
37
- with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_tts:
38
- await communicate.save(tmp_tts.name)
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(tts_audio.volumex(0.85).set_start(0))
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 = video_original.resize((target_width, target_height))
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
93
 
94
- video_con_audio = video_resized.set_audio(audio_final)
 
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