config.py

import os
from dotenv import load_dotenv

# Load environment variables from .env file
load_dotenv()

# Hugging Face API token
HF_TOKEN = os.getenv("HF_TOKEN", "")

# Default model for text to image
DEFAULT_TEXT2IMG_MODEL = "stabilityai/stable-diffusion-2-1"

# Default model for image to image
DEFAULT_IMG2IMG_MODEL = "lllyasviel/sd-controlnet-depth"

# 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"))