import io
import PIL
import base64
import re
import random
import aiohttp
import asyncio
from aiogoogletrans import Translator
from typing import Any, Dict
from discord import File

translator = Translator()
weight = [0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.7, 0.8, 0.9]
cfg = [6,7,8,9,10,11]
#def check_curse(text: str):
#    return korcen_checker.check(text)
def is_korean(string):
    pattern = re.compile(r"[ㄱ-ㅎㅏ-ㅣ가-힣]")
    match = pattern.search(string)
    return bool(match)
    
async def process_prompt(prompt: str, remove: str, res: list, isnsfw: bool, style1: float, style2: float, afterprocess: float):
    if is_korean(prompt):
        prompt = await translator.translate(prompt, dest="en")
        prompt = prompt.text
    if is_korean(remove):
        remove = await translator.translate(remove, dest="en")
        remove = remove.text
    default_negative = """(KHFB, AuroraNegative, easynegative, negative_hand-neg, verybadimagenegative_v1.3:0.8), (Worst Quality, Low Quality:1.4), border, skimpy, grayscale, multiple_girls, 3d, realistic, string, multiple hands, chinese, thick abs, chubby abs, lowres, bad anatomy, asymmetric wings, elf, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, large areolae, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, Multiple legs, multiple feet, tie, (necktie:1.5), several hands, three feet, four legs, three legs,animal ears"""
    if style1 != 0:
        prompt = prompt + f"<lora:addDetail:{style1}>"
    if style2 != 0:
        prompt = prompt + f"<lora:光影:{style2}>"
    if remove != None:
        negative_prompt = default_negative + "," + remove
    add_prompt = random.choice([True, False])
    if add_prompt == True:
        qprompt = prompt + f"<lora:canistermix1.1:{random.choice(weight)}>"
    payloads = {
        "prompt": prompt,
        "negative_prompt": negative_prompt,
        "seed": random.randint(0, 1000000000),
        "steps": 25,
        "cfg_scale": random.choice(cfg),
        "width": res[0],
        "height": res[1],
        "sampler_index": "DPM++ 2M Karras",
        "refiner_checkpoint": "smooREFINERV2R10_half",
        "refiner_switch_at": afterprocess,
        "alwayson_scripts": {
            "ADetailer": {
                'args': [
                    {
                        'ad_model': 'face_yolov8n.pt',
                        'ad_inpaint_only_masked': True        
                    }]
            }}}
    return payloads


async def post_gpu_server(url: str, payload: Dict[str, Any]):
    async with aiohttp.ClientSession() as session:
        try:
            async with session.post(url, json=payload, timeout=300) as response:
                if response.status == 200:
                    return {"status": True, "data": await response.json()} 
                else:
                    return {"status": False, "data": None}
        except Exception as e:
            if isinstance(e, asyncio.TimeoutError):
                return {"status": False, "data": None}
            
            
async def image_to_base64(image) -> str:
    img = PIL.Image.open(image)
    img = img.convert("RGB")
    img_byte_arr = io.BytesIO()
    img.save(img_byte_arr, format='PNG')
    img_byte_arr = img_byte_arr.getvalue()
    img_str = base64.b64encode(img_byte_arr).decode("utf-8")
    return img_str


async def base64_to_image(base642) -> File:
    attachment = File(io.BytesIO(base64.b64decode(base642)), filename="image.png")
    return attachment