File size: 1,016 Bytes
6165ba9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | import os
from pathlib import Path
import tomllib
# Base Directory Setup
BASE_DIR = Path(__file__).resolve().parent
OUTPUT_DIR = os.getenv("AIBOM_OUTPUT_DIR") or "/tmp/aibom_output"
# Ensure absolute path for security
if not os.path.isabs(OUTPUT_DIR):
OUTPUT_DIR = os.path.abspath(OUTPUT_DIR)
def get_project_metadata() -> tuple[str, str]:
try:
pyproject_path = BASE_DIR.parent / "pyproject.toml"
with open(pyproject_path, "rb") as f:
data = tomllib.load(f)
return data["project"]["name"], data["project"]["version"]
except Exception:
return "owasp-aibom-generator", "1.0.2"
AIBOM_GEN_NAME, AIBOM_GEN_VERSION = get_project_metadata()
TEMPLATES_DIR = BASE_DIR / "templates"
# Cleanup Configuration
MAX_AGE_DAYS = 7
MAX_FILES = 1000
CLEANUP_INTERVAL = 100
# Hugging Face Setup
HF_REPO = "owasp-genai-security-project/aisbom-usage-log"
HF_TOKEN = os.getenv("HF_TOKEN")
RECAPTCHA_SITE_KEY = os.getenv("RECAPTCHA_SITE_KEY")
|