serve/constants.py

import os

LOGDIR = os.getenv("LOGDIR", "./GenAI-Arena-hf-logs/vote_log")
IMAGE_DIR = os.getenv("IMAGE_DIR", f"{LOGDIR}/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", "https://tigerai.ca")
LOG_SERVER_SUBDOMAIN = os.getenv("LOG_SERVER_SUBDOMAIN", "/GenAI-Arena-hf-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"

IMAGE_GENERATION_MODELS = ['fal_LCM(v1.5/XL)_text2image','fal_SDXLTurbo_text2image','fal_SDXL_text2image', 'imagenhub_PixArtAlpha_generation', 'fal_PixArtSigma_text2image',
                            'imagenhub_OpenJourney_generation','fal_SDXLLightning_text2image', 'fal_StableCascade_text2image',
                            'playground_PlayGroundV2_generation', 'playground_PlayGroundV2.5_generation']
IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
                        'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition',
                        'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition',
                        'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition']
VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
                           'fal_AnimateDiffTurbo_text2video',
                           'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
                           'videogenhub_ModelScope_generation', 'videogenhub_OpenSora_generation']