Update app.py

app.py

@@ -3,8 +3,8 @@ import moviepy.editor as mp
 import numpy as np
 import librosa
 from PIL import Image, ImageDraw
-import tempfile
 import os
+import tempfile
 import logging
 
 # Configuración de logging
@@ -29,20 +29,18 @@ def generate_video(audio_file, image_file):
 
         # 3. Analizar audio
         audio_envelope = np.abs(y) / np.max(np.abs(y))  # Normalizar
-        audio_envelope_zoom = audio_envelope * 0.2 + 0.9  # Escalar para zoom (0.9x a 1.1x)
-        audio_envelope_wave = audio_envelope * (img_h // 6)  # Para el waveform
+        audio_envelope_zoom = audio_envelope * 0.2 + 0.9  # Para zoom
+        audio_envelope_wave = audio_envelope * (img_h // 6)  # Para waveform
 
-        # 4. Generar frames con zoom y waveform
+        # 4. Generar frames
         def make_frame(t):
-            # Calcular posición en el audio
+            # --- Zoom automático ---
             time_idx = int(t * sr)
-            
-            # --- Efecto de Zoom ---
             zoom_factor = audio_envelope_zoom[time_idx] if time_idx < len(audio_envelope_zoom) else 1.0
             new_size = (int(img_w * zoom_factor), int(img_h * zoom_factor))
             zoomed_img = img.resize(new_size, Image.LANCZOS)
             
-            # Centrar la imagen ampliada
+            # Recortar al tamaño original
             x_offset = (new_size[0] - img_w) // 2
             y_offset = (new_size[1] - img_h) // 2
             cropped_img = zoomed_img.crop((
@@ -52,11 +50,9 @@ def generate_video(audio_file, image_file):
                 y_offset + img_h
             ))
 
-            # --- Dibujar Waveform ---
+            # --- Waveform ---
             frame = ImageDraw.Draw(cropped_img)
-            
-            # Posición vertical del waveform (abajo)
-            start_y = int(img_h * 0.75)  # 75% hacia abajo
+            start_y = int(img_h * 0.8)  # 80% hacia abajo
             
             # Extraer slice de audio
             start = max(0, time_idx - sr//10)
@@ -72,7 +68,7 @@ def generate_video(audio_file, image_file):
                 points.extend([(x, y_pos), (x, y_neg)])
             
             if len(points) > 2:
-                frame.polygon(points, fill=(255, 0, 0, 150))  # Rojo semitransparente
+                frame.polygon(points, fill=(255, 0, 0, 150))
             
             return np.array(cropped_img)
 
@@ -82,21 +78,23 @@ def generate_video(audio_file, image_file):
         video = video.set_audio(mp.AudioFileClip(audio_file))
 
         # 6. Guardar video
-        with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmpfile:
-            video.write_videofile(
-                tmpfile.name,
-                codec="libx264",
-                audio_codec="aac",
-                fps=24,
-                logger=None
-            )
-            
-            logger.info(f"Video guardado: {tmpfile.name}")
-            return tmpfile.name
+        temp_dir = tempfile.mkdtemp()
+        output_path = os.path.join(temp_dir, "output.mp4")
+        
+        video.write_videofile(
+            output_path,
+            codec="libx264",
+            audio_codec="aac",
+            fps=24,
+            logger=None
+        )
+        
+        logger.info(f"Video guardado en: {output_path}")
+        return output_path  # Retornar la ruta completa
 
     except Exception as e:
         logger.error(f"Error crítico: {str(e)}")
-        return f"Error: {str(e)}"
+        return None  # Retornar None en caso de error
 
 # Interfaz Gradio
 iface = gr.Interface(
@@ -105,7 +103,7 @@ iface = gr.Interface(
         gr.Audio(type="filepath", label="Audio (WAV/MP3)"),
         gr.Image(type="filepath", label="Imagen de Fondo")
     ],
-    outputs=gr.File(label="Descargar Video"),
+    outputs=gr.File(label="Descargar Video"),  # Muestra el enlace de descarga
     title="Generador de Video Musical",
     description="Crea videos con zoom automático y efectos de audio sincronizados"
 )