ColorComic/config.py at main · vikast908/ColorComic · GitHub

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
import os

BASE_DIR = os.path.dirname(os.path.abspath(__file__))


class Config:
    SECRET_KEY = os.environ.get("SECRET_KEY", os.urandom(24).hex())
    UPLOAD_FOLDER = os.path.join(BASE_DIR, "uploads")
    OUTPUT_FOLDER = os.path.join(BASE_DIR, "output")
    MAX_CONTENT_LENGTH = 200 * 1024 * 1024  # 200 MB

    # Image processing
    PAGE_DPI = 300
    PREVIEW_DPI = 150

    # ── manga-colorization-v2 (auto mode) ─────────────────────────────
    WEIGHTS_DIR = os.path.join(BASE_DIR, "models", "weights")
    GENERATOR_WEIGHTS_PATH = os.path.join(WEIGHTS_DIR, "generator.zip")
    EXTRACTOR_WEIGHTS_PATH = os.path.join(WEIGHTS_DIR, "extractor.pth")
    DENOISER_WEIGHTS_DIR = os.path.join(WEIGHTS_DIR, "denoiser")
    ML_DEVICE = os.environ.get("COLORCOMIC_DEVICE", "auto")
    COLOR_TRANSFER_STRENGTH = float(os.environ.get("COLOR_TRANSFER_STRENGTH", "0.7"))

    # ── MangaNinja (reference mode) ───────────────────────────────────
    MANGANINJA_WEIGHTS_DIR = os.path.join(WEIGHTS_DIR, "manganinja")
    MANGANINJA_DENOISING_UNET = os.path.join(MANGANINJA_WEIGHTS_DIR, "denoising_unet.pth")
    MANGANINJA_REFERENCE_UNET = os.path.join(MANGANINJA_WEIGHTS_DIR, "reference_unet.pth")
    MANGANINJA_POINTNET = os.path.join(MANGANINJA_WEIGHTS_DIR, "point_net.pth")
    MANGANINJA_CONTROLNET = os.path.join(MANGANINJA_WEIGHTS_DIR, "controlnet.pth")

    MANGANINJA_HF_REPO = "Johanan0528/MangaNinja"

    # Paths for SD 1.5 / CLIP / ControlNet base models (HuggingFace cache)
    SD15_MODEL_PATH = os.environ.get(
        "SD15_MODEL_PATH", "stable-diffusion-v1-5/stable-diffusion-v1-5"
    )
    CLIP_VISION_PATH = os.environ.get(
        "CLIP_VISION_PATH", "openai/clip-vit-large-patch14"
    )
    CONTROLNET_LINEART_PATH = os.environ.get(
        "CONTROLNET_LINEART_PATH", "lllyasviel/control_v11p_sd15_lineart"
    )
    LINEART_ANNOTATOR_PATH = os.path.join(MANGANINJA_WEIGHTS_DIR, "annotators")

    MANGANINJA_DENOISE_STEPS = int(os.environ.get("MANGANINJA_DENOISE_STEPS", "30"))

    # ── Real-ESRGAN upscaler ──────────────────────────────────────────
    ESRGAN_MODEL_PATH = os.path.join(WEIGHTS_DIR, "RealESRGAN_x4plus_anime_6B.pth")
    ESRGAN_MODEL_URL = (
        "https://github.com/xinntao/Real-ESRGAN/releases/download/"
        "v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth"
    )
    ESRGAN_SCALE = 4
    ESRGAN_TILE = 256

    # ── Post-processing toggles ───────────────────────────────────────
    POSTPROCESS_L_CHANNEL = os.environ.get("POSTPROCESS_L_CHANNEL", "1") == "1"
    POSTPROCESS_GUIDED_FILTER = os.environ.get("POSTPROCESS_GUIDED_FILTER", "1") == "1"
    POSTPROCESS_UPSCALE = os.environ.get("POSTPROCESS_UPSCALE", "0") == "1"