from invoke import task
import preprocess
import train
import results
import predict


@task
def preprocess(ctx):
    print("Running preprocess step...")
    # Add invoke logic for preprocessing
    preprocess.run_preprocessing()


@task
def train(ctx):
    print("Running training step...")
    # Add invoke logic for training
    best_params = train.run_training(max_depth=10, min_samples_split=2)
    print(f"Best Parameters: {best_params}")
    
    analyze(ctx)


@task
def analyze(ctx):
    print("Running analysis step...")
    # Add invoke logic for result analysis
    results.evaluate_results()


@task
def predict(ctx):
    print("Running prediction step...")
    # Add invoke logic for prediction
    predict.run_prediction()