Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import subprocess
 import logging
-from typing import Optional, List
+from typing import Optional
 from fastapi import FastAPI, HTTPException
 import gradio as gr
 
@@ -20,6 +20,16 @@ os.makedirs(UPLOAD_DIR, exist_ok=True)
 VIDEO_FORMATS = ['.mp4', '.avi', '.mov', '.mkv', '.webm', '.flv', '.wmv']
 AUDIO_FORMATS = ['.mp3', '.wav', '.aac', '.flac', '.ogg', '.m4a', '.wma']
 
+def detect_media_type(file_path: str) -> str:
+    """Detectar si el archivo es de audio o video."""
+    ext = os.path.splitext(file_path)[1].lower()
+    if ext in VIDEO_FORMATS:
+        return 'video'
+    elif ext in AUDIO_FORMATS:
+        return 'audio'
+    else:
+        return 'unsupported'
+
 def sanitize_filename(filename: str) -> str:
     """Limpiar y validar nombre de archivo para prevenir riesgos de seguridad."""
     return ''.join(c for c in filename if c.isalnum() or c in ('.', '_', '-')).rstrip()
@@ -43,20 +53,21 @@ def make_ffmpeg_executable(ffmpeg_path: str):
         logger.error(f"Error al configurar permisos de FFmpeg: {e}")
         raise
 
-def convert_media(input_file: str, output_dir: str, output_format: str = 'mp4') -> str:
+def convert_media(input_file: str, output_dir: str) -> str:
     """Convertir archivos multimedia con configuraciones optimizadas."""
     try:
+        media_type = detect_media_type(input_file)
         base_name = os.path.basename(input_file)
+        
+        # Elegir extensión de salida según el tipo de medio
+        output_extension = 'mp4' if media_type == 'video' else 'm4a'
         output_filename = ensure_unique_filename(
             output_dir, 
-            f"{os.path.splitext(base_name)[0]}_converted.{output_format}"
+            f"{os.path.splitext(base_name)[0]}_converted.{output_extension}"
         )
         output_file = os.path.join(output_dir, output_filename)
         
-        # Determinar comandos según sea video o audio
-        input_extension = os.path.splitext(input_file)[1].lower()
-        
-        if input_extension in VIDEO_FORMATS:
+        if media_type == 'video':
             # Configuraciones de conversión de video
             ffmpeg_command = [
                 FFMPEG_PATH,
@@ -71,21 +82,19 @@ def convert_media(input_file: str, output_dir: str, output_format: str = 'mp4')
                 '-movflags', '+faststart',  # Web optimization
                 output_file
             ]
-        elif input_extension in AUDIO_FORMATS:
+        elif media_type == 'audio':
             # Configuraciones de conversión de audio
             ffmpeg_command = [
                 FFMPEG_PATH,
                 '-i', input_file,
                 '-vn',  # Ignorar video
-                '-c:a', 'libfdk_aac',  # Codificador según formato
-                -'profile:a', 'aac_he_v2',
-                '-b:a', '48k',  # Calidad de audio alta
+                '-c:a', 'libfdk_aac',  # Codificador AAC para M4A
+                '-b:a', '192k',  # Calidad de audio alta
                 '-ar', '44100',  # Frecuencia de muestreo
-                '-preset', 'slow',
                 output_file
             ]
         else:
-            raise ValueError(f"Formato no soportado: {input_extension}")
+            raise ValueError("Formato no soportado")
         
         # Ejecutar conversión
         result = subprocess.run(
@@ -104,17 +113,17 @@ def convert_media(input_file: str, output_dir: str, output_format: str = 'mp4')
         logger.error(f"Error inesperado durante la conversión: {e}")
         raise HTTPException(status_code=500, detail="Error inesperado durante la conversión")
 
-def process_media(file_path: str, output_format: str = 'mp4') -> str:
+def process_media(file_path: str) -> str:
     """Procesar medio completo."""
-    return convert_media(file_path, UPLOAD_DIR, output_format)
+    return convert_media(file_path, UPLOAD_DIR)
 
-def gradio_interface(media: Optional[str], output_format: str) -> Optional[str]:
+def gradio_interface(media: Optional[str]) -> Optional[str]:
     """Interfaz de Gradio para conversión de medios."""
     if not media:
         raise gr.Error("No se ha subido ningún medio. Por favor, sube un archivo.")
     
     try:
-        converted_media = process_media(media, output_format)
+        converted_media = process_media(media)
         return converted_media
     except Exception as e:
         raise gr.Error(f"Conversión fallida: {str(e)}")
@@ -128,24 +137,17 @@ app = FastAPI()
 # Configurar interfaz de Gradio
 iface = gr.Interface(
     fn=gradio_interface,
-    inputs=[
-        gr.File(
-            label="Subir Archivo", 
-            type="filepath",  # Cambié esto de 'file' a 'filepath'
-        ),
-        gr.Dropdown(
-            label="Formato de Salida", 
-            choices=['mp4', 'mp3', 'wav', 'avi', 'mkv', 'flac'],
-            value='mp4'
-        )
-    ],
+    inputs=gr.File(
+        label="Subir Archivo", 
+        type="filepath"
+    ),
     outputs=gr.File(
         label="Descargar Archivo Convertido", 
-        type="filepath"  # Cambié esto de 'file' a 'filepath'
+        type="filepath"
     ),
-    title="🎥🎵 Convertidor Universal de Medios",
-    description="Convierte videos y audios a múltiples formatos con configuraciones optimizadas",
-    theme="huggingface"  # Tema de HuggingFace
+    title="🎥🎵 Convertidor Universal",
+    description="Convierte videos a MP4 y audios a M4A con configuraciones optimizadas",
+    theme="huggingface"
 )
 
 # Lanzar interfaz de Gradio