UI.py

# UI.py
import gradio as gr

# Ya no importamos process_and_plot aquí, se pasará como argumento a create_interface

def create_interface(process_function_for_button): # <-- AÑADIDO: process_function_for_button
    """
    Esta función crea la interfaz de usuario y la devuelve.
    Conecta el botón de submit a la 'process_function_for_button' proporcionada.
    """
    
    with gr.Blocks(theme='upsatwal/mlsc_tiet') as demo:
    # with gr.Blocks(theme=gr.themes.Soft()) as demo: 
        gr.Markdown("# Modelado de Bioprocesos con Ecuaciones Personalizadas y Análisis por IA")

        with gr.Row():
            with gr.Column(scale=2):
                gr.Markdown("### Carga de Datos y Configuración General")
                file_input = gr.File(label="Subir archivo Excel (.xlsx)", file_types=[".xlsx"])
                
                show_legend_ui = gr.Checkbox(label="Mostrar leyenda en gráficos", value=True)
                show_params_ui = gr.Checkbox(label="Mostrar parámetros ajustados en gráficos", value=True)
                legend_position_ui = gr.Dropdown(
                    label="Posición de la leyenda",
                    choices=['best', 'upper right', 'upper left', 'lower right', 'lower left', 'center left', 'center right', 'lower center', 'upper center', 'center'],
                    value='best'
                )
            with gr.Column(scale=1):
                gr.Markdown("### Conteo de Ecuaciones")
                biomass_eq_count_ui = gr.Number(label="Número de ecuaciones de Biomasa a probar (1-3)", value=1, minimum=1, maximum=3, precision=0, step=1)
                substrate_eq_count_ui = gr.Number(label="Número de ecuaciones de Sustrato a probar (1-3)", value=1, minimum=1, maximum=3, precision=0, step=1)
                product_eq_count_ui = gr.Number(label="Número de ecuaciones de Producto a probar (1-3)", value=1, minimum=1, maximum=3, precision=0, step=1)

        with gr.Accordion("Ecuaciones y Parámetros de Biomasa", open=True):
            with gr.Row():
                with gr.Column():
                    biomass_eq1_ui = gr.Textbox(label="Ecuación de Biomasa 1 (ej: Xm * (1 - exp(-um * t)))", value="Xm * (1 - exp(-um * t_lag))", lines=2)
                    biomass_param1_ui = gr.Textbox(label="Parámetros Biomasa 1 (coma sep., ej: Xm, um, t_lag)", value="Xm, um, t_lag", info="Use 't' para tiempo. X_val para S/P.")
                    biomass_bound1_ui = gr.Textbox(label="Límites Biomasa 1 (ej: (0,inf),(0,5),(0,inf))", value="(0, inf), (0, inf), (0, inf)", info="Formato: (low,high). Use np.inf.")
                biomass_col2 = gr.Column(visible=False)
                with biomass_col2:
                    biomass_eq2_ui = gr.Textbox(label="Ecuación de Biomasa 2", value="X0 * exp(um * t)", lines=2)
                    biomass_param2_ui = gr.Textbox(label="Parámetros Biomasa 2", value="X0, um")
                    biomass_bound2_ui = gr.Textbox(label="Límites Biomasa 2", value="(0, inf), (0, inf)")
                biomass_col3 = gr.Column(visible=False)
                with biomass_col3:
                    biomass_eq3_ui = gr.Textbox(label="Ecuación de Biomasa 3", lines=2)
                    biomass_param3_ui = gr.Textbox(label="Parámetros Biomasa 3")
                    biomass_bound3_ui = gr.Textbox(label="Límites Biomasa 3")
        
        with gr.Accordion("Ecuaciones y Parámetros de Sustrato", open=True):
            gr.Markdown("Para Sustrato/Producto, usa `X_val` para X(t). Ejemplo: `S0 - (1/YXS) * (X_val - X0_bio)` donde X0_bio es un parámetro o valor.")
            with gr.Row():
                with gr.Column():
                    substrate_eq1_ui = gr.Textbox(label="Ecuación de Sustrato 1", value="S0 - (X_val / YXS) - mS * t", lines=2)
                    substrate_param1_ui = gr.Textbox(label="Parámetros Sustrato 1", value="S0, YXS, mS")
                    substrate_bound1_ui = gr.Textbox(label="Límites Sustrato 1", value="(0, inf), (0.01, inf), (0, inf)")
                substrate_col2 = gr.Column(visible=False)
                with substrate_col2:
                    substrate_eq2_ui = gr.Textbox(label="Ecuación de Sustrato 2", lines=2)
                    substrate_param2_ui = gr.Textbox(label="Parámetros Sustrato 2")
                    substrate_bound2_ui = gr.Textbox(label="Límites Sustrato 2")
                substrate_col3 = gr.Column(visible=False)
                with substrate_col3:
                    substrate_eq3_ui = gr.Textbox(label="Ecuación de Sustrato 3", lines=2)
                    substrate_param3_ui = gr.Textbox(label="Parámetros Sustrato 3")
                    substrate_bound3_ui = gr.Textbox(label="Límites Sustrato 3")

        with gr.Accordion("Ecuaciones y Parámetros de Producto", open=True):
            with gr.Row():
                with gr.Column():
                    product_eq1_ui = gr.Textbox(label="Ecuación de Producto 1", value="P0 + YPX * X_val + mP * t", lines=2)
                    product_param1_ui = gr.Textbox(label="Parámetros Producto 1", value="P0, YPX, mP")
                    product_bound1_ui = gr.Textbox(label="Límites Producto 1", value="(0, inf), (0, inf), (0, inf)")
                product_col2 = gr.Column(visible=False)
                with product_col2:
                    product_eq2_ui = gr.Textbox(label="Ecuación de Producto 2", lines=2)
                    product_param2_ui = gr.Textbox(label="Parámetros Producto 2")
                    product_bound2_ui = gr.Textbox(label="Límites Producto 2")
                product_col3 = gr.Column(visible=False)
                with product_col3:
                    product_eq3_ui = gr.Textbox(label="Ecuación de Producto 3", lines=2)
                    product_param3_ui = gr.Textbox(label="Parámetros Producto 3")
                    product_bound3_ui = gr.Textbox(label="Límites Producto 3")

        def update_visibility(count): # Simplificado, devuelve tupla de dicts
            return {"visible": count >= 2}, {"visible": count >= 3}

        biomass_eq_count_ui.change(fn=update_visibility, inputs=biomass_eq_count_ui, outputs=[biomass_col2, biomass_col3])
        substrate_eq_count_ui.change(fn=update_visibility, inputs=substrate_eq_count_ui, outputs=[substrate_col2, substrate_col3])
        product_eq_count_ui.change(fn=update_visibility, inputs=product_eq_count_ui, outputs=[product_col2, product_col3])
        
        submit_button = gr.Button("Procesar y Analizar", variant="primary", scale=1)
        
        gr.Markdown("## Resultados del Análisis")
        with gr.Row():
            image_output = gr.Image(label="Gráfico Generado", type="pil", width=600, height=900, scale=2)
            with gr.Column(scale=3):
                analysis_output = gr.Markdown(label="Análisis del Modelo por IA")
        
        all_inputs = [
            file_input,
            biomass_eq1_ui, biomass_eq2_ui, biomass_eq3_ui,
            biomass_param1_ui, biomass_param2_ui, biomass_param3_ui,
            biomass_bound1_ui, biomass_bound2_ui, biomass_bound3_ui,
            substrate_eq1_ui, substrate_eq2_ui, substrate_eq3_ui,
            substrate_param1_ui, substrate_param2_ui, substrate_param3_ui,
            substrate_bound1_ui, substrate_bound2_ui, substrate_bound3_ui,
            product_eq1_ui, product_eq2_ui, product_eq3_ui,
            product_param1_ui, product_param2_ui, product_param3_ui,
            product_bound1_ui, product_bound2_ui, product_bound3_ui,
            legend_position_ui,
            show_legend_ui,
            show_params_ui,
            biomass_eq_count_ui,
            substrate_eq_count_ui,
            product_eq_count_ui
        ]
        
        # --- CONEXIÓN DEL BOTÓN DENTRO DEL CONTEXTO DE BLOCKS ---
        submit_button.click(
            fn=process_function_for_button, # Usa la función pasada como argumento
            inputs=all_inputs,
            outputs=[image_output, analysis_output]
        )

        # Inicializar la visibilidad correctamente al cargar la demo
        # No es necesario demo.load para esto si la visibilidad por defecto es False
        # y la función `update_visibility` se llama con el valor inicial del Number input.
        # Para asegurar que se llama al inicio:
        demo.load(lambda val_b, val_s, val_p: (
            update_visibility(val_b)[0], update_visibility(val_b)[1], # Para biomasa
            update_visibility(val_s)[0], update_visibility(val_s)[1], # Para sustrato
            update_visibility(val_p)[0], update_visibility(val_p)[1]  # Para producto
            ), 
            inputs=[biomass_eq_count_ui, substrate_eq_count_ui, product_eq_count_ui], 
            outputs=[biomass_col2, biomass_col3, substrate_col2, substrate_col3, product_col2, product_col3]
        )

    return demo