From 0e0465dc6411e2d4f1f2cd268a60ff31ce5946c2 Mon Sep 17 00:00:00 2001
From: xiaohai-huang <jeromehuang18@gmail.com>
Date: Sun, 20 Aug 2023 18:30:58 +0800
Subject: [PATCH] Build and publish Docker image using GitHub workflow (#140)

* ci: add publish docker image workflow

* ci: move workflow file to the correct folder

* docs: update changelog

* ci: change on push branch to `main`

* ci: use variable instead of hard code docker username
---
 .github/workflows/publish-docker-image.yaml | 26 +++++++++++++++++++++
 README.md                                   |  6 +++++
 2 files changed, 32 insertions(+)
 create mode 100644 .github/workflows/publish-docker-image.yaml

diff --git a/.github/workflows/publish-docker-image.yaml b/.github/workflows/publish-docker-image.yaml
new file mode 100644
index 00000000..dcf750f7
--- /dev/null
+++ b/.github/workflows/publish-docker-image.yaml
@@ -0,0 +1,26 @@
+name: Docker Image CI
+
+on:
+  push:
+    branches: [main]
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v3
+
+      - name: Login to DockerHub
+        uses: docker/login-action@v2
+        with:
+          username: ${{ secrets.DOCKER_USERNAME }}
+          password: ${{ secrets.DOCKER_PASSWORD }}
+
+      - name: Build and Push the Docker image
+        uses: docker/build-push-action@v4
+        with:
+          context: .
+          push: true
+          tags: ${{ secrets.DOCKER_USERNAME }}/tts-generation-webui
diff --git a/README.md b/README.md
index 1baa68c6..ddd4b6cd 100644
--- a/README.md
+++ b/README.md
@@ -34,6 +34,11 @@ Google Colab demo: [![Open In Colab](https://colab.research.google.com/assets/co
 https://rsxdalv.github.io/bark-speaker-directory/
 
 ## Changelog
+
+
+Aug 18:
+* CI: Add a GitHub Action to automatically publish docker image.
+
 Aug 16:
 * Add "name" to tortoise generation parameters
 
@@ -60,6 +65,7 @@ Aug 7:
 Aug 6:
 * Fix audiogen + mbd error, add tortoise fix for colab 
 
+
 Aug 4:
 * Add MultiBandDiffusion option to MusicGen https://github.com/rsxdalv/tts-generation-webui/pull/109
 * MusicGen/AudioGen save tokens on generation as .npz files.