Update app.py
Browse files
app.py
CHANGED
@@ -25,160 +25,98 @@ def clean_text_for_search(text):
|
|
25 |
text = re.sub(r'[^\w\s]', '', text).strip()
|
26 |
return text
|
27 |
|
28 |
-
def resize_and_blur_video(clip,
|
29 |
"""
|
30 |
-
Redimensiona y aplica desenfoque al video para mantener aspecto 16:9
|
31 |
-
Los videos más pequeños se redimensionan y los verticales se convierten en horizontales con blur.
|
32 |
-
|
33 |
-
Args:
|
34 |
-
clip: VideoFileClip object
|
35 |
-
target_width: Ancho objetivo (default 1920 para 1080p)
|
36 |
-
target_height: Alto objetivo (default 1080 para 1080p)
|
37 |
"""
|
38 |
try:
|
39 |
w, h = clip.size
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
foreground = clip.resize(width=new_width, height=new_height)
|
62 |
-
|
63 |
-
# Centrar el video
|
64 |
-
x_center = (target_width - new_width) // 2
|
65 |
-
y_center = (target_height - new_height) // 2
|
66 |
-
|
67 |
-
final = CompositeVideoClip(
|
68 |
-
[background,
|
69 |
-
foreground.set_position((x_center, y_center))],
|
70 |
-
size=(target_width, target_height)
|
71 |
)
|
72 |
-
|
73 |
-
return final
|
74 |
-
|
75 |
else: # Video horizontal
|
76 |
-
|
77 |
-
return clip.resize(width=target_width, height=target_height)
|
78 |
-
|
79 |
-
except Exception as e:
|
80 |
-
print(f"Error en resize_and_blur_video: {e}")
|
81 |
-
return clip.resize(width=target_width, height=target_height)
|
82 |
|
83 |
except Exception as e:
|
84 |
print(f"Error en resize_and_blur_video: {e}")
|
85 |
return clip
|
86 |
|
87 |
-
def concatenate_pexels_videos(keywords, num_videos_per_keyword=1
|
88 |
"""
|
89 |
-
Concatena videos de Pexels
|
90 |
-
|
91 |
-
Args:
|
92 |
-
keywords (str): Palabras clave separadas por comas
|
93 |
-
num_videos_per_keyword (int): Número de videos por palabra clave
|
94 |
-
target_width (int): Ancho objetivo para los videos
|
95 |
-
target_height (int): Alto objetivo para los videos
|
96 |
"""
|
97 |
keyword_list = [keyword.strip() for keyword in keywords.split(",") if keyword.strip()]
|
98 |
if not keyword_list:
|
99 |
-
raise
|
100 |
|
101 |
video_clips = []
|
102 |
-
processed_keywords = []
|
103 |
|
104 |
for keyword in keyword_list:
|
105 |
try:
|
106 |
-
print(f"Buscando videos para
|
107 |
-
|
108 |
-
clean_keyword = clean_text_for_search(keyword)
|
109 |
-
|
110 |
-
# Obtener videos con la nueva API
|
111 |
-
links = search_pexels(clean_keyword, num_results=num_videos_per_keyword)
|
112 |
if not links:
|
113 |
-
print(f"No se encontraron videos para
|
114 |
continue
|
115 |
|
116 |
-
|
117 |
-
|
118 |
-
|
119 |
-
|
120 |
-
|
121 |
-
|
122 |
-
|
123 |
-
|
124 |
-
|
125 |
-
|
126 |
-
|
127 |
-
if chunk:
|
128 |
-
tmp_video.write(chunk)
|
129 |
-
|
130 |
-
tmp_video.flush()
|
131 |
-
|
132 |
-
# Procesar el video
|
133 |
-
try:
|
134 |
-
clip = VideoFileClip(tmp_video.name)
|
135 |
-
|
136 |
-
# Verificar duración mínima
|
137 |
-
if clip.duration < 3:
|
138 |
-
print(f"Video demasiado corto ({clip.duration}s), saltando...")
|
139 |
-
clip.close()
|
140 |
-
continue
|
141 |
-
|
142 |
-
# Procesar y agregar el clip
|
143 |
-
processed_clip = resize_and_blur_video(clip, target_width, target_height)
|
144 |
-
if processed_clip:
|
145 |
-
video_clips.append(processed_clip)
|
146 |
-
processed_keywords.append(keyword)
|
147 |
-
print(f"Video procesado exitosamente para: '{keyword}'")
|
148 |
-
|
149 |
-
except Exception as e:
|
150 |
-
print(f"Error procesando video: {e}")
|
151 |
-
if 'clip' in locals():
|
152 |
-
clip.close()
|
153 |
-
continue
|
154 |
-
|
155 |
-
finally:
|
156 |
-
# Limpiar archivo temporal
|
157 |
-
if os.path.exists(tmp_video.name):
|
158 |
-
try:
|
159 |
-
os.unlink(tmp_video.name)
|
160 |
-
except Exception as e:
|
161 |
-
print(f"Error eliminando archivo temporal: {e}")
|
162 |
|
163 |
except Exception as e:
|
164 |
print(f"Error procesando palabra clave '{keyword}': {e}")
|
165 |
continue
|
166 |
|
167 |
if not video_clips:
|
168 |
-
raise Exception("No se pudieron obtener videos válidos
|
169 |
|
170 |
-
|
171 |
-
|
172 |
-
# Aleatorizar el orden de los clips
|
173 |
random.shuffle(video_clips)
|
174 |
-
|
175 |
-
|
|
|
|
|
176 |
try:
|
177 |
-
|
178 |
-
|
179 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
180 |
except Exception as e:
|
181 |
-
|
|
|
182 |
|
183 |
def combine_audio_video(audio_file, video_clip, music_clip=None):
|
184 |
try:
|
@@ -210,57 +148,38 @@ def combine_audio_video(audio_file, video_clip, music_clip=None):
|
|
210 |
|
211 |
def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch, keywords):
|
212 |
try:
|
213 |
-
# Validar entrada de texto
|
214 |
if text.strip():
|
215 |
final_text = text
|
216 |
elif txt_file is not None:
|
217 |
final_text = txt_file.decode("utf-8")
|
218 |
else:
|
219 |
-
|
220 |
-
return None
|
221 |
|
222 |
-
# Validar voces
|
223 |
voices = asyncio.run(get_voices())
|
224 |
if selected_voice not in voices:
|
225 |
-
return
|
226 |
|
227 |
-
# Generar audio
|
228 |
try:
|
229 |
audio_file = asyncio.run(text_to_speech(final_text, selected_voice, rate, pitch))
|
230 |
except Exception as e:
|
231 |
-
|
232 |
-
return None
|
233 |
|
234 |
-
# Procesar videos
|
235 |
try:
|
236 |
video_clip = concatenate_pexels_videos(keywords, num_videos_per_keyword=1)
|
237 |
except Exception as e:
|
238 |
-
|
239 |
-
return None
|
240 |
|
241 |
-
# Procesar música de fondo si existe
|
242 |
if mp3_file is not None:
|
243 |
music_clip = adjust_background_music(video_clip.duration, mp3_file.name)
|
244 |
else:
|
245 |
music_clip = None
|
246 |
|
247 |
-
# Combinar audio y video
|
248 |
final_video_path = combine_audio_video(audio_file, video_clip, music_clip)
|
249 |
-
if final_video_path is None:
|
250 |
-
return None
|
251 |
-
|
252 |
-
# Subir a Google Drive
|
253 |
upload_to_google_drive(final_video_path)
|
254 |
-
|
255 |
-
# Verificar que el archivo existe antes de retornarlo
|
256 |
-
if os.path.exists(final_video_path):
|
257 |
-
return final_video_path
|
258 |
-
else:
|
259 |
-
return None
|
260 |
|
261 |
except Exception as e:
|
262 |
-
|
263 |
-
return None
|
264 |
|
265 |
def upload_to_google_drive(file_path):
|
266 |
try:
|
|
|
25 |
text = re.sub(r'[^\w\s]', '', text).strip()
|
26 |
return text
|
27 |
|
28 |
+
def resize_and_blur_video(clip, target_aspect_ratio=16/9):
|
29 |
"""
|
30 |
+
Redimensiona y aplica desenfoque al fondo del video para mantener el aspecto 16:9.
|
|
|
|
|
|
|
|
|
|
|
|
|
31 |
"""
|
32 |
try:
|
33 |
w, h = clip.size
|
34 |
+
current_aspect_ratio = w / h
|
35 |
+
|
36 |
+
print(f"Procesando video: {w}x{h}, ratio: {current_aspect_ratio}")
|
37 |
+
if abs(current_aspect_ratio - target_aspect_ratio) < 0.1:
|
38 |
+
return clip
|
39 |
+
|
40 |
+
if current_aspect_ratio < target_aspect_ratio: # Video vertical
|
41 |
+
target_w = int(h * target_aspect_ratio)
|
42 |
+
target_h = h
|
43 |
+
|
44 |
+
background = clip.resize(width=target_w)
|
45 |
+
try:
|
46 |
+
background = background.fx(vfx.blur, sigma=50)
|
47 |
+
except Exception as e:
|
48 |
+
print(f"Error al aplicar blur: {e}")
|
49 |
+
|
50 |
+
foreground = clip.resize(height=target_h)
|
51 |
+
x_center = (target_w - foreground.w) / 2
|
52 |
+
return CompositeVideoClip(
|
53 |
+
[background, foreground.set_position((x_center, 0))],
|
54 |
+
size=(target_w, target_h)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
55 |
)
|
|
|
|
|
|
|
56 |
else: # Video horizontal
|
57 |
+
return clip.resize(width=int(h * target_aspect_ratio), height=h)
|
|
|
|
|
|
|
|
|
|
|
58 |
|
59 |
except Exception as e:
|
60 |
print(f"Error en resize_and_blur_video: {e}")
|
61 |
return clip
|
62 |
|
63 |
+
def concatenate_pexels_videos(keywords, num_videos_per_keyword=1):
|
64 |
"""
|
65 |
+
Concatena videos de Pexels basados en palabras clave proporcionadas por el usuario.
|
66 |
+
:param keywords: Palabras clave separadas por comas (ejemplo: "universo, galaxia, bosque, gato").
|
|
|
|
|
|
|
|
|
|
|
67 |
"""
|
68 |
keyword_list = [keyword.strip() for keyword in keywords.split(",") if keyword.strip()]
|
69 |
if not keyword_list:
|
70 |
+
raise Exception("No se proporcionaron palabras clave válidas.")
|
71 |
|
72 |
video_clips = []
|
|
|
73 |
|
74 |
for keyword in keyword_list:
|
75 |
try:
|
76 |
+
print(f"Buscando videos para la palabra clave '{keyword}'...")
|
77 |
+
links = search_pexels(keyword, num_results=num_videos_per_keyword)
|
|
|
|
|
|
|
|
|
78 |
if not links:
|
79 |
+
print(f"No se encontraron videos para la palabra clave '{keyword}'.")
|
80 |
continue
|
81 |
|
82 |
+
link = links[0] # Usamos solo el primer video encontrado
|
83 |
+
video_response = requests.get(link)
|
84 |
+
if video_response.status_code != 200:
|
85 |
+
print(f"Error al descargar video desde {link}: Código de estado {video_response.status_code}")
|
86 |
+
continue
|
87 |
+
|
88 |
+
with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmp_video:
|
89 |
+
tmp_video.write(video_response.content)
|
90 |
+
clip = VideoFileClip(tmp_video.name)
|
91 |
+
processed_clip = resize_and_blur_video(clip)
|
92 |
+
video_clips.append(processed_clip)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
93 |
|
94 |
except Exception as e:
|
95 |
print(f"Error procesando palabra clave '{keyword}': {e}")
|
96 |
continue
|
97 |
|
98 |
if not video_clips:
|
99 |
+
raise Exception("No se pudieron obtener videos válidos.")
|
100 |
|
101 |
+
# Aleatorizar el orden de los clips si es necesario
|
|
|
|
|
102 |
random.shuffle(video_clips)
|
103 |
+
|
104 |
+
return concatenate_videoclips(video_clips, method="compose")
|
105 |
+
|
106 |
+
def adjust_background_music(video_duration, music_file):
|
107 |
try:
|
108 |
+
music = AudioFileClip(music_file)
|
109 |
+
if music.duration < video_duration:
|
110 |
+
repetitions = int(video_duration / music.duration) + 1
|
111 |
+
music_clips = [music] * repetitions
|
112 |
+
music = concatenate_audioclips(music_clips)
|
113 |
+
if music.duration > video_duration:
|
114 |
+
music = music.subclip(0, video_duration)
|
115 |
+
music = music.volumex(0.2)
|
116 |
+
return music
|
117 |
except Exception as e:
|
118 |
+
print(f"Error ajustando música: {e}")
|
119 |
+
return None
|
120 |
|
121 |
def combine_audio_video(audio_file, video_clip, music_clip=None):
|
122 |
try:
|
|
|
148 |
|
149 |
def process_input(text, txt_file, mp3_file, selected_voice, rate, pitch, keywords):
|
150 |
try:
|
|
|
151 |
if text.strip():
|
152 |
final_text = text
|
153 |
elif txt_file is not None:
|
154 |
final_text = txt_file.decode("utf-8")
|
155 |
else:
|
156 |
+
return "No input provided"
|
|
|
157 |
|
|
|
158 |
voices = asyncio.run(get_voices())
|
159 |
if selected_voice not in voices:
|
160 |
+
return f"La voz '{selected_voice}' no es válida. Por favor, seleccione una de las siguientes voces: {', '.join(voices.keys())}"
|
161 |
|
|
|
162 |
try:
|
163 |
audio_file = asyncio.run(text_to_speech(final_text, selected_voice, rate, pitch))
|
164 |
except Exception as e:
|
165 |
+
return f"Error generando audio: {e}"
|
|
|
166 |
|
|
|
167 |
try:
|
168 |
video_clip = concatenate_pexels_videos(keywords, num_videos_per_keyword=1)
|
169 |
except Exception as e:
|
170 |
+
return f"Error concatenando videos: {e}"
|
|
|
171 |
|
|
|
172 |
if mp3_file is not None:
|
173 |
music_clip = adjust_background_music(video_clip.duration, mp3_file.name)
|
174 |
else:
|
175 |
music_clip = None
|
176 |
|
|
|
177 |
final_video_path = combine_audio_video(audio_file, video_clip, music_clip)
|
|
|
|
|
|
|
|
|
178 |
upload_to_google_drive(final_video_path)
|
179 |
+
return final_video_path
|
|
|
|
|
|
|
|
|
|
|
180 |
|
181 |
except Exception as e:
|
182 |
+
return f"Error durante el procesamiento: {e}"
|
|
|
183 |
|
184 |
def upload_to_google_drive(file_path):
|
185 |
try:
|