From 3ca79c609f4239d3982504d9fb69828e72efb051 Mon Sep 17 00:00:00 2001
From: remsky <jeremy.braun@ucalgary.ca>
Date: Thu, 2 Jan 2025 01:59:25 -0700
Subject: [PATCH] fix: longer timeouts, fix on hf model pull

---
 .github/workflows/docker-publish.yml |  2 --
 docker-compose.cpu.yml               | 20 ++++++++++++++++----
 docker-compose.yml                   |  8 ++++----
 3 files changed, 20 insertions(+), 10 deletions(-)

diff --git a/.github/workflows/docker-publish.yml b/.github/workflows/docker-publish.yml
index bcb8bdd..94b9dae 100644
--- a/.github/workflows/docker-publish.yml
+++ b/.github/workflows/docker-publish.yml
@@ -72,7 +72,6 @@ jobs:
           tags: ${{ steps.meta-gpu.outputs.tags }}
           labels: ${{ steps.meta-gpu.outputs.labels }}
           platforms: linux/amd64
-          target: runtime
 
       # Build and push CPU version
       - name: Build and push CPU Docker image
@@ -84,7 +83,6 @@ jobs:
           tags: ${{ steps.meta-cpu.outputs.tags }}
           labels: ${{ steps.meta-cpu.outputs.labels }}
           platforms: linux/amd64
-          target: runtime
 
   create-release:
     needs: build
diff --git a/docker-compose.cpu.yml b/docker-compose.cpu.yml
index 2daeb46..faea2fe 100644
--- a/docker-compose.cpu.yml
+++ b/docker-compose.cpu.yml
@@ -8,7 +8,7 @@ services:
       sh -c "
         rm -f .git/index.lock;
         if [ -z \"$(ls -A .)\" ]; then
-          git clone https://huggingface.co/hexgrad/Kokoro-82M
+          git clone https://huggingface.co/hexgrad/Kokoro-82M .
           touch .cloned;
         else
           rm -f .git/index.lock && \
@@ -20,9 +20,9 @@ services:
       "
     healthcheck:
       test: ["CMD", "test", "-f", ".cloned"]
-      interval: 3s
-      timeout: 1s
-      retries: 120
+      interval: 5s
+      timeout: 2s
+      retries: 300
       start_period: 1s
 
   kokoro-tts:
@@ -39,3 +39,15 @@ services:
     depends_on:
       model-fetcher:
         condition: service_healthy
+
+  # Gradio UI service [Comment out everything below if you don't need it]
+  gradio-ui:
+    build:
+      context: ./ui
+    ports:
+      - "7860:7860"
+    volumes:
+      - ./ui/data:/app/ui/data
+      - ./ui/app.py:/app/app.py  # Mount app.py for hot reload
+    environment:
+      - GRADIO_WATCH=True  # Enable hot reloading
diff --git a/docker-compose.yml b/docker-compose.yml
index 2722208..2e7a86f 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -8,7 +8,7 @@ services:
       sh -c "
         rm -f .git/index.lock;
         if [ -z \"$(ls -A .)\" ]; then
-          git clone https://huggingface.co/hexgrad/Kokoro-82M
+          git clone https://huggingface.co/hexgrad/Kokoro-82M .
           touch .cloned;
         else
           rm -f .git/index.lock && \
@@ -20,9 +20,9 @@ services:
       "
     healthcheck:
       test: ["CMD", "test", "-f", ".cloned"]
-      interval: 3s
-      timeout: 1s
-      retries: 120
+      interval: 5s
+      timeout: 2s
+      retries: 300
       start_period: 1s
 
   kokoro-tts: