仪评指标联分析模块

2025-09-24 13:59:00 +08:00
parent 0ab82b00d6
commit d2775f60a7
28 changed files with 1106 additions and 181 deletions
@@ -0,0 +1,60 @@
+import uuid
+from uuid import UUID
+
+import config.minIO as minIO
+import db.postgres as pg
+from config.minIO import minio_client
+from llm.ticketLLM import *
+
+
+def process_ticket_image(
+    img_bytes: bytes,
+    file_name: str = None,
+    project_name: str = None,
+    user_id: UUID = None,
+):
+    """
+    处理票据图片的核心逻辑，供不同接口调用
+    """
+
+    # 上传到 OSS，使用 UUID 做对象名
+    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
+    json_data = get_ticket_response(oss_url)
+    # 解析条码
+    barcode = decode_barcode(BytesIO(img_bytes))
+    json_data["barcode"] = barcode
+
+    # 获取图片分辨率和大小
+    img = Image.open(BytesIO(img_bytes))
+    resolution = f"{img.width}x{img.height}"
+    size_kb = len(img_bytes) / 1024
+
+    # 插入数据库
+    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],
+        moisture_content=json_data.get("moisture_content"),
+        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=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