import gradio as gr
import subprocess

def entrenar_modelo(_):
    result = subprocess.run(["python", "train.py"], capture_output=True, text=True)
    return result.stdout + "\n\n" + result.stderr

gr.Interface(
    fn=entrenar_modelo,
    inputs=gr.Textbox(label="output"),
    outputs="text",
    title="Entrenamiento del modelo LoRA",
    description="Presiona para iniciar el entrenamiento con el dataset 'woman_latina'."
).launch()