| # modules/constants.py | |
| # constants.py contains all the constants used in the project | |
| import os | |
| from pathlib import Path | |
| from dotenv import load_dotenv | |
| # Load environment variables from .env file | |
| 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 |