import os

# ---------------------------
# Persistent folders (tracked in Git via .gitkeep)
# ---------------------------
persistent_folders = [
    "stable-diffusion-webui/deforum/input",                     # init videos
    "stable-diffusion-webui/deforum/output_committed/frames",  # persistent frames
    "stable-diffusion-webui/deforum/output_committed/video",   # final video storage
    "stable-diffusion-webui/models/Stable-diffusion",          # base SD models
    "stable-diffusion-webui/models/Lora",                      # LoRA models
    "stable-diffusion-webui/extensions/ControlNet/models"      # ControlNet weights
]

# ---------------------------
# Session-only / runtime folders
# ---------------------------
session_folders = [
    "stable-diffusion-webui/deforum/output_temp/frames"        # temp frames during render
]

# ---------------------------
# Create all folders
# ---------------------------
for folder in persistent_folders + session_folders:
    os.makedirs(folder, exist_ok=True)
    print(f"Folder ensured: {folder}")