constant.py

serve/constants.py

@@ -0,0 +1,63 @@
+import os
+
+LOGDIR = os.getenv("LOGDIR", "./ksort-logs/vote_log")
+IMAGE_DIR = os.getenv("IMAGE_DIR", f"{LOGDIR}/images")
+KSORT_IMAGE_DIR = os.getenv("KSORT_IMAGE_DIR", f"{LOGDIR}/ksort_images")
+VIDEO_DIR = os.getenv("VIDEO_DIR", f"{LOGDIR}/videos")
+
+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://127.0.0.1:22005")
+LOG_SERVER_SUBDOMAIN = os.getenv("LOG_SERVER_SUBDOMAIN", "/ksort-logs")
+LOG_SERVER_ADDR = os.getenv("LOG_SERVER_ADDR", f"{LOG_SERVER}{LOG_SERVER_SUBDOMAIN}")
+# LOG SERVER API ENDPOINTS
+APPEND_JSON = "append_json"
+SAVE_IMAGE = "save_image"
+SAVE_VIDEO = "save_video"
+SAVE_LOG = "save_log"
+
+SSH_MSCOCO = os.getenv("SSH_MSCOCO", "/root/MSCOCO_prompt")
+
+
+SSH_SERVER = os.getenv("SSH_SERVER", "default_value")
+SSH_PASSWORD = os.getenv("SSH_PASSWORD", "default_value")
+SSH_PORT = os.getenv("SSH_PORT", "default_value")
+SSH_USER = os.getenv("SSH_USER", "default_value")
+SSH_LOG = os.getenv("SSH_LOG", "/home/zhendongucb/ksort/ksort_log/log")
+SSH_VIDEO_LOG = os.getenv("SSH_LOG", "/home/zhendongucb/ksort/ksort_log/video_log")
+SSH_SKILL = os.getenv("SSH_SKILL", "/home/zhendongucb/ksort/TrueSkill/trueskill_data.json")
+SSH_VIDEO_SKILL = os.getenv("SSH_VIDEO_SKILL", "/home/zhendongucb/ksort/TrueSkill/trueskill_video_data.json")
+
+SSH_CACHE_OPENSOURCE = os.getenv("SSH_CACHE_OPENSOURCE", "/home/zhendongucb/ksort/ksort_video_cache/Kaiyuan/")
+SSH_CACHE_ADVANCE = os.getenv("SSH_CACHE_ADVANCE", "/home/zhendongucb/ksort/ksort_video_cache/Advance/")
+SSH_CACHE_PIKA = os.getenv("SSH_CACHE_PIKA", "/home/zhendongucb/ksort/ksort_video_cache/Pika-Beta/")
+SSH_CACHE_SORA = os.getenv("SSH_CACHE_SORA", "/home/zhendongucb/ksort/ksort_video_cache/Sora/")
+
+SSH_CACHE_IMAGE = os.getenv("SSH_CACHE_IMAGE", "/home/zhendongucb/ksort/ksort_image_cache/")
+
+# COLOR1=(128, 214, 255)
+# COLOR2=(237, 247, 152)
+# COLOR3=(250, 181, 122)
+# COLOR4=(240, 104, 104)
+
+# COLOR1=(112, 161, 215)
+# COLOR2=(161, 222, 147)
+# COLOR3=(247, 244, 139)
+# COLOR4=(244, 124, 124)
+
+COLOR1=(168, 230, 207)
+COLOR2=(253, 255, 171)
+COLOR3=(255, 211, 182)
+COLOR4=(255, 170, 165)
+
+# COLOR1=(255, 212, 96)
+# COLOR2=(240, 123, 63)
+# COLOR3=(234, 84, 85)
+# COLOR4=(45, 64, 89)
+
+# COLOR1=(255, 189, 57)
+# COLOR2=(230, 28, 93)
+# COLOR3=(147, 0, 119)
+# COLOR4=(58, 0, 136)