Merge pull request #22 from ainize-team/refactor/donghoon/update-envs-and-docker-compose

refactor: update envs and docker compose

Author: DHBaek

.env.sample

@@ -1,4 +1,4 @@
 BROKER_BASE_URI=amqp://<user_name>:<password>@<rabbitmq_container_name>:5672
-DATABASE_URL=<filebase_database_url>
-STORAGE_BUCKET=<filebase_storage_bucket>
-APP_NAME=text-to-art
+FIREBASE_DATABASE_URL=<filebase_database_url>
+FIREBASE_STORAGE_BUCKET=<filebase_storage_bucket>
+FIREBASE_APP_NAME=text-to-art

README.md

@@ -46,8 +46,8 @@ docker build -t tti-worker .
 ```shell
 docker run -d --name <worker_container_name> \
 --gpus='"device=0"' -e BROKER_URI=<broker_uri> \
--e DATABASE_URL=<firebase_realtime_database_url> \
--e STORAGE_BUCKET=<firebase_storage_url> \
+-e FIREBASE_DATABASE_URL=<firebase_realtime_database_url> \
+-e FIREBASE_STORAGE_BUCKET=<firebase_storage_url> \
 -v <firebase_credential_path>:/app/key -v <model_local_path>:/app/model \
 tti-worker
 ```

docker-compose.yml

@@ -1,29 +1,6 @@
 version: '3'
 
 services:
-  stable-diffusion-v2:
-    build:
-      context: .
-      dockerfile: Dockerfile
-    container_name: stable-diffuion-v2-worker
-    env_file:
-      - .env
-    environment:
-      - VHOST_NAME=stable-diffusion-v2
-    volumes:
-      - <firebase_credential_path>:/app/key
-      - <model_local_path>:/app/model
-    networks:
-      - tti-fastapi_dev
-    restart: unless-stopped
-    deploy:
-      resources:
-        reservations:
-          devices:
-            - driver: nvidia
-              device_ids: [<gpu_id>]
-              capabilities: [gpu]
-
   stable-diffusion-v2-1:
     build:
       context: .

src/configs/config.py

@@ -15,9 +15,9 @@ class ModelSettings(BaseSettings):
 
 class FirebaseSettings(BaseSettings):
     cred_path: str = "/app/key/serviceAccountKey.json"
-    database_url: str
-    storage_bucket: str
-    app_name: str = "text-to-art"
+    firebase_database_url: str
+    firebase_storage_bucket: str
+    firebase_app_name: str = "text_to_art"
 
 
 celery_worker_settings = CeleryWorkerSettings()

src/utils.py

@@ -13,7 +13,7 @@
 from schemas import ImageGenerationResponse, ImageGenerationResult, ImageGenerationWorkerOutput
 
 
-app_name = firebase_settings.app_name
+app_name = firebase_settings.firebase_app_name
 
 
 def clear_memory() -> None:

src/worker.py

@@ -17,7 +17,7 @@
 firebase_admin.initialize_app(
     cred,
     {
-        "databaseURL": firebase_settings.database_url,
-        "storageBucket": firebase_settings.storage_bucket,
+        "databaseURL": firebase_settings.firebase_database_url,
+        "storageBucket": firebase_settings.firebase_storage_bucket,
     },
 )