name: RIA Hub Workflow Demo

on:
  push:
    branches:
      [main]
  pull_request:
    branches:
      [main]

jobs:
  ria-demo:
    runs-on: ubuntu-latest-2080

    steps:
      - name: Print GPU information
        run: |
          if command -v nvidia-smi &> /dev/null; then
            echo "✅ NVIDIA GPU is available"
            nvidia-smi
          else
            echo "⚠️ No NVIDIA GPU found"
          fi

      - name: Checkout code
        uses: actions/checkout@v4
        with:
          lfs: true

      - name: Set up Python
        uses: actions/setup-python@v4
        with:
          python-version: "3.10"

      - name: Install dependencies
        run: |
          python -m pip install --upgrade pip
          pip install -r requirements.txt

      # - name: 1. Build HDF5 Dataset
      #   run: |
      #    mkdir -p data/dataset
      #    PYTHONPATH=. python data/scripts/produce_dataset.py
      #    echo "datasets produced successfully"
      #   shell: bash

      - name: 2. Train Model
        env:
            NO_NNPACK: 1
            PYTORCH_NO_NNPACK: 1
        run: |
          PYTHONPATH=. python data/training/train.py 2>/dev/null
          echo "training model"

      - name: 3. Build inference app
        run: |
          PYTHONPATH=. python convert_to_onnx.py
          echo "building inference app"

      - name: Upload Dataset Artifacts
        uses: actions/upload-artifact@v3
        with:
          name: ria-dataset
          path: data/dataset/**

      - name: Upload Checkpoints
        uses: actions/upload-artifact@v3
        with:
          name: ria-checkpoints
          path: checkpoint_files/inference_recognition_model.ckpt


      - name: Upload Inference App
        uses: actions/upload-artifact@v3
        with:
          name: ria-demo-app
          path: onnx_files/inference_recognition_model.onnx