- Getting containers ready for the cloud

eveai_workers/__init__.py

@@ -1,19 +1,26 @@
 import logging
 import logging.config
 from flask import Flask
+import os
 
 from common.utils.celery_utils import make_celery, init_celery
 from common.extensions import db, minio_client
 from config.logging_config import LOGGING
+from config.config import get_config
 
 
 def create_app(config_file=None):
     app = Flask(__name__)
 
-    if config_file is None:
-        app.config.from_object('config.config.DevConfig')
-    else:
-        app.config.from_object(config_file)
+    environment = os.getenv('FLASK_ENV', 'development')
+
+    match environment:
+        case 'development':
+            app.config.from_object(get_config('dev'))
+        case 'production':
+            app.config.from_object(get_config('prod'))
+        case _:
+            app.config.from_object(get_config('dev'))
 
     logging.config.dictConfig(LOGGING)
     app.embed_tuning_logger = logging.getLogger('embed_tuning')