import json
import pandas as pd
from PIL import Image
from .model import model
from .data import pepsodent_qpds_data






async def detection(model, img_content, confidence):
    try:
        img = Image.open(img_content)
        result = model(source=img, device=0, conf=confidence)
        detection = {}
        data = json.loads(result[0].tojson())
        if not result or not data:
            return {}
        else:
            name_counts = {}
            for item in data:
                name = item['name']
                if name in name_counts:
                    name_counts[name] += 1
                else:
                    name_counts[name] = 1
            detection.update(name_counts)

        return detection
    except Exception as e:
        return {}


async def format_result(convertData,resData):
    try:
        result = {}
        for aiName,sysName in convertData.items():
            if aiName in resData:
                result.update({sysName:resData[aiName]})
        return result
    except Exception as e:
        pass




async def main(img):
    try:
        result = await detection(model,img,0.3)
        formattedResult = await format_result(pepsodent_qpds_data,result)

        return formattedResult
    except Exception as e:
        pass