txt2img-consumer / Helpers.py
akinv's picture
Fix
cb83b1e
from dotenv import load_dotenv
import os
import sentry_sdk
from natsort import natsorted
from glob import glob
load_dotenv()
def name_formatter(name):
Tr2Eng = name.maketrans("çğıöşüÇĞİÖŞÜ", "cgiosucgiosu")
return os.getenv('NAME_PREFIX') + name.replace(' ', '').translate(Tr2Eng).lower()
def weights_dir():
return natsorted(glob(os.getenv('OUTPUT_DIR') + os.sep + "*"))[-1]
def capture_message(data):
sentry_sdk.init(
dsn="https://3ff299ace6234e20a444c0e5fb6f7cbd@o4504356072128512.ingest.sentry.io/4504356073308160",
traces_sample_rate=1.0
)
sentry_sdk.capture_message(data)