Update app.py

app.py

@@ -3,14 +3,32 @@ import subprocess
 import os
 import uuid
 
+def get_audio_duration(audio_path):
+    """Obtiene duración en segundos del audio usando ffprobe"""
+    result = subprocess.run([
+        "ffprobe", "-v", "error", "-show_entries",
+        "format=duration", "-of",
+        "default=noprint_wrappers=1:nokey=1", audio_path
+    ], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+    
+    try:
+        return float(result.stdout.strip())
+    except:
+        return None
+
 def generate_video(audio_path, image_path):
     output_filename = f"output_{uuid.uuid4().hex[:8]}.mp4"
+    duration = get_audio_duration(audio_path)
+    if not duration:
+        return "❌ Could not get audio duration. Check your audio file."
 
     try:
         cmd = [
             "ffmpeg", "-y", "-loop", "1", "-framerate", "1", "-i", image_path,
             "-i", audio_path, "-c:v", "libx264", "-preset", "ultrafast", "-tune", "stillimage",
-            "-c:a", "aac", "-b:a", "192k", "-shortest", "-r", "1", "-pix_fmt", "yuv420p", output_filename
+            "-c:a", "aac", "-b:a", "192k",
+            "-t", str(duration),    # <- aquí limitamos la duración del video al audio
+            "-r", "1", "-pix_fmt", "yuv420p", output_filename
         ]
         subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
 
@@ -22,7 +40,6 @@ def generate_video(audio_path, image_path):
     except Exception as e:
         return f"⚠️ Error: {str(e)}"
 
-# UI with fixed types
 with gr.Blocks(theme="Taithrah/Minimal") as app:
     gr.Markdown("# 🎶 Audio to Video Tool\nUpload an audio file and an image to generate a simple MP4 video (like TunesToTube)")
     with gr.Row():