#!/bin/bash
export TZ=Asia/Shanghai

export OUTPUT_DIR=/opt/data/output
export IMAGES_DIR=/opt/data/images
export MODELS_PATH=/opt/data/models
export DEEPFACE_HOME=/opt/data/models
export FAISS_INDEX_DIR=/opt/data/faiss
export CELEBRITY_SOURCE_DIR=/opt/data/chinese_celeb_dataset
export GENDER_CONFIDENCE=1
export UPSCALE_SIZE=2
export AGE_CONFIDENCE=1.0
export DRAW_SCORE=true
export FACE_CONFIDENCE=0.7

export ENABLE_DDCOLOR=true
export ENABLE_GFPGAN=true
export ENABLE_REALESRGAN=true
export ENABLE_ANIME_STYLE=true
export ENABLE_RVM=true
export ENABLE_REMBG=true
export ENABLE_CLIP=false

export CLEANUP_INTERVAL_HOURS=1
export CLEANUP_AGE_HOURS=1

export BEAUTY_ADJUST_GAMMA=0.8
export BEAUTY_ADJUST_MIN=1.0
export BEAUTY_ADJUST_MAX=9.0
export ENABLE_ANIME_PRELOAD=true
export ENABLE_LOGGING=true
export BEAUTY_ADJUST_ENABLED=true

export RVM_LOCAL_REPO=/opt/data/models/RobustVideoMatting
export RVM_WEIGHTS_PATH=/opt/data/models/torch/hub/checkpoints/rvm_resnet50.pth
export RVM_MODEL=resnet50

export AUTO_INIT_GFPGAN=false
export AUTO_INIT_DDCOLOR=false
export AUTO_INIT_REALESRGAN=false
export AUTO_INIT_ANIME_STYLE=true
export AUTO_INIT_CLIP=false
export AUTO_INIT_RVM=false
export AUTO_INIT_REMBG=false

export ENABLE_WARMUP=true
export REALESRGAN_MODEL=realesr-general-x4v3
export CELEBRITY_FIND_THRESHOLD=0.87
export FEMALE_AGE_ADJUSTMENT=4

uvicorn app:app --workers 1 --loop asyncio --http httptools --host 0.0.0.0 --port 7860 --timeout-keep-alive 600