【NovelAI】在QQ群中部署AI画图机器人

目录

一、NovelAI 

二、UIautomation和pywin32

三、代码 

1、AI画图相关

 2、QQ群消息抓取相关

四、代码效果 

五、后记 

---

一、NovelAI 

NovelAI是一个用来画二次元图片的开源算法，部署方式见  b站链接

二、UIautomation和pywin32

UIautomation和pywin32是python的库，主要用于窗口句柄的抓取和操作，本代码中用于接收、发送QQ群消息 

三、代码 

1、AI画图相关

通过使用NovelAI代码包里的txt2img.py文件，实现从文字到图片的转化：

def txt2img(prompt: str, negative_prompt: str, prompt_style: str, prompt_style2: str, steps: int, sampler_index: int, restore_faces: bool, tiling: bool, n_iter: int, batch_size: int, cfg_scale: float, seed: int, subseed: int, subseed_strength: float, seed_resize_from_h: int, seed_resize_from_w: int, seed_enable_extras: bool, height: int, width: int, enable_hr: bool, scale_latent: bool, denoising_strength: float, *args):
    p = StableDiffusionProcessingTxt2Img(
        sd_model=shared.sd_model,
        outpath_samples=opts.outdir_samples or opts.outdir_txt2img_samples,
        outpath_grids=opts.outdir_grids or opts.outdir_txt2img_grids,
        prompt=prompt,
        styles=[prompt_style, prompt_style2],
        negative_prompt=negative_prompt,
        seed=seed,
        subseed=subseed,
        subseed_strength=subseed_strength,
        seed_resize_from_h=seed_resize_from_h,
        seed_resize_from_w=seed_resi