Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import subprocess
 import logging
-from typing import Optional
+from typing import Optional, List
 from fastapi import FastAPI, HTTPException
 import gradio as gr
 
@@ -16,6 +16,10 @@ FFMPEG_PATH = "./ffmpeg"
 # Crear directorios
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 
+# Definir formatos de video y audio soportados
+VIDEO_FORMATS = ['.mp4', '.avi', '.mov', '.mkv', '.webm', '.flv', '.wmv']
+AUDIO_FORMATS = ['.mp3', '.wav', '.aac', '.flac', '.ogg', '.m4a', '.wma']
+
 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()
@@ -39,30 +43,49 @@ def make_ffmpeg_executable(ffmpeg_path: str):
         logger.error(f"Error al configurar permisos de FFmpeg: {e}")
         raise
 
-def convert_video_to_mp4(input_file: str, output_dir: str) -> str:
-    """Convertir video a formato MP4 con configuraciones optimizadas."""
+def convert_media(input_file: str, output_dir: str, output_format: str = 'mp4') -> str:
+    """Convertir archivos multimedia con configuraciones optimizadas."""
     try:
-        # Generar nombre de archivo único
+        base_name = os.path.basename(input_file)
         output_filename = ensure_unique_filename(
             output_dir, 
-            os.path.basename(input_file).rsplit('.', 1)[0] + "_converted.mp4"
+            f"{os.path.splitext(base_name)[0]}_converted.{output_format}"
         )
         output_file = os.path.join(output_dir, output_filename)
         
-        # Comando FFmpeg con configuraciones optimizadas
-        ffmpeg_command = [
-            FFMPEG_PATH,
-            '-i', input_file,
-            '-vf', 'fps=24',  # Cambiar la tasa de fotogramas
-            '-c:v', 'libx264',  # Codificador de video
-            '-c:a', 'libfdk_aac',  # Codificador de audio
-            '-profile:a', 'aac_he_v2',  # Perfil de audio
-            '-crf', '28',  # Tasa de compresión
-            '-b:a', '32k',  # Tasa de bits de audio
-            '-preset', 'slow',  # Preajuste de codificación
-            '-movflags', '+faststart',  # Web optimization
-            output_file
-        ]
+        # Determinar comandos según sea video o audio
+        input_extension = os.path.splitext(input_file)[1].lower()
+        
+        if input_extension in VIDEO_FORMATS:
+            # Configuraciones de conversión de video
+            ffmpeg_command = [
+                FFMPEG_PATH,
+                '-i', input_file,
+                '-vf', 'fps=24',  # Cambiar la tasa de fotogramas
+                '-c:v', 'libx264',  # Codificador de video
+                '-c:a', 'libfdk_aac',  # Codificador de audio
+                '-profile:a', 'aac_he_v2',  # Perfil de audio
+                '-crf', '28',  # Tasa de compresión
+                '-b:a', '28k',  # Tasa de bits de audio
+                '-preset', 'slow',  # Preajuste de codificación
+                '-movflags', '+faststart',  # Web optimization
+                output_file
+            ]
+        elif input_extension in AUDIO_FORMATS:
+            # 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
+                '-ar', '44100',  # Frecuencia de muestreo
+                '-preset', 'slow',
+                output_file
+            ]
+        else:
+            raise ValueError(f"Formato no soportado: {input_extension}")
         
         # Ejecutar conversión
         result = subprocess.run(
@@ -71,28 +94,28 @@ def convert_video_to_mp4(input_file: str, output_dir: str) -> str:
             capture_output=True, 
             text=True
         )
-        logger.info(f"Video convertido exitosamente: {output_file}")
+        logger.info(f"Medio convertido exitosamente: {output_file}")
         return output_file
     
     except subprocess.CalledProcessError as e:
         logger.error(f"Error de conversión de FFmpeg: {e.stderr}")
-        raise HTTPException(status_code=500, detail=f"Fallo en la conversión de video: {e.stderr}")
+        raise HTTPException(status_code=500, detail=f"Fallo en la conversión de medio: {e.stderr}")
     except Exception as e:
         logger.error(f"Error inesperado durante la conversión: {e}")
-        raise HTTPException(status_code=500, detail="Error inesperado durante la conversión de video")
+        raise HTTPException(status_code=500, detail="Error inesperado durante la conversión")
 
-def process_video(file_path: str) -> str:
-    """Procesar el video completo."""
-    return convert_video_to_mp4(file_path, UPLOAD_DIR)
+def process_media(file_path: str, output_format: str = 'mp4') -> str:
+    """Procesar medio completo."""
+    return convert_media(file_path, UPLOAD_DIR, output_format)
 
-def gradio_interface(video: Optional[str]) -> Optional[str]:
-    """Interfaz de Gradio para conversión de video."""
-    if not video:
-        raise gr.Error("No se ha subido ningún video. Por favor, sube un video.")
+def gradio_interface(media: Optional[str], output_format: 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_video = process_video(video)
-        return converted_video
+        converted_media = process_media(media, output_format)
+        return converted_media
     except Exception as e:
         raise gr.Error(f"Conversión fallida: {str(e)}")
 
@@ -102,14 +125,29 @@ make_ffmpeg_executable(FFMPEG_PATH)
 # Crear aplicación FastAPI
 app = FastAPI()
 
-# Modificación de la interfaz de Gradio
+# Configurar interfaz de Gradio
 iface = gr.Interface(
     fn=gradio_interface,
-    inputs=gr.Video(label="Subir Video"),  # Eliminar los argumentos 'type' y 'sources'
-    outputs=gr.File(label="Descargar MP4 Convertido"),
-    title="🎥 Convertidor de Video",
-    description="Convierte videos a formato MP4 optimizado con configuraciones estándar",
-    theme="huggingface"
+    inputs=[
+        gr.File(
+            label="Subir Archivo", 
+            type="filepath",
+            file_count="single",
+            file_types=VIDEO_FORMATS + AUDIO_FORMATS
+        ),
+        gr.Dropdown(
+            label="Formato de Salida", 
+            choices=['ts', 'mp4', 'mp3', 'wav', 'avi', 'mkv', 'flac'],
+            value='mp4'
+        )
+    ],
+    outputs=gr.File(
+        label="Descargar Archivo Convertido", 
+        type="file"
+    ),
+    title="🎥🎵 Convertidor Universal de Medios",
+    description="Convierte videos y audios a múltiples formatos con configuraciones optimizadas",
+    theme="huggingface"  # Tema de HuggingFace
 )
 
 # Lanzar interfaz de Gradio