From 280291f2aa8b47712d9ea6b2f71fbed969fc7dd2 Mon Sep 17 00:00:00 2001
From: remsky <jeremy.braun@ucalgary.ca>
Date: Sat, 4 Jan 2025 20:59:30 -0700
Subject: [PATCH] ci: update Docker configurations to use pre-built images and
 UI services

---
 .github/workflows/docker-publish.yml |  3 +++
 docker-compose.cpu.yml               | 14 +++++++++-----
 docker-compose.yml                   | 12 ++++++++----
 3 files changed, 20 insertions(+), 9 deletions(-)

diff --git a/.github/workflows/docker-publish.yml b/.github/workflows/docker-publish.yml
index 5a2b3af..138fd7c 100644
--- a/.github/workflows/docker-publish.yml
+++ b/.github/workflows/docker-publish.yml
@@ -38,6 +38,7 @@ jobs:
             type=semver,pattern={{version}}
             type=semver,pattern={{major}}.{{minor}}
             type=semver,pattern={{major}}
+            type=raw,value=latest
 
       # Extract metadata for CPU image
       - name: Extract metadata (tags, labels) for CPU Docker
@@ -51,6 +52,7 @@ jobs:
             type=semver,pattern={{version}}
             type=semver,pattern={{major}}.{{minor}}
             type=semver,pattern={{major}}
+            type=raw,value=latest
 
       # Build and push GPU version
       - name: Build and push GPU Docker image
@@ -86,6 +88,7 @@ jobs:
             type=semver,pattern={{version}}
             type=semver,pattern={{major}}.{{minor}}
             type=semver,pattern={{major}}
+            type=raw,value=latest
 
       # Build and push UI version
       - name: Build and push UI Docker image
diff --git a/docker-compose.cpu.yml b/docker-compose.cpu.yml
index f44f2d4..f3975df 100644
--- a/docker-compose.cpu.yml
+++ b/docker-compose.cpu.yml
@@ -26,9 +26,11 @@ services:
       start_period: 1s
 
   kokoro-tts:
-    build:
-      context: .
-      dockerfile: Dockerfile.cpu
+    image: ghcr.io/remsky/kokoro-fastapi:v0.0.2-cpu
+    # Uncomment below to build from source instead of using the released image
+    # build:
+    #   context: .
+    #   dockerfile: Dockerfile.cpu
     volumes:
       - ./api/src:/app/api/src
       - ./Kokoro-82M:/app/Kokoro-82M
@@ -49,8 +51,10 @@ services:
 
   # Gradio UI service [Comment out everything below if you don't need it]
   gradio-ui:
-    build:
-      context: ./ui
+    image: ghcr.io/remsky/kokoro-fastapi:v0.0.2-ui
+    # Uncomment below to build from source instead of using the released image
+    # build:
+    #   context: ./ui
     ports:
       - "7860:7860"
     volumes:
diff --git a/docker-compose.yml b/docker-compose.yml
index 2e7a86f..46ec691 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -26,8 +26,10 @@ services:
       start_period: 1s
 
   kokoro-tts:
-    build:
-      context: .
+    image: ghcr.io/remsky/kokoro-fastapi:v0.0.2
+    # Uncomment below to build from source instead of using the released image
+    # build:
+    #   context: .
     volumes:
       - ./api/src:/app/api/src
       - ./Kokoro-82M:/app/Kokoro-82M
@@ -48,8 +50,10 @@ services:
 
   # Gradio UI service [Comment out everything below if you don't need it]
   gradio-ui:
-    build:
-      context: ./ui
+    image: ghcr.io/remsky/kokoro-fastapi:v0.0.2-ui
+    # Uncomment below to build from source instead of using the released image
+    # build:
+    #   context: ./ui
     ports:
       - "7860:7860"
     volumes: