pipeline {
    agent any

    stages {
        stage('Clone repository') {
            steps {
                checkout scm
            }
        }

        stage('Copy artifacts') {
            agent {
                dockerfile {
                    filename 'Dockerfile'
                    reuseNode true
                }
            }

            steps {
                copyArtifacts(projectName: 'z-s464863-create-dataset', filter: 'datasets/*', selector: lastSuccessful())
            }
        }

        stage('Create model') {
            agent {
                dockerfile {
                    filename 'Dockerfile'
                    reuseNode true
                }
            }

            steps {
                sh "chmod +x ./create_model.py"
                sh "python3 ./create_model.py"
                archiveArtifacts artifacts: 'models/model.pth', onlyIfSuccessful: true
            }
        }
    }
}