import streamlit as st
import subprocess

def prepare_data():
    st.text("Préparation des données en cours...")
    subprocess.run(["python", "prepare.py"])
    st.text("Préparation des données terminée.")

def train_model():
    st.text("Entraînement du modèle en cours...")
    subprocess.run(["python", "train.py", "config/train_shakespeare_char.py", "--device=cpu", "--compile=False", "--eval_iters=20", "--log_interval=1", "--block_size=64", "--batch_size=12", "--n_layer=4", "--n_head=4", "--n_embd=128", "--max_iters=2000", "--lr_decay_iters=2000", "--dropout=0.0"])
    st.text("Entraînement du modèle terminé.")

def generate_samples():
    st.text("Génération d'échantillons en cours...")
    subprocess.run(["python", "sample.py", "--out_dir=out-shakespeare-char", "--device=cpu"])
    st.text("Génération d'échantillons terminée.")

def main():
    st.title("Application de Commandes")
    
    if st.button("Préparer les données"):
        prepare_data()
    
    if st.button("Entraîner le modèle"):
        train_model()
    
    if st.button("Générer des échantillons"):
        generate_samples()

if __name__ == "__main__":
    main()