IconEval / app.py
hailey2024's picture
use data dir for db
f33202d
"""App for the IconEval interface."""
import os
import subprocess
from argparse import Namespace
from src.eval_server import app_main
REPO_ROOT = os.path.dirname(os.path.abspath(__file__))
DB_PATH = os.path.join(REPO_ROOT, 'data', 'eval_round_r1.db')
IMAGE_TEMPLATE = os.path.join(REPO_ROOT, 'images', '{label}', '{path}')
REDUNDANCY = int(os.environ['REDUNDANCY'])
ALLOWED_USERS_STR = os.environ['ALLOW_USERS']
ALLOWED_USERS = [u.strip() for u in ALLOWED_USERS_STR.split(',') if u.strip()]
ROUND_ID = 'r1'
AUTO_ALLO_NUM = int(os.environ['AUTO_ALLO_NUM'])
AUTO_COMMIT = int(os.environ.get('AUTO_COMMIT', '10'))
USER_EMAIL = os.getenv('USER_EMAIL', '[email protected]')
USER_NAME = os.getenv('USER_NAME', 'Janeding')
subprocess.run(['git', 'config', '--global', 'user.email', USER_EMAIL], check=True)
subprocess.run(['git', 'config', '--global', 'user.name', USER_NAME], check=True)
args = Namespace(
db_path=DB_PATH,
round_id=ROUND_ID,
redundancy=REDUNDANCY,
image_template=IMAGE_TEMPLATE,
allowed_users=ALLOWED_USERS,
auto_allo_num=AUTO_ALLO_NUM,
auto_commit=AUTO_COMMIT,
)
app_main(args)