base_prompt = {
    'model_id': 'sdxl',
    'init_image': '',
    'mask_image': None,
    'control_image': None,
    'width': '512',
    'height': '512',
    'prompt': '',
    'guess_mode': None,
    'use_karras_sigmas': None,
    'tomesd': None,
    'vae': 'sd-ft-mse',
    'embeddings': None,
    'lora_strength': None,
    'multi_lingual': None,
    'upscale': None,
    'strength': 1,
    'negative_prompt': 'high contrast, ugly, duplicate, morbid, mutilated, out of frame, extra fingers, mutated hands, poorly drawn hands, poorly drawn face, mutation, deformed, blurry, bad anatomy, bad proportions, extra limbs, cloned face, disfigured, gross proportions, malformed limbs, missing arms, missing legs, extra arms, extra legs, fused fingers, too many fingers, long neck, low resolution, 3d model, deformed hands, deformed feet, deformed face, deformed body parts, same haircut, eyes without pupils, doubled image, mid aged man, old men, logo in frame, gun, man with more than one penis on body, scared facial expression, drawing, painting, blur focus, blur, photo effects, skinny guy, make-up on male, angry facial expression, same human face in one frame, illustration, anime, cartoon, ugly face, bruises, cartoon, anime, painting, red color saturation, unattractive face, jpeg artifacts, frame, Violence, Gore, Blood, War, Weapons, Death, Destruction, Fire, Explosions, Pollution, Garbage, Graffiti, Vandalism, Rust, Decay, Filth, Disease, Insects, Rodents, Vermin, Darkness, Shadows, Nightmares, Fear, Horror, Sadness, Depression, Pain, Suffering, Anguish, Despair, Loneliness, Isolation, Neglect, Abandonment, Negativity, Hate, Racism, Sexism, Homophobia, Discrimination, Intolerance, Prejudice, Ignorance, Arrogance, Greed, Selfishness, Cruelty, Insanity, Madness, lowres, text, error, cropped, worst quality, low quality, jpeg artifacts, ugly, duplicate, morbid, mutilated, out of frame, extra fingers, mutated hands, poorly drawn hands, poorly drawn face, mutation, deformed, blurry, dehydrated, bad anatomy, bad proportions, extra limbs, cloned face, disfigured, gross proportions, malformed limbs, missing arms, missing legs, extra arms, extra legs, fused fingers, too many fingers, long neck, deformed, ugly, mutilated, disfigured, text, extra limbs, face cut, head cut, extra fingers, extra arms, poorly drawn face, mutation, bad proportions, cropped head, malformed limbs, mutated hands, fused fingers, long neck, illustration, painting, drawing, art, sketch, disfigured, kitsch, ugly, oversaturated, grain, low-res, Deformed, blurry, bad anatomy, disfigured, poorly drawn face, mutation, mutated, extra limb, ugly, poorly drawn hands, missing limb, blurry, floating limbs, disconnected limbs, malformed hands, blur, out of focus, long neck, long body, ugly, disgusting, poorly drawn, childish, mutilated, , mangled, old, surreal',  # noqa
    'guidance': '10',
    'samples': '1',
    'safety_checker': 'yes',
    'auto_hint': 'yes',
    'steps': 20,
    'seed': 27582,
    'webhook': None,
    'track_id': None,
    'scheduler': 'EulerAncestralDiscreteScheduler',
    'base64': None,
    'clip_skip': None,
    'controlnet_conditioning_scale': None,
    'temp': None,
    'controlnet_type': None,
    'controlnet_model': 'softedge',
    'lora': ''
}

loras_types = {
    'ghibli style': {
        'lora': 'ghibli-style'
    },
    'naruto style, a ninja': {
        'lora': 'naruto-oneai'
    },
    'anime style, 2d, masterpiece': {
        'lora': ''
    },
    'kimetsu no yaiba style': {
        'lora': 'vampire-killer'
    },
    'toriyama akira': {
        'lora': 'akira-toriyama'
    },
    'wanostyle': {
        'lora': 'wanostyle'
    },
}


async def sdf_request(prompt: str, photo: str) -> dict:
    lora = loras_types.get(prompt, {}).get('lora', '')
    base_prompt['key'] = os.environ.get('STABILITY_KEY')
    base_prompt['prompt'] = prompt
    base_prompt['init_image'] = photo
    base_prompt['seed'] = randint(1, 30000)
    base_prompt['lora'] = lora
    async with ClientSession() as session:
        response = await session.post(url_stable,
                                      headers=headers_stable,
                                      data=json.dumps(base_prompt))
        body = await response.json()
        logger.debug(body)
        response.close()
    if body['status'] == 'processing':
        while body['status'] == 'processing':
            data = {'key': os.environ.get('STABILITY_KEY'),
                    'request_id': body['id']}
            async with ClientSession() as session:
                response = await session.post(url_stable,
                                              headers=headers_stable,
                                              data=json.dumps(data))
                logger.debug(body)
                body = await response.json()
                response.close()
    if body['status'] == 'failed':
        return {}
    return {'photo_url': body['output'][0],
            'prompt': base_prompt,
            'generation_time': body['generationTime']}