| import os | |
| from dotenv import load_dotenv | |
| # Load environment variables from .env file | |
| load_dotenv() | |
| # Hugging Face API token | |
| # First try to get from environment variables (Hugging Face Spaces secrets) | |
| # Then fall back to .env file for local development | |
| HF_TOKEN = os.getenv("HF_TOKEN", "") | |
| # Default model for text to image | |
| DEFAULT_TEXT2IMG_MODEL = "stabilityai/stable-diffusion-3-medium-diffusers" | |
| # Default model for image to image | |
| DEFAULT_IMG2IMG_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5" | |
| # ControlNet configuration | |
| USE_CONTROLNET = True # Set to False to disable ControlNet in case of issues | |
| CONTROLNET_MODEL = "lllyasviel/sd-controlnet-depth" | |
| BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5" | |
| # Default prompts - used as placeholders in UI and defaults in API | |
| DEFAULT_TEXT2IMG_PROMPT = "A beautiful landscape with mountains and a lake" | |
| DEFAULT_IMG2IMG_PROMPT = "Transform this image with fantasy elements" | |
| # Default negative prompts - used to improve image quality by avoiding common issues | |
| DEFAULT_NEGATIVE_PROMPT = "blurry, low quality, distorted, deformed, ugly, pixelated, noise, grain, text, watermark, poor composition, bad proportions, disfigured, mutation, mutated, extra limbs, extra fingers, fused fingers, malformed hands, poorly drawn face, bad anatomy, amateur drawing, low resolution, duplicate, cropped, out of frame, worst quality, jpeg artifacts, compression artifacts, glitch, overexposed, underexposed, low contrast, washed out colors, oversaturated, signature, username, artist name, logo, unnatural lighting, uneven lighting, harsh shadows, motion blur, out of focus, chromatic aberration, film grain, scratches, stains, dark spots, light leaks, fuzzy edges, jagged edges, broken lines, stretched image, elongated features, asymmetrical eyes, misaligned features, missing details, incomplete rendering" | |
| # API settings | |
| API_HOST = os.getenv("API_HOST", "0.0.0.0") | |
| API_PORT = int(os.getenv("API_PORT", "8000")) | |
| # Gradio settings | |
| GRADIO_HOST = os.getenv("GRADIO_HOST", "0.0.0.0") | |
| GRADIO_PORT = int(os.getenv("GRADIO_PORT", "7860")) | |