File size: 3,591 Bytes
667fbb2 3c073d8 667fbb2 73f006c 667fbb2 0a60a6a 667fbb2 | 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 | """
Configuration settings for Indian Art Generator
"""
import os
from typing import Dict
class Config:
"""Application configuration"""
# Model Settings
BASE_MODEL = os.getenv("BASE_MODEL", "stable-diffusion-v1-5/stable-diffusion-v1-5")
# LoRA Weights Path
# Options:
# 1. Local: "./models/..."
# 2. HuggingFace Hub: "username/repo-name"
LORA_PATH = os.getenv("LORA_PATH", "mehradity/lora_weights")
LORA_ADAPTER_NAME = os.getenv("LORA_ADAPTER_NAME", "indian_art")
# Generation Defaults
DEFAULT_WIDTH = int(os.getenv("DEFAULT_WIDTH", "512"))
DEFAULT_HEIGHT = int(os.getenv("DEFAULT_HEIGHT", "512"))
DEFAULT_STEPS = int(os.getenv("DEFAULT_STEPS", "20" if os.getenv("DEVICE") == "cpu" else "30"))
DEFAULT_GUIDANCE = float(os.getenv("DEFAULT_GUIDANCE", "7.5"))
DEFAULT_LORA_SCALE = float(os.getenv("DEFAULT_LORA_SCALE", "0.8"))
# Device
DEVICE = os.getenv("DEVICE", "cuda")
# Cache dir for models (use /tmp for HF Spaces)
CACHE_DIR = os.getenv("CACHE_DIR", "/tmp/huggingface")
# Art-specific Negative Prompts (Cultural Preservation)
# Prevents Western art styles and modern elements
DEFAULT_NEGATIVE_PROMPT = """photorealistic, 3d render, photography, western art style,
oil painting, renaissance, baroque, impressionist, modern abstract,
graffiti, street art, pop art, contemporary western,
urban elements, skyscrapers, cars, modern architecture,
blurry, low quality, distorted, deformed, ugly, duplicate,
bad anatomy, disfigured, poorly drawn face, mutation, mutated, extra limbs,
watermark, signature, text, logo, cropped, out of frame,
worst quality, low resolution, jpeg artifacts, noise, grainy,
grayscale, monochrome, boring, plain background,
western clothing, modern fashion, jeans, t-shirt, sunglasses, sneakers"""
# Art Style Prefixes (Auto-enhance prompts)
ART_STYLES: Dict[str, str] = {
"madhubani": "madhubani painting, intricate double border, geometric patterns, natural dyes, no gaps, fine line work, traditional motifs,",
"warli": "warli tribal art, white rice paste drawing on dark background, stick figures, traditional hut, daily life scene, harvest festival,",
"gond": "gond painting, intricate dot patterns, bright flat colors, nature inspired, line work, patangarh style, dighna patterns,",
"pattachitra": "pattachitra scroll art, mythological theme, bold outlines, natural colors, intricate border, odisha tradition, palm leaf texture,",
"tanjore": "tanjore painting, gold foil work, rich colors, devotional iconography, embossed texture, tamil nadu style, semi-precious stones,",
"miniature": "mughal miniature painting, fine detailing, persian influence, solid colors, intricate borders, court scene,",
"kalamkari": "kalamkari art, hand drawn with kalam, natural dyes, mythological narratives, floral motifs, andhra pradesh style,",
"phad": "phad painting, scroll narrative, bold lines, bright orange red colors, rajasthani folk, deity stories,",
"mysore": "mysore painting, muted colors, intricate details, gold leaf, karnataka tradition, hindu gods,",
"none": ""
}
# UI Configuration
GRADIO_THEME = os.getenv("GRADIO_THEME", "soft")
GRADIO_SHARE = os.getenv("GRADIO_SHARE", "false").lower() == "true"
@classmethod
def validate(cls):
"""Validate configuration"""
if cls.LORA_PATH is None:
print(" Warning: LORA_PATH not set. Using base SDXL model only.")
return True
|