pipeline {
  agent { 
    dockerfile true 
  }

  triggers {
    upstream(upstreamProjects: 'z-s464913-create-dataset', threshold: hudson.model.Result.SUCCESS)
  }
  
  parameters {
    buildSelector(
      defaultSelector: lastSuccessful(),
      description: 'Which build to use for copying artifacts',
      name: 'BUILD_SELECTOR'
    )
    string(name: 'LEARNING_RATE', defaultValue: '0.001', description: 'Learning rate')
    string(name: 'EPOCHS', defaultValue: '5', description: 'Number of epochs')
  }

  stages {
    stage('Clone Repository') {
      steps {
        git branch: 'training', url: 'https://git.wmi.amu.edu.pl/s464913/ium_464913.git'
      }
    }

    stage('Copy Artifacts') {
      steps {
        copyArtifacts filter: 'data/*', projectName: 'z-s464913-create-dataset', selector: buildParameter('BUILD_SELECTOR')
      }
    }

    stage('Run train_model script') {
      steps {
        sh 'chmod +x train_model.py'
        sh "python3 ./train_model.py ${params.LEARNING_RATE} ${params.EPOCHS}"
      }
    }

    stage('Archive Artifacts') {
      steps {
        archiveArtifacts artifacts: 'model/*', onlyIfSuccessful: true
      }
    }
  }
}