Update app.py

app.py

@@ -3,68 +3,57 @@ import moviepy.editor as mp
 import numpy as np
 import librosa
 import matplotlib.pyplot as plt
-from io import BytesIO
-import logging
+from PIL import Image
+import tempfile
+import os
 
-# Configuración de logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger("audio_to_video")
-
-def generate_waveform_video(audio_path, image_path):
+def generate_waveform_video(audio_file, image_file):
     try:
         # 1. Cargar audio
-        logger.info("Cargando archivo de audio...")
-        y, sr = librosa.load(audio_path)
+        y, sr = librosa.load(audio_file)
         duration = librosa.get_duration(y=y, sr=sr)
-        logger.info(f"Duración del audio: {duration:.2f} segundos")
 
         # 2. Cargar imagen
-        logger.info("Procesando imagen...")
-        img_clip = mp.ImageClip(image_path).set_duration(duration)
-        img_width, img_height = img_clip.size
+        img_clip = mp.ImageClip(image_file).set_duration(duration)
+        img_w, img_h = img_clip.size
 
         # 3. Crear efecto de waveform
-        logger.info("Generando efecto visual...")
-        audio_envelope = np.abs(y)  # Envelope del audio
-        audio_envelope = (audio_envelope / np.max(audio_envelope)) * (img_height // 3)
+        audio_envelope = np.abs(y)
+        audio_envelope = (audio_envelope / np.max(audio_envelope)) * (img_h // 3)
 
         def make_frame(t):
-            fig, ax = plt.subplots(figsize=(img_width/100, img_height/100), dpi=100)
+            fig, ax = plt.subplots(figsize=(img_w/100, img_h/100), dpi=100)
             ax.set_xlim(0, duration)
-            ax.set_ylim(-img_height//2, img_height//2)
+            ax.set_ylim(-img_h//2, img_h//2)
             ax.axis('off')
-            
-            time_index = int(t * sr)
-            start = max(0, time_index - sr//10)
-            end = min(len(audio_envelope), time_index + sr//10)
+
+            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]
-            
-            x_values = np.linspace(t-0.1, t+0.1, len(wave_slice))
-            ax.fill_between(x_values, wave_slice - img_height//4, -wave_slice + img_height//4, 
+
+            x = np.linspace(t-0.1, t+0.1, len(wave_slice))
+            ax.fill_between(x, wave_slice - img_h//4, -wave_slice + img_h//4, 
                            facecolor='red', alpha=0.7)
-            
-            buf = BytesIO()
+
+            # Convertir figura a numpy array
+            buf = io.BytesIO()
             plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0)
             plt.close(fig)
-            return mp.ImageClip(buf).get_frame(0)
+            return np.array(Image.open(buf))  # Formato correcto para MoviePy
 
-        logger.info("Renderizando video...")
+        # Crear video
         effect_clip = mp.VideoClip(make_frame, duration=duration).set_fps(24)
         final_clip = mp.CompositeVideoClip([img_clip, effect_clip.set_pos("center")])
-        
-        # 4. Combinar con audio
-        final_clip = final_clip.set_audio(mp.AudioFileClip(audio_path))
-        
-        # 5. Guardar en memoria
-        buffer = BytesIO()
-        final_clip.write_videofile(buffer, fps=24, codec="libx264", 
-                                  audio_codec="aac", logger=None)
-        buffer.seek(0)
-        logger.info("Video generado exitosamente")
-        return buffer
+        final_clip = final_clip.set_audio(mp.AudioFileClip(audio_file))
+
+        # Guardar en archivo temporal
+        with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmpfile:
+            final_clip.write_videofile(tmpfile.name, fps=24, codec="libx264", 
+                                      audio_codec="aac", logger=None)
+            return tmpfile.name
 
     except Exception as e:
-        logger.error(f"Error durante la generación: {str(e)}")
         return f"Error: {str(e)}"
 
 # Interfaz Gradio
@@ -76,10 +65,8 @@ iface = gr.Interface(
     ],
     outputs=gr.Video(label="Video Resultante", format="mp4"),
     title="Generador de Video con Efectos de Audio",
-    description="Crea videos con efectos visuales sincronizados con el audio. Actualmente soporta efecto de waveform.",
-    allow_flagging="never"
+    description="Crea videos con efectos visuales sincronizados con el audio. Actualmente soporta efecto de waveform."
 )
 
 if __name__ == "__main__":
-    logger.info("Iniciando aplicación Gradio...")
-    iface.queue().launch(share=False, debug=True)
+    iface.queue().launch()