import datetime
import webuiapi
class SDApi:
    def __init__(self):
        # self.api = webuiapi.WebUIApi(host='127.0.0.1', port=7860)
        # self.api = webuiapi.WebUIApi(host='stable-diffusion-api.fc-stable-diffusion-plus.1021266052978015.cn-shanghai.fc.devsapp.net', port=80)
        self.api = webuiapi.WebUIApi(host="region-8.seetacloud.com", port=40794)
        webuiapi.ControlNetUnit = webuiapi.ControlNetUnit
        pass

    def get_now_model(self):
        return self.api.util_get_current_model()

    def set_now_model(self, model_name):
        options = {}
        options['sd_model_checkpoint'] = model_name
        return self.api.set_options(options)

    def get_all_models(self):
        return self.api.get_sd_models()

    def txt2img(self,prompt):
        try:

            result = self.api.txt2img(
                prompt=prompt,
                # negative_prompt="NSFW, illustration, 3d, sepia, painting, cartoons, sketch, (worst quality:2), (low quality:2), (normal quality:2), lowres, bad anatomy, bad hands, normal quality, ((monochrome)), ((grayscale:1.2)), futanari, full-package_futanari, penis_from_girl, newhalf, collapsed eyeshadow, multiple eyebrows, vaginas in breasts,holes on breasts, fleckles, stretched nipples, gigantic penis, nipples on buttocks, analog, analogphoto, signatre, logo,2 faces",
                # negative_prompt="(worst quality, low quality:1.4), ((text, signature, captions))",
                negative_prompt="",

                batch_size=4,
                seed=-1,
                styles=[],
                cfg_scale=7,
                sampler_index='Euler a',
                restore_faces=False,
                # restore_faces=True,
                width=762,
                height=512,
                steps=20,
                enable_hr=False,
                hr_scale=2,
                hr_upscaler=webuiapi.HiResUpscaler.Latent,
                hr_second_pass_steps=20,
                hr_resize_x=1536,
                hr_resize_y=1024,
                denoising_strength=0.4,)

            return result
        except Exception as e:
            print(e)
            return self.txt2img(prompt)





if __name__ == '__main__':
    webuiapi.ControlNetUni
# api = SDApi()
# now_model = api.get_now_model()
# print("当前使用 models ：",now_model)
# if "kidsmix" in now_model:
#     print("不需要切换模型")
# else:
#     all_models = api.get_all_models()
#     model_name = None
#     for m in all_models:
#         if "kidsmix" in m.get("title"):
#             model_name = m.get("title")
#             break
#     current_time = datetime.datetime.now()
#     milliseconds = current_time.timestamp() * 1000
#     print(api.set_now_model(model_name))
#     current_time = datetime.datetime.now()
#     end_milliseconds = current_time.timestamp() * 1000
#     print("耗时：" + str(int(end_milliseconds)-int(milliseconds)) + "ms")