pipeline{
    agent any

    parameters {
        string(
            name: 'EPOCHS',
            description: 'Number of epochs',
            defaultValue: '10'
        )
    }

    stages{
        stage('Copy dataset'){
            steps{
                build job: 'z-s487175-create-dataset', propagate: true
                sh "docker cp \$(docker ps -l -q):/app/dane ${env.WORKSPACE}"
            }
        }

        stage('Train model'){
            steps{
                def epochs = params.EPOCHS ?: '10' // Domyślna wartość 10, jeśli parametr EPOCHS nie jest przekazany
                sh "docker run -v ${env.WORKSPACE}/dane:/app/dane ium python3 /app/DL-model.py --epochs ${epochs}"
            }
        }

        stage('Archive model'){
            steps{
                archiveArtifacts artifacts: 'model_with_data.pickle', fingerprint: true
            }
        }
    }
}