BBIT-Kai
109 lines
3.4 KiB
Python
109 lines
3.4 KiB
Python
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
|