From c94dfa10e7969633b2d0f39435f929daeb76beff Mon Sep 17 00:00:00 2001
From: ziesorx <nwisessan@hotmail.com>
Date: Wed, 24 Sep 2025 22:35:24 +0700
Subject: [PATCH] chore: update CICD

---
 .gitea/workflows/build.yml |  6 +++---
 Dockerfile.base            | 13 +++++++++++--
 requirements.txt           |  1 -
 3 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/.gitea/workflows/build.yml b/.gitea/workflows/build.yml
index 585009f..316c4dc 100644
--- a/.gitea/workflows/build.yml
+++ b/.gitea/workflows/build.yml
@@ -51,7 +51,7 @@ jobs:
           registry: git.siwatsystem.com
           username: ${{ github.actor }}
           password: ${{ secrets.RUNNER_TOKEN }}
-        
+
       - name: Build and push base Docker image
         uses: docker/build-push-action@v4
         with:
@@ -79,7 +79,7 @@ jobs:
           registry: git.siwatsystem.com
           username: ${{ github.actor }}
           password: ${{ secrets.RUNNER_TOKEN }}
-        
+
       - name: Build and push Docker image
         uses: docker/build-push-action@v4
         with:
@@ -109,4 +109,4 @@ jobs:
           else
             echo "Deploying staging stack..."
             ssh -i ~/.ssh/id_rsa ${{ vars.DEPLOY_USER_CMS }}@${{ vars.DEPLOY_HOST_CMS }} "cd ~/cms-system-k8s && docker compose -f docker-compose.staging.yml pull && docker compose -f docker-compose.staging.yml up -d"
-          fi
\ No newline at end of file
+          fi
diff --git a/Dockerfile.base b/Dockerfile.base
index 3700920..60999b1 100644
--- a/Dockerfile.base
+++ b/Dockerfile.base
@@ -1,8 +1,17 @@
 # Base image with all ML dependencies
-FROM python:3.13-bookworm
+FROM pytorch/pytorch:2.8.0-cuda12.6-cudnn9-runtime
 
 # Install system dependencies
-RUN apt update && apt install -y libgl1 && rm -rf /var/lib/apt/lists/*
+RUN apt update && apt install -y \
+    libgl1 \
+    libglib2.0-0 \
+    libgstreamer1.0-0 \
+    libgtk-3-0 \
+    libavcodec58 \
+    libavformat58 \
+    libswscale5 \
+    libgomp1 \
+    && rm -rf /var/lib/apt/lists/*
 
 # Copy and install base requirements (ML dependencies that rarely change)
 COPY requirements.base.txt .
diff --git a/requirements.txt b/requirements.txt
index 256c766..034d18e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,6 +4,5 @@ websockets
 fastapi[standard]
 redis
 urllib3<2.0.0
-opencv-python
 numpy
 requests
\ No newline at end of file