pipeline {
    agent {
        docker {
            image 'python:3.11'
            args '-v /root/.cache:/root/.cache -u root -v /tmp/mlruns:/tmp/mlruns -v /mlruns:/mlruns'
        }
    }
    
    stages {
        stage('Przygotowanie) {
            steps {
                sh 'pip install pandas tensorflow numpy scikit-learn mlflow'
            }
        }
        stage('Zaladuj dane i model') {
            steps {
                script {
                    copyArtifacts(projectName: 's407409-training', fingerprintArtifacts: true)
                }
            }
        }
        stage('Dokonaj predykcji') {
            steps {
                script {
                    sh "python3 model_test.py"
                }
            }
        }
        stage('Zarchiwizuj wyniki') {
            steps {
                archiveArtifacts artifacts: 'election_predictions.csv', fingerprint: true
            }
        }
    }
}