NovelAI 5ch Wiki

async def main():
    async with ClientSession() as session:
        api = NovelAI_API(session, logger = logger)

        login = await api.high_level.login(username, password)
        logger.info(login)

        preset = ImagePreset()
        preset['n_samples'] = 1
        preset['resolution'] = (512, 512)

        contains_always = ['nsfw', '{{{masterpiece}}}', 'sex']
        positions = [
            'reverse cowgirl position',
            'cowgirl position',
            'straddling',
            'amazon position',
            'spooning',
            'spitroast',
            'top-down bottom-up',
            'missionary',
            'doggystyle',
            'girl on top',
            'boy on top',
            'kneeling',
            'lying',
        ]

        for i in range(100):
            now = datetime.datetime.now().strftime('%Y%m%d%H%M%S')
            random.shuffle(positions)
            word = ', '.join(contains_always + positions[:1]) + ','
            async for img in api.high_level.generate_image(word, ImageModel.Anime_Full, preset):
                with open(f'{now}.png', 'wb') as f:
                    f.write(base64.b64decode(img))
                with open(f'{now}.txt', 'wt') as f:
                    f.write(word)
            time.sleep(1)