constants.py

import os
from pathlib import Path

os.chdir(Path.cwd())

LOGDIR = os.getenv("LOGDIR", "./3DGen-Arena-logs/vote_log")
IMAGE_DIR = os.getenv("IMAGE_DIR", f"{LOGDIR}/images")
OFFLINE_DIR = "./offline"
OFFLINE_GIF_DIR = os.path.join(OFFLINE_DIR, "gifs")

SERVER_PORT = os.getenv("SERVER_PORT", 7860)
ROOT_PATH = os.getenv("ROOT_PATH", None)
ELO_RESULTS_DIR = os.getenv("ELO_RESULTS_DIR", "./arena_elo/results/latest")

LOG_SERVER = os.getenv("LOG_SERVER", "http://2.0.0.1:8800")
LOG_SERVER_SUBDOAMIN = os.getenv("LOG_SERVER_SUBDIR", "3DGen-Arena-hf-logs")
LOG_SERVER_ADDR = os.getenv("LOG_SERVER_ADDR", f"{LOG_SERVER}/{LOG_SERVER_SUBDOAMIN}")
# LOG SERVER API ENDPOINTS
APPEND_JSON = "append_json"
SAVE_IMAGE = "save_image"
SAVE_LOG = "save_log"

NUM_SIDES = 2
EVALUATE_DIMS = 5
TEXT_PROMPT_PATH = "offline/prompts_110.json"
IMAGE_PROMPT_PATH = "offline/image_urls.txt"

MAX_ATTEMPTS = 5
REPLICATE_API_TOKEN = os.getenv("REPLICATE_API_TOKEN", "r8_0BaoQW0G8nWFXY8YWBCCUDurANxCtY72rarv9")