template_sdxl.js

const fs = require('fs');
const https = require('https');

const TARGET_DIR = process.argv[2] ?? 'sdxl';
const BASE_SRC = `../${TARGET_DIR}/template.ipynb`;

async function fetchModel() {
    return new Promise((resolve, reject)=>{
        // JSON 데이터를 가져올 URL
        const url = 'https://raw.githubusercontent.com/ninjaneural/webui/master/misc/checkpoints_sdxl.json';

        // HTTP GET 요청 보내기
        https.get(url, (response) => {
            let data = '';

            // 데이터를 수신할 때마다 호출되는 콜백 함수
            response.on('data', (chunk) => {
                data += chunk;
            });

            // 데이터 수신이 완료되면 호출되는 콜백 함수
            response.on('end', () => {
                resolve(data);
            });
        }).on('error', (error) => {
            reject(error);
        });
    })
}

let checkpoints = [];

async function copy_files() {
    let templateCode = fs.readFileSync(BASE_SRC, { encoding: 'utf8' });
    let readme = [];
    readme.push(`| Colab                                                                                                                                                                                            | Model                                                                                  | VAE  | Memo                    |`);
    readme.push(`| ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------ | -------------------------------------------------------------------------------------- | ---- | ----------------------- |`);
    const list = checkpoints.filter(x=>!x.auth).sort((a, b) => a.name.toLowerCase() > b.name.toLowerCase() ? 1 : a.name.toLowerCase() < b.name.toLowerCase() ? -1 : 0).map(item=>{
        let id = item.name.toLowerCase().replace(/[- ]/g, '_').replace(/[^a-z0-9]/g, '');
        if (!item.ipynb) {
            item.ipynb = id+'_webui_colab';
        }
        if (!item.checkpoint_file) {
            item.checkpoint_file = id+'.safetensors';
        }
        if (!item.sdmodel) {
            item.sdmodel = 'sdxl';
        }
        return item;
    });
    list.forEach((item) => {
        let code = templateCode;
        console.log(`${item.ipynb} 복사`);
        code = code.replaceAll('#template_checkpoint_default#', item.checkpoint);
        code = code.replaceAll('#template_checkpoint_default_name#', item.checkpoint_file);
        code = code.replaceAll('#template_refine_checkpoint_default#', item.refine_checkpoint ?? '');
        code = code.replaceAll('#template_refine_checkpoint_default_name#', item.refine_checkpoint_file ?? '');
        code = code.replaceAll('#template_notebook#', TARGET_DIR);
        fs.writeFileSync(`../${TARGET_DIR}/${item.ipynb}.ipynb`, code);

        readme.push(`| [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/ninjaneural/webui/blob/master/${TARGET_DIR}/${item.ipynb}.ipynb) | [${item.name}](${item.model})                    | ${item.bakedVAE ? '' : '선택'} | ${item.type}                      |`)
    });

    readmeText = readme.join('\n');
    fs.writeFileSync(`../${TARGET_DIR}/README.md`, readmeText);

    return true;
}

async function make_readme2() {
    let readme = [];
    const list = checkpoints.filter(x=>!x.auth).sort((a, b) => a.name.toLowerCase() > b.name.toLowerCase() ? 1 : a.name.toLowerCase() < b.name.toLowerCase() ? -1 : 0).map(item=>{
        let id = item.name.toLowerCase().replace(/[- ]/g, '_').replace(/[^a-z0-9]/g, '');
        if (!item.ipynb) {
            item.ipynb = id+'_webui_colab';
        }
        if (!item.checkpoint_file) {
            item.checkpoint_file = id+'.safetensors';
        }
        if (!item.sdmodel) {
            item.sdmodel = 'sdxl';
        }
        return item;
    });
    readme.push(`| 바로실행                                                                                                                                                                                        | 설치버전(install)                                                                                                                                                                                | Model                                                                                  | VAE  | Memo                    |`);
    readme.push(`| ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- | ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------ | -------------------------------------------------------------------------------------- | ---- | ----------------------- |`);
    list.filter(x=>x.sdmodel=='sdxl').forEach((item) => {
        readme.push(`| [![Open In Colab](https://raw.githubusercontent.com/neuralninja22/colab/master/icons/colab-badge.svg)](https://colab.research.google.com/github/ninjaneural/webui/blob/master/sdxl/${item.ipynb}.ipynb) | [![Open In Colab](https://raw.githubusercontent.com/neuralninja22/colab/master/icons/colab-badge-install.svg)](https://colab.research.google.com/github/ninjaneural/webui/blob/master/install_sdxl/${item.ipynb}.ipynb) | [${item.name}](${item.model})                    | ${item.bakedVAE ? '' : '선택'} | ${item.type}                      |`)
    });
    readme.push(``);

    readme.push(`| 바로실행                                                                                                                                                                                        | 설치버전(install)                                                                                                                                                                                | Model                                                                                  | VAE  | Memo                    |`);
    readme.push(`| ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- | ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------ | -------------------------------------------------------------------------------------- | ---- | ----------------------- |`);
    list.filter(x=>x.sdmodel=='pony').forEach((item) => {
        readme.push(`| [![Open In Colab](https://raw.githubusercontent.com/neuralninja22/colab/master/icons/colab-badge.svg)](https://colab.research.google.com/github/ninjaneural/webui/blob/master/sdxl/${item.ipynb}.ipynb) | [![Open In Colab](https://raw.githubusercontent.com/neuralninja22/colab/master/icons/colab-badge-install.svg)](https://colab.research.google.com/github/ninjaneural/webui/blob/master/install_sdxl/${item.ipynb}.ipynb) | [${item.name}](${item.model})                    | ${item.bakedVAE ? '' : '선택'} | ${item.type}                      |`)
    });

    readmeText = readme.join('\n');
    fs.writeFileSync(`../COLAB_SDXL_ALL.md`, readmeText);

    let variable = []
    variable.push(
        `LocalCheckpointName = 'juggernaut_xl_v8.safetensors' #@param [${list.map((a)=>'"'+a.checkpoint_file+'"').join(',')}]`
    )
    variable.push(
        `LocalCheckpointMap = {${list.map((a)=>'"'+a.checkpoint_file+'":"'+a.checkpoint+'"').join(',')}}`
    )    
    fs.writeFileSync(`../install/SDXL_VARIABLES.md`, variable.join('\n'));

    return true;
}


(async function () {
    try {
        const model = await fetchModel();
        checkpoints = JSON.parse(model);
        copy_files();
        make_readme2();
    } catch (e) {
        console.error(e);
    }
})();