pipeline {
    agent {
        docker {
            image 'python:3.11'
            args '-v /root/.cache:/root/.cache -u root -v /tmp/mlruns:/tmp/mlruns -v /mlruns:/mlruns'
        }
    }
    parameters {
        string(name: 'EPOCHS', defaultValue: '10', description: 'Liczba Epok')
    }
    stages {
        stage('Przygotowanie) {
            steps {
                sh 'pip install pandas tensorflow numpy scikit-learn mlflow'
            }
        }
        stage('Pobierz dane') {
            steps {
                script {
                    copyArtifacts(projectName: 's407409-create-dataset', fingerprintArtifacts: true)
                }
            }
        }
        stage('Trenuj model') {
            steps {
                script {
                    sh "python3 model_train.py"
                }
            }
        }
        stage('Zarchiwizuj model') {
            steps {
                archiveArtifacts artifacts: 'model_election.h5', fingerprint: true
            }
        }
    }
}