import gradio as gr
import subprocess
import tempfile
import os
import uuid

def convert_to_epub(uploaded_file, title, author):
    """
    Convierte un archivo de entrada (docx, pdf, md, etc.) a EPUB utilizando Pandoc.
    Permite incluir opcionalmente metadatos (título y autor).
    """
    # Verificar que se haya subido un archivo
    if uploaded_file is None:
        return "No se ha subido ningún archivo."
    
    # Obtener la extensión del archivo mediante su nombre original
    filename = uploaded_file.name if hasattr(uploaded_file, "name") else "input_file"
    ext = os.path.splitext(filename)[1].lower()
    if ext == "":
        return "El archivo subido no tiene extensión reconocible."
    
    # Crear un directorio temporal para manejar archivos de entrada y salida
    with tempfile.TemporaryDirectory() as tmpdir:
        input_filename = f"input_{uuid.uuid4().hex}{ext}"
        input_path = os.path.join(tmpdir, input_filename)
        output_path = os.path.join(tmpdir, "output.epub")
        
        # Escribir el archivo subido en disco
        with open(input_path, "wb") as f:
            with open(uploaded_file, "rb") as uploaded:
                f.write(uploaded.read())

        # Construir el comando para Pandoc
        cmd = ["pandoc", input_path, "-o", output_path, "--toc"]
        
        # Incluir metadatos opcionales (si el usuario los provee)
        if title.strip():
            cmd.extend(["--metadata", f"title={title.strip()}"])
        if author.strip():
            cmd.extend(["--metadata", f"author={author.strip()}"])
        
        # Ejecutar Pandoc y capturar errores, si los hubiera
        try:
            result = subprocess.run(
                cmd,
                check=True,
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                text=True
            )
        except subprocess.CalledProcessError as e:
            error_message = e.stderr if e.stderr else str(e)
            return f"Error en la conversión: {error_message}"
        
        # Verificar que el archivo EPUB se generó
        if not os.path.exists(output_path):
            return "Error: El archivo EPUB no se generó correctamente."
        
        # Leer el archivo EPUB generado
        with open(output_path, "rb") as f:
            epub_data = f.read()
    
    # Para que Gradio ofrezca el archivo para descarga, lo salvamos temporalmente
    output_filename = "output.epub"
    with open(output_filename, "wb") as f:
        f.write(epub_data)
        
    return output_filename

# Configuración de la interfaz de Gradio
iface = gr.Interface(
    fn=convert_to_epub,
    inputs=[
        gr.File(label="Sube un archivo (docx, pdf, md, etc.)"),
        gr.Textbox(label="Título (opcional)", placeholder="Título del documento"),
        gr.Textbox(label="Autor (opcional)", placeholder="Autor del documento")
    ],
    outputs=gr.File(label="Archivo EPUB"),
    title="Conversor a EPUB con Pandoc",
    description=(
        "Sube un documento en uno de los formatos soportados (docx, pdf, md, etc.) y conviértelo a EPUB utilizando Pandoc.\n\n"
    ),
    allow_flagging="never",
)

if __name__ == "__main__":
    iface.launch()