Update app.py

app.py

@@ -6,23 +6,29 @@ from PIL import Image, ImageDraw
 import tempfile
 import os
 import logging
+from tqdm import tqdm
 
 # Configuración de logging
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(message)s')
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    handlers=[logging.StreamHandler()]
+)
+logger = logging.getLogger("audio_to_video")
 
-def generate_video(audio_path, image_path):
+def generate_video(audio_file, image_file):
     try:
         # 1. Cargar audio
-        y, sr = librosa.load(audio_path, sr=None)
+        y, sr = librosa.load(audio_file)
         duration = librosa.get_duration(y=y, sr=sr)
-        logging.info(f"Audio cargado ({duration:.1f} segundos)")
+        logger.info(f"Audio cargado: {duration:.1f} segundos")
 
         # 2. Cargar imagen
-        img = Image.open(image_path).convert('RGB')
+        img = Image.open(image_file).convert('RGB')
         img_w, img_h = img.size
-        logging.info(f"Imagen cargada ({img_w}x{img_h})")
+        logger.info(f"Imagen cargada: {img_w}x{img_h}")
 
-        # 3. Crear efecto de onda
+        # 3. Crear efecto waveform
         audio_envelope = np.abs(y) / np.max(np.abs(y))  # Normalizar
         audio_envelope *= img_h // 4  # Escalar al 25% de la altura
 
@@ -31,13 +37,11 @@ def generate_video(audio_path, image_path):
             frame = img.copy()
             draw = ImageDraw.Draw(frame)
             
-            # Calcular posición en el audio
             time_idx = int(t * sr)
             start = max(0, time_idx - sr//10)
             end = min(len(audio_envelope), time_idx + sr//10)
             wave_slice = audio_envelope[start:end]
             
-            # Dibujar onda
             points = []
             for i, val in enumerate(wave_slice):
                 x = int((i / len(wave_slice)) * img_w)
@@ -46,35 +50,47 @@ def generate_video(audio_path, image_path):
                 points.extend([(x, y_pos), (x, y_neg)])
             
             if len(points) > 2:
-                draw.polygon(points, fill=(255, 0, 0, 128))  # Rojo semitransparente
+                draw.polygon(points, fill=(255, 0, 0, 128))
             
             return np.array(frame)
 
         # 5. Crear video
         video = mp.VideoClip(make_frame, duration=duration)
         video.fps = 24
-        video = video.set_audio(mp.AudioFileClip(audio_path))
-
-        # 6. Guardar video temporalmente
-        temp_file = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False)
-        video.write_videofile(temp_file.name, codec="libx264", audio_codec="aac", fps=24, logger=None)
-        logging.info(f"Video guardado temporalmente en: {temp_file.name}")
+        video = video.set_audio(mp.AudioFileClip(audio_file))
 
-        # Retornar el archivo como descargable
-        return temp_file.name
+        # 6. Guardar video
+        with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmpfile:
+            # Mostrar barra de progreso
+            with tqdm(total=100, desc="Exportando") as pbar:
+                video.write_videofile(
+                    tmpfile.name,
+                    codec="libx264",
+                    audio_codec="aac",
+                    fps=24,
+                    logger=None,
+                    progress_callback=lambda gd: pbar.update(gd['percent'] - pbar.n)
+                )
+            
+            # Asegurar que el archivo existe
+            if not os.path.exists(tmpfile.name):
+                raise Exception("Error al guardar el video temporal")
+            
+            logger.info(f"Video guardado: {tmpfile.name}")
+            return tmpfile.name  # Retornar la ruta del archivo
 
     except Exception as e:
-        logging.error(f"Error: {str(e)}")
+        logger.error(f"Error crítico: {str(e)}")
         return f"Error: {str(e)}"
 
 # Interfaz Gradio
 iface = gr.Interface(
     fn=generate_video,
     inputs=[
-        gr.Audio(type="filepath", label="Subir Audio (WAV/MP3)"),
-        gr.Image(type="filepath", label="Subir Imagen de Fondo")
+        gr.Audio(type="filepath", label="Audio (WAV/MP3)"),
+        gr.Image(type="filepath", label="Imagen de Fondo")
     ],
-    outputs=gr.File(label="Descargar Video"),  # Cambiado a gr.File para descargar
+    outputs=gr.File(label="Descargar Video"),  # Usar File para descargar
     title="Generador de Video Musical",
     description="Crea videos con efectos de audio sincronizados. Sube un audio y una imagen."
 )