|
|
|
|
|
|
|
|
import os |
|
|
from pathlib import Path |
|
|
from dotenv import load_dotenv |
|
|
|
|
|
|
|
|
dotenv_path = Path(__file__).parent.parent / '.env' |
|
|
load_dotenv(dotenv_path) |
|
|
|
|
|
IS_SHARED_SPACE = "Surn/3D-Viewer" in os.environ.get('SPACE_ID', '') |
|
|
|
|
|
HF_API_TOKEN = os.getenv("HF_TOKEN") |
|
|
if not HF_API_TOKEN: |
|
|
raise ValueError("HF_TOKEN is not set. Please check your .env file.") |
|
|
try: |
|
|
if os.environ['TMPDIR']: |
|
|
TMPDIR = os.environ['TMPDIR'] |
|
|
else: |
|
|
TMPDIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp') |
|
|
except: |
|
|
TMPDIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp') |
|
|
|
|
|
os.makedirs(TMPDIR, exist_ok=True) |
|
|
|
|
|
model_extensions = {".glb", ".gltf", ".obj", ".ply"} |
|
|
model_extensions_list = list(model_extensions) |
|
|
image_extensions = {".png", ".jpg", ".jpeg", ".webp"} |
|
|
image_extensions_list = list(image_extensions) |
|
|
upload_file_types = model_extensions_list + image_extensions_list |
|
|
|
|
|
default_slider_images = [ |
|
|
"images/slider/beeuty_545jlbh1_v12_alpha96_300dpi.png", |
|
|
"images/slider/beeuty_545jlbh1_v12_alpha96_300dpi_depth.png" |
|
|
] |
|
|
|
|
|
default_model_3d = "models/beeuty_545jlbh1_300dpi.glb" |
