import gradio as gr
from modules.extractive import TFIDFSummarizer, TextRankSummarizer, CombinedSummarizer, BERTSummarizer
from modules.abstractive import load_summarizers, abstractive_summary
from modules.preprocessing import Preprocessor, PDFProcessor
from modules.utils import handle_long_text

# Cargar modelos abstractivos finetuneados
summarizers = load_summarizers()

# Ejemplo de texto
EXAMPLE_TEXT = """
La inteligencia artificial (IA) ha revolucionado múltiples industrias, desde la medicina hasta el transporte.
Los avances en modelos de lenguaje, como T5, BART y PEGASUS, permiten aplicaciones innovadoras como la generación automática de resúmenes, lo cual facilita el acceso a información clave en documentos extensos.
"""

# Función para procesar el archivo cargado
def process_file(file):
    """
    Procesa un archivo cargado y extrae texto si es un PDF válido.

    Args:
        file (UploadedFile): Archivo subido por el usuario.

    Returns:
        str: Texto extraído del archivo o mensaje de error.
    """
    if file is not None:
        pdf_processor = PDFProcessor()
        input_text = pdf_processor.pdf_to_text(file.name)
        if input_text.strip():
            return input_text
        return "El archivo no contiene texto procesable."
    return "Por favor, cargue un archivo válido."

# Función para cargar un ejemplo de texto
def load_example_text():
    """
    Devuelve un ejemplo de texto predefinido.

    Returns:
        str: Texto de ejemplo.
    """
    return EXAMPLE_TEXT

# Función para cargar y limpiar el contenido de un archivo
def process_uploaded_file(file):
    """
    Procesa un archivo, lo limpia y devuelve su contenido en texto.

    Args:
        file (UploadedFile): Archivo subido.

    Returns:
        str: Texto limpio extraído del archivo.
    """
    raw_text = process_file(file)
    if "El archivo no contiene" not in raw_text and "Por favor" not in raw_text:
        preprocessor = Preprocessor()
        return preprocessor.clean_text(raw_text)
    return raw_text

# Interfaz dinámica
with gr.Blocks() as interface:
    gr.Markdown("# Aplicación Híbrida para Resumir Documentos de Forma Extractiva y Abstractiva")

    # Entrada de texto o archivo
    with gr.Row():
        with gr.Column(scale=2):
            input_text = gr.Textbox(lines=9, label="Ingrese texto", interactive=True)
            with gr.Row():
                load_example_button = gr.Button("Load Example")
                upload_file_button = gr.Button("Upload File")
        with gr.Column(scale=1):
            file = gr.File(label="Subir archivo (PDF, TXT)")

    # Acciones de botones
    load_example_button.click(
        load_example_text,
        inputs=[],
        outputs=[input_text],
    )

    upload_file_button.click(
        process_uploaded_file,
        inputs=[file],
        outputs=[input_text],
    )

    # Selección de tipo de resumen y opciones dinámicas
    summary_type = gr.Radio(
        ["Extractivo", "Abstractivo", "Combinado"],
        label="Tipo de resumen",
        value="Extractivo",
    )
    
    method = gr.Radio(
        ["TF-IDF", "TextRank", "BERT", "TF-IDF + TextRank"],
        label="Método Extractivo",
        visible=True,
    )
    num_sentences = gr.Slider(
        1, 10, value=3, step=1, label="Número de oraciones (Extractivo)", visible=True
    )
    model_name = gr.Radio(
        ["Pegasus", "T5", "BART"],
        label="Modelo Abstractivo",
        visible=False,
    )
    max_length = gr.Slider(
        50, 300, value=128, step=10, label="Longitud máxima (Abstractivo)", visible=False
    )
    num_beams = gr.Slider(
        1, 10, value=4, step=1, label="Número de haces (Abstractivo)", visible=False
    )

    def update_options(summary_type):
        if summary_type == "Extractivo":
            return (
                gr.update(visible=True), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False),
                gr.update(visible=False))
        elif summary_type == "Abstractivo":
            return (
                gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=True),
                gr.update(visible=True))
        elif summary_type == "Combinado":
            return (gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True),
                    gr.update(visible=True))
        else:
            return (
                gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False),
                gr.update(visible=False))

    summary_type.change(
        update_options,
        inputs=[summary_type],
        outputs=[method, num_sentences, model_name, max_length, num_beams],
    )

    summarize_button = gr.Button("Generar Resumen")
    output = gr.Textbox(lines=10, label="Resumen generado", interactive=True)
    copy_button = gr.Button("Copiar Resumen")

    summarize_button.click(
        summarize,
        inputs=[input_text, file, summary_type, method, num_sentences, model_name, max_length, num_beams],
        outputs=output,
    )

    def copy_summary(summary):
        return summary

    copy_button.click(
        fn=copy_summary,
        inputs=[output],
        outputs=[output],
        js="""function(summary) { navigator.clipboard.writeText(summary); return summary; }""",
    )

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