Create constrants.py

constrants.py

@@ -0,0 +1,28 @@
+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", "https://tigerai.ca")
+LOG_SERVER_SUBDOAMIN = os.getenv("LOG_SERVER_SUBDIR", "GenAI-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
+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")