pipeline {
    agent {
        docker {
            image 'python:3.11'
            args '-v /root/.cache:/root/.cache'
        }
    }
    parameters {
        string(name: 'EPOCHS', defaultValue: '10', description: 'Liczba Epok')
    }
    stages {
        stage('Pobierz dane') {
            steps {
                script {
                    copyArtifacts(projectName: 's487187-create-dataset', fingerprintArtifacts: true)
                }
            }
        }
        stage('Trenuj model') {
            steps {
                script {
                    // sh 'python3 train.py ${params.EPOCHS}'
                    sh 'python3 train.py'
                }
            }
        }
        stage('Zarchiwizuj model') {
            steps {
                archiveArtifacts artifacts: 'model.h5', fingerprint: true
            }
        }
    }
}