diff --git a/docker/cpu/docker-compose.yml b/docker/cpu/docker-compose.yml
index 6e89c76..a545122 100644
--- a/docker/cpu/docker-compose.yml
+++ b/docker/cpu/docker-compose.yml
@@ -35,3 +35,5 @@ services:
       - GRADIO_WATCH=True  # Enable hot reloading
       - PYTHONUNBUFFERED=1  # Ensure Python output is not buffered
       - DISABLE_LOCAL_SAVING=false  # Set to 'true' to disable local saving and hide file view
+      - API_HOST=kokoro-tts  # Set TTS service URL
+      - API_PORT=8880  # Set TTS service PORT
diff --git a/ui/Dockerfile b/ui/Dockerfile
index 0266b19..e1726fb 100644
--- a/ui/Dockerfile
+++ b/ui/Dockerfile
@@ -11,5 +11,8 @@ RUN mkdir -p data/inputs data/outputs
 # Copy the application files
 COPY . .
 
+ENV API_HOST=kokoro-tts
+ENV API_PORT=8880
+
 # Run the Gradio app
 CMD ["python", "app.py"]
diff --git a/ui/lib/config.py b/ui/lib/config.py
index 2a4aeb7..406cdf0 100644
--- a/ui/lib/config.py
+++ b/ui/lib/config.py
@@ -1,7 +1,9 @@
 import os
 
 # API Configuration
-API_URL = "http://kokoro-tts:8880"
+API_HOST = os.getenv("API_HOST", "kokoro-tts")
+API_PORT = os.getenv("API_PORT", "8880")
+API_URL = f"http://{API_HOST}:{API_PORT}"
 
 # File paths
 INPUTS_DIR = "/app/ui/data/inputs"