Spaces:
Running
Running
File size: 3,375 Bytes
82a1838 | 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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 | # modules/constants.py
# constants.py contains all the constants used in the project such as the default LUT example image, prompts, negative prompts, pre-rendered maps, models, LoRA weights, and more.
# execptions made for some environmental variables
import os
from pathlib import Path
from dotenv import load_dotenv
import numpy as np
IS_SHARED_SPACE = "Surn/SeparateTracks" in os.environ.get('SPACE_ID', '')
# Load environment variables from .env file
dotenv_path = Path(__file__).parent.parent / '.env'
load_dotenv(dotenv_path)
# Function to load env vars from .env and create Python variables
def load_env_vars(env_path):
try:
with open(env_path, 'r') as file:
for line in file:
# Skip empty lines or comments
line = line.strip()
if line and not line.startswith('#'):
# Split on the first '=' only
if '=' in line:
key, value = line.split('=', 1)
key = key.strip()
value = value.strip()
# Dynamically create a Python variable with the key name
globals()[key] = value
# Also update os.environ (optional, for consistency)
os.environ[key] = value
except FileNotFoundError:
print(f"Warning: .env file not found at {env_path}")
USE_FLASH_ATTENTION = os.getenv("USE_FLASH_ATTENTION", "0") == "1"
HF_API_TOKEN = os.getenv("HF_TOKEN", None)
CRYPTO_PK = os.getenv("CRYPTO_PK", None)
if not HF_API_TOKEN:
raise ValueError("HF_TOKEN is not set. Please check your .env file.")
default_lut_example_img = "./LUT/daisy.jpg"
MAX_SEED = np.iinfo(np.int32).max
TARGET_SIZE = (2688,1536)
BASE_HEIGHT = 640
SCALE_FACTOR = (12/5)
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)
SPACE_NAME = os.getenv('SPACE_NAME', 'Surn/SeparateTracks')
# Constants for URL shortener and storage
HF_REPO_ID = os.getenv("HF_REPO_ID", "Surn/Storage") # Replace with your Hugging Face repository ID
SHORTENER_JSON_FILE = "shortener.json"
model_extensions = {".glb", ".gltf", ".obj", ".ply"}
model_extensions_list = list(model_extensions)
image_extensions = {".png", ".jpg", ".jpeg", ".webp"}
image_extensions_list = list(image_extensions)
audio_extensions = {".mp3", ".wav", ".ogg", ".flac"}
audio_extensions_list = list(audio_extensions)
video_extensions = {".mp4"}
video_extensions_list = list(video_extensions)
doc_extensions = {".json"}
doc_extensions_list = list(doc_extensions)
upload_file_types = model_extensions_list + image_extensions_list + audio_extensions_list + video_extensions_list + doc_extensions_list
umg_mcp_server = "https://surn-unlimitedmusicgen.hf.space/gradio_api/mcp/sse"
#umg_mcp_server = "http://127.0.0.1:7860/gradio_api/mcp/sse"
badge_negative_prompt = "low quality, blurry, copyright, cropped, worst quality, bad text, missing text, normal quality, jpeg artifacts, signature, watermark, username, missing_transparent_background"
default_badge = "assets/openbadge.png"
default_badge_512_url = os.getenv("DEFAULT_BADGE_512_URL",None)
|