120 lines
3.1 KiB
Python
120 lines
3.1 KiB
Python
import os
|
|
|
|
import pika
|
|
from dotenv import load_dotenv
|
|
from pydantic import BaseSettings
|
|
|
|
BASE_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), '../../'))
|
|
load_dotenv(os.path.join(BASE_DIR, '.env'))
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
PROJECT_NAME = os.getenv('PROJECT_NAME', 'FASTAPI BASE')
|
|
SECRET_KEY = os.getenv('SECRET_KEY', '')
|
|
API_PREFIX = ''
|
|
BACKEND_CORS_ORIGINS = ['*']
|
|
DATABASE_URL = os.getenv('SQL_DATABASE_URL', '')
|
|
ACCESS_TOKEN_EXPIRE_SECONDS: int = 60 * 60 * 24 * 7 # Token expired after 7 days
|
|
SECURITY_ALGORITHM = 'HS256'
|
|
LOGGING_CONFIG_FILE = os.path.join(BASE_DIR, 'logging_env.py')
|
|
|
|
|
|
DEBUG = False
|
|
if DEBUG:
|
|
LOGS_PATH = "logs/"
|
|
CATEGORY_PATH = "service/attribute/config/descriptor/category/category_dis.csv"
|
|
FACE_CLASSIFIER = "service/generate_image/utils/haarcascade_frontalface_alt.xml"
|
|
else:
|
|
LOGS_PATH = "app/logs/"
|
|
CATEGORY_PATH = "app/service/attribute/config/descriptor/category/category_dis.csv"
|
|
FACE_CLASSIFIER = 'app/service/generate_image/utils/haarcascade_frontalface_alt.xml'
|
|
|
|
# RABBITMQ_ENV = "" # 生产环境
|
|
# RABBITMQ_ENV = "-dev" # 开发环境
|
|
RABBITMQ_ENV = "-local" # 本地测试环境
|
|
|
|
settings = Settings()
|
|
|
|
# minio 配置
|
|
MINIO_URL = "www.minio.aida.com.hk:9000"
|
|
MINIO_ACCESS = 'vXKFLSJkYeEq2DrSZvkB'
|
|
MINIO_SECRET = 'uKTZT3x7C43WvPN9QTc99DiRkwddWZrG9Uh3JVlR'
|
|
MINIO_SECURE = True
|
|
|
|
# redis 配置
|
|
REDIS_HOST = "10.1.1.240"
|
|
REDIS_PORT = "6379"
|
|
REDIS_DB = "2"
|
|
|
|
# rabbitmq config
|
|
RABBITMQ_PARAMS = {
|
|
"host": "18.167.251.121",
|
|
"port": 5672,
|
|
"credentials": pika.credentials.PlainCredentials(username='rabbit', password='123456'),
|
|
"virtual_host": "/"
|
|
}
|
|
|
|
# milvus 配置
|
|
MILVUS_DB_HOST = "10.1.1.240"
|
|
MILVUS_ALIAS = "default"
|
|
MILVUS_PORT = "19530"
|
|
MILVUS_TABLE_KEYPOINT = "keypoint_cache"
|
|
MILVUS_TABLE_SEG = "seg_cache"
|
|
|
|
# attribute service config
|
|
ATT_TRITON_URL = "10.1.1.240:8020"
|
|
|
|
# SR service config
|
|
SR_MODEL_NAME = "super_resolution"
|
|
SR_TRITON_URL = "10.1.1.240:10031"
|
|
SR_MINIO_BUCKET = "aida-users"
|
|
SR_RABBITMQ_QUEUES = os.getenv("SR_RABBITMQ_QUEUES", f"SuperResolution{RABBITMQ_ENV}")
|
|
|
|
# GenerateImage service config
|
|
GI_MODEL_NAME = 'stable_diffusion_xl'
|
|
GI_MODEL_URL = '10.1.1.240:10041'
|
|
GI_MINIO_BUCKET = "aida-users"
|
|
GI_RABBITMQ_QUEUES = os.getenv("GI_RABBITMQ_QUEUES", f"GenerateImage{RABBITMQ_ENV}")
|
|
GI_SYS_IMAGE_URL = "aida-sys-image/generate_image/white_image.jpg"
|
|
|
|
# SEG service config
|
|
SEG_MODEL_URL = '10.1.1.240:10000'
|
|
SEGMENTATION = {
|
|
"name": "seg_ocrnet_hr18",
|
|
"input": "seg_input__0",
|
|
"output": "seg_output__0",
|
|
}
|
|
|
|
# DESIGN config
|
|
DESIGN_MODEL_URL = '10.1.1.240:9000'
|
|
|
|
AIDA_CLOTHING = "aida-clothing"
|
|
|
|
# 优先级
|
|
PRIORITY_DICT = {
|
|
'earring_front': 99,
|
|
'bag_front': 98,
|
|
'hairstyle_front': 97,
|
|
'outwear_front': 20,
|
|
'tops_front': 19,
|
|
'dress_front': 18,
|
|
'blouse_front': 17,
|
|
'skirt_front': 16,
|
|
'trousers_front': 15,
|
|
'bottoms_front': 14,
|
|
'shoes_right': 1,
|
|
'shoes_left': 1,
|
|
'body': 0,
|
|
'bottoms_back': -14,
|
|
'trousers_back': -15,
|
|
'skirt_back': -16,
|
|
'blouse_back': -17,
|
|
'dress_back': -18,
|
|
'tops_back': -19,
|
|
'outwear_back': -20,
|
|
'hairstyle_back': -97,
|
|
'bag_back': -98,
|
|
'earring_back': -99,
|
|
}
|
|
|