pipeline {
    agent { 
        dockerfile  true
        args '-v /mlruns:/mlruns'
    }

    parameters {
        string(
            defaultValue: 'kamilab0bkowska',
            description: 'kaggle username',
            name: 'KAGGLE_USERNAME',
            trim: false
             )
        password(
            defaultValue: '',
            description: 'kaggle passowrd',
            name: 'KAGGLE_KEY'
        )
    }

    stages {
        stage('Stage 1') {
            steps {
                echo 'Hello world!' 
            }
        }

        stage('Run mlflow on nn_train') {
            steps {
                withEnv(["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}",
                "KAGGLE_KEY=${params.KAGGLE_KEY}"]) {
                    sh 'python3 ./nn_train_mlflow.py'
                    archiveArtifacts artifacts: 'mlruns/**'
                    sh 'rm -r my_model_mlflow'
                }
            }
        }
    }
}