PyTorch Support and GitHub Actions

.dockerignore

@@ -0,0 +1,2 @@
+.venv
+data

.github/workflows/README.md

@@ -0,0 +1,64 @@
+# GitHub Workflows
+
+This directory contains GitHub Actions workflows for the No-Code Classification Toolkit.
+
+## Docker Build and Push Workflow
+
+**File:** `docker-build-push.yml`
+
+### Purpose
+Automatically builds and pushes Docker images to GitHub Container Registry (ghcr.io) whenever code is pushed to the repository.
+
+### Triggers
+The workflow runs on:
+- **Push to main branch**: Builds and pushes images with `latest` tag
+- **Push tags matching `v*`**: Builds and pushes versioned releases (e.g., `v1.0.0`)
+- **Pull requests to main**: Builds images for testing (does not push)
+- **Manual trigger**: Can be triggered manually via GitHub Actions UI
+
+### Docker Image Tags
+The workflow creates multiple tags for each build:
+
+- `latest` - Latest build from the main branch
+- `main` - Latest build from the main branch
+- `v1.2.3` - Semantic version tags (for tagged releases)
+- `v1.2` - Major.minor version tags
+- `v1` - Major version tags
+- `main-<sha>` - Branch name with commit SHA
+
+### Docker Registry
+Images are pushed to GitHub Container Registry:
+```
+ghcr.io/animikhaich/no-code-classification-toolkit
+```
+
+### Usage
+
+#### Pull the latest image:
+```bash
+docker pull ghcr.io/animikhaich/no-code-classification-toolkit:latest
+```
+
+#### Pull a specific version:
+```bash
+docker pull ghcr.io/animikhaich/no-code-classification-toolkit:v1.0.0
+```
+
+#### Run the container:
+```bash
+docker run -it --runtime nvidia --net host -v /path/to/dataset:/data ghcr.io/animikhaich/no-code-classification-toolkit:latest
+```
+
+### Permissions
+The workflow requires:
+- `contents: read` - To checkout the repository
+- `packages: write` - To push images to GitHub Container Registry
+
+### Features
+- **Docker Buildx**: Uses BuildKit for efficient multi-platform builds
+- **Layer caching**: Leverages GitHub Actions cache for faster builds
+- **Metadata extraction**: Automatically generates Docker labels and tags
+- **Security**: Uses `GITHUB_TOKEN` for authentication (no manual secrets needed)
+
+### Monitoring
+You can monitor workflow runs in the [Actions tab](https://github.com/animikhaich/No-Code-Classification-Toolkit/actions) of the repository.

.github/workflows/docker-build-push.yml

@@ -0,0 +1,90 @@
+name: Build and Push Docker Images
+
+on:
+    push:
+        branches:
+            - main
+        tags:
+            - "v*"
+    pull_request:
+        branches:
+            - main
+    workflow_dispatch:
+
+env:
+    REGISTRY: ghcr.io
+    IMAGE_NAME: ${{ github.repository }}
+
+jobs:
+    build-and-push:
+        runs-on: ubuntu-latest
+        permissions:
+            contents: read
+            packages: write
+
+        steps:
+            - name: Checkout repository
+              uses: actions/checkout@v4
+
+            - name: Set up Docker Buildx
+              uses: docker/setup-buildx-action@v3
+
+            - name: Log in to GitHub Container Registry
+              if: github.event_name != 'pull_request'
+              uses: docker/login-action@v3
+              with:
+                  registry: ${{ env.REGISTRY }}
+                  username: ${{ github.actor }}
+                  password: ${{ secrets.GITHUB_TOKEN }}
+
+            # TensorFlow image
+            - name: Extract metadata (tensorflow)
+              id: meta-tensorflow
+              uses: docker/metadata-action@v5
+              with:
+                  images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}-tensorflow
+                  tags: |
+                      type=ref,event=branch
+                      type=ref,event=pr
+                      type=semver,pattern={{version}}
+                      type=semver,pattern={{major}}.{{minor}}
+                      type=semver,pattern={{major}}
+                      type=sha,prefix={{branch}}-
+                      type=raw,value=latest,enable={{is_default_branch}}
+
+            - name: Build and push TensorFlow image
+              uses: docker/build-push-action@v5
+              with:
+                  context: .
+                  file: Dockerfile.tensorflow
+                  push: ${{ github.event_name != 'pull_request' }}
+                  tags: ${{ steps.meta-tensorflow.outputs.tags }}
+                  labels: ${{ steps.meta-tensorflow.outputs.labels }}
+                  cache-from: type=gha
+                  cache-to: type=gha,mode=max
+
+            # PyTorch image
+            - name: Extract metadata (pytorch)
+              id: meta-pytorch
+              uses: docker/metadata-action@v5
+              with:
+                  images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}-pytorch
+                  tags: |
+                      type=ref,event=branch
+                      type=ref,event=pr
+                      type=semver,pattern={{version}}
+                      type=semver,pattern={{major}}.{{minor}}
+                      type=semver,pattern={{major}}
+                      type=sha,prefix={{branch}}-
+                      type=raw,value=latest,enable={{is_default_branch}}
+
+            - name: Build and push PyTorch image
+              uses: docker/build-push-action@v5
+              with:
+                  context: .
+                  file: Dockerfile.pytorch
+                  push: ${{ github.event_name != 'pull_request' }}
+                  tags: ${{ steps.meta-pytorch.outputs.tags }}
+                  labels: ${{ steps.meta-pytorch.outputs.labels }}
+                  cache-from: type=gha
+                  cache-to: type=gha,mode=max

.gitignore

@@ -143,4 +143,7 @@ dmypy.json
 model/
 logs*
 old_data/
-test*
+test*
+
+
+data

Dockerfile.both

@@ -0,0 +1,23 @@
+FROM astral/uv:python3.12-bookworm-slim
+
+ARG DEBIAN_FRONTEND=noninteractive
+
+# Install Python and necessary tools
+RUN apt-get update && apt-get install -y \
+    parallel \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create app directory
+RUN mkdir /app
+WORKDIR /app
+
+# Copy requirements and install dependencies
+COPY ./requirements-tensorflow.txt /app
+COPY ./requirements-pytorch.txt /app
+RUN uv pip install --no-cache-dir -r requirements-tensorflow.txt --system
+RUN uv pip install --no-cache-dir -r requirements-pytorch.txt --system
+
+# Copy application files
+COPY . /app
+
+CMD ["sh", "launch.sh"]

Dockerfile.pytorch

@@ -0,0 +1,15 @@
+FROM astral/uv:python3.12-bookworm-slim
+
+ARG DEBIAN_FRONTEND=noninteractive
+
+RUN apt-get update
+RUN apt-get install -y parallel
+
+RUN mkdir /app
+WORKDIR /app
+COPY ./requirements-pytorch.txt /app
+
+RUN uv pip install -r requirements-pytorch.txt --system
+COPY . /app
+
+CMD ["sh", "launch.sh"]

Dockerfile.tensorflow

@@ -0,0 +1,15 @@
+FROM astral/uv:python3.12-bookworm-slim
+
+ARG DEBIAN_FRONTEND=noninteractive
+
+RUN apt-get update
+RUN apt-get install -y parallel
+
+RUN mkdir /app
+WORKDIR /app
+COPY ./requirements-tensorflow.txt /app
+
+RUN uv pip install -r requirements-tensorflow.txt --system
+COPY . /app
+
+CMD ["sh", "launch.sh"]