import uuid
from uuid import UUID

import config.minIO as minIO
import db.postgres as pg
from agent.licenseImageAgent import get_license_response
from config.minIO import minio_client
from llm.ticketLLM import *
from llm.ticketLLMv2 import get_ticket_response_v2


def process_ticket_image(
    version: int,
    needBarcode: bool = False,
    img_bytes=None,
    file_name: str = None,
    project_name: str = None,
    user_id: UUID = None,
):
    """
    处理票据图片的核心逻辑，供不同接口调用
    """

    # 上传到 OSS，使用 UUID 做对象名
    if img_bytes is None:
        img_bytes = []
    object_name = str(uuid.uuid4())
    file_bytes = BytesIO(img_bytes)
    bucket_name = "image-ticket"
    if not minio_client.bucket_exists(bucket_name):
        minio_client.make_bucket(bucket_name)

    minIO.push_file(bucket_name, object_name, file_bytes, img_bytes, "image/jpeg")
    oss_url = minIO.get_temp_url(bucket_name, object_name)

    # 调用分析方法获取 JSON
    if version == 1:
        json_data = get_ticket_response(oss_url)
    elif version == 2:
        json_data = get_ticket_response_v2(oss_url, needBarcode)
    else:
        json_data = get_ticket_response(oss_url)
    # 获取图片分辨率和大小
    img = Image.open(BytesIO(img_bytes))
    resolution = f"{img.width}x{img.height}"
    size_kb = round(len(img_bytes) / 1024, 2)

    # 插入数据库
    pg.insert_ticket_image(
        created_by=user_id,
        file_name=file_name,
        resolution=resolution,
        size=size_kb,
        name=project_name if project_name else object_name[:8],
        dead_pupa_count=json_data.get("dead_pupa_count") if version == 2 else 0,
        moisture_content=json_data.get("moisture_content") if version == 2 else 0,
        cocoon_weight=json_data.get("cocoon_weight"),
        defective_pupa_count=json_data.get("defective_pupa_count"),
        fresh_shell_weight=json_data.get("fresh_shell_weight"),
        sample_count=json_data.get("sample_count"),
        barcode=json_data.get("barcode"),
        oss=object_name,
        net_weight_total=json_data.get("net_weight_total"),
        evaluator=json_data.get("evaluator"),
        reviewer=json_data.get("reviewer"),
    )

    return json_data


def process_license_image(
    img_bytes=None,
    file_name: str = None,
    project_name: str = None,
    user_id: UUID = None,
):
    # 上传到 OSS，使用 UUID 做对象名
    if img_bytes is None:
        img_bytes = []
    object_name = str(uuid.uuid4())
    file_bytes = BytesIO(img_bytes)
    bucket_name = "image-license"
    if not minio_client.bucket_exists(bucket_name):
        minio_client.make_bucket(bucket_name)

    minIO.push_file(bucket_name, object_name, file_bytes, img_bytes, "image/jpeg")
    oss_url = minIO.get_temp_url(bucket_name, object_name)

    # 调用分析方法获取 JSON
    json_data = get_license_response(oss_url)
    # 获取图片分辨率和大小
    img = Image.open(BytesIO(img_bytes))
    resolution = f"{img.width}x{img.height}"
    size_kb = round(len(img_bytes) / 1024, 2)

    # 插入数据库
    pg.insert_license_image(
        created_by=user_id,
        file_name=file_name,
        resolution=resolution,
        size=size_kb,
        name=project_name if project_name else object_name[:8],
        oss=object_name,
        type=json_data.get("type"),
        content=json_data.get("content"),
    )

    return json_data