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

docker-compose.yml

@@ -2,9 +2,7 @@ version: '3'
 
 services:
   stable-diffusion-v2:
-    build:
-      context: .
-      dockerfile: Dockerfile
+    image: devainetwork/tti-worker
     container_name: stable-diffuion-v2-worker
     env_file:
       - .env
@@ -25,9 +23,7 @@ services:
               capabilities: [gpu]
 
   stable-diffusion-v2-1:
-    build:
-      context: .
-      dockerfile: Dockerfile
+    image: devainetwork/tti-worker
     container_name: stable-diffuion-v2-1-worker
     env_file:
       - .env

src/configs/config.py

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

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,
     },
 )