"""Centralized configuration settings for the application.""" import os from pathlib import Path from typing import Optional from dotenv import load_dotenv # Load environment variables load_dotenv() # Base paths BASE_DIR = Path(__file__).parent.parent # HF Spaces: use persistent storage when available (/data). Fallback to repo-local. _PERSISTENT_ROOT = Path("/data") _DEFAULT_OUTPUT_ROOT = _PERSISTENT_ROOT if _PERSISTENT_ROOT.exists() else BASE_DIR OUTPUT_DIR = _DEFAULT_OUTPUT_ROOT / os.getenv("OUTPUT_DIR", "generated_images") ASSETS_DIR = BASE_DIR / os.getenv("ASSETS_DIR", "assets") EXAMPLES_DIR = ASSETS_DIR / "examples" # Model configuration # For HF Inference / InferenceClient, prefer the canonical repo id (no "models/"). MODEL_ID = os.getenv("MODEL_ID", "gokaygokay/Flux-Seamless-Texture-LoRA") MODEL_PROVIDER = os.getenv("MODEL_PROVIDER", "hf-inference") # MCP configuration MCP_ENABLED = True MCP_ENDPOINT = "/gradio_api/mcp/" # Application settings MAX_IMAGES = int(os.getenv("MAX_IMAGES", "100")) CLEANUP_AFTER_DAYS = int(os.getenv("CLEANUP_AFTER_DAYS", "7")) ENABLE_WATERMARK = os.getenv("ENABLE_WATERMARK", "false").lower() == "true" # Logging LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO") LOG_FILE = os.getenv("LOG_FILE", "app.log") # Default generation parameters DEFAULT_PARAMS = { "guidance_scale": 7.5, "num_inference_steps": 50, "seed": None, "width": 1024, "height": 1024, "cfg_scale": 7.5, "negative_prompt": "", "lora_strength": 1.0, } # Create directories if they don't exist OUTPUT_DIR.mkdir(parents=True, exist_ok=True) ASSETS_DIR.mkdir(parents=True, exist_ok=True) EXAMPLES_DIR.mkdir(parents=True, exist_ok=True)