pipeline {
    agent {
        docker {
            image 'python:3.11'
            args '-v /root/.cache:/root/.cache'
        }
    }
    parameters {
        string(name: 'TRAIN_PARAMS', defaultValue: '', description: 'Parametry dla skryptu trenującego')
    }
    stages {
        stage('Pobierz dane') {
            steps {
                script {
                    // Skopiuj dane z poprzedniego projektu (s123456-create-dataset)
                    copyArtifacts(projectName: 's123456-create-dataset', filter: '*.csv', target: 'data', fingerprintArtifacts: true)
                }
            }
        }
        stage('Trenuj model') {
            steps {
                script {
                    // Uruchom skrypt trenujący z parametrami
                    sh 'python3 train.py ${params.TRAIN_PARAMS}'
                }
            }
        }
        stage('Zarchiwizuj model') {
            steps {
                archiveArtifacts artifacts: 'model.h5', fingerprint: true
            }
        }
    }
}