pipeline{
    agent any

    parameters {
        gitParameter(
            name: 'BRANCH',
            type: 'branch',
            defaultValue: 'master',
            description: 'Select the branch of training'
        )
    }

    stages{
         stage('Copy model artifacts with test dataset') {
            steps {
                script {
                    def myProject = "z-s487175-training/master"
                    copyArtifacts(
                        projectName: myProject,
                        fingerprintArtifacts: true,
                        selector: [$class: 'SpecificBuildSelector', buildNumber: "lastSuccessfulBuild"]
                    )
                }
            }
        }

        stage('Check workspace content') {
             steps {
                echo "${env.WORKSPACE}" 
            }
        }

        stage('Train model'){
            steps{
                sh "docker run -v ${env.WORKSPACE} ium python3 DL-prediction.py"
            }
        }

        stage('Archive prediction results'){
            steps{
                sh "docker cp \$(docker ps -l -q):/app/results_prediction.csv ${env.WORKSPACE}"
                archiveArtifacts artifacts: 'results_prediction.csv', fingerprint: true
            }
        }
    }
}