新增证件识别接口

bbit_ai/app/routers/Public.py

@@ -4,8 +4,9 @@ from fastapi import APIRouter
 
 from config.app import F8_SERVER_USER_ID
 from models.BaseResponse import BaseResponse
+from models.F8ImageRequest import F8ImageRequest
 from models.F8ImageRequestV2 import F8ImageRequestV2
-from service.vision import process_ticket_image
+from service.vision import process_ticket_image, process_license_image
 from utils import MyUtils
 
 publicRouter = APIRouter()
@@ -30,3 +31,27 @@ async def cocoonTicket(data: F8ImageRequestV2):
         return BaseResponse(data=json_data)
     except Exception as e:
         return BaseResponse(status=False, message=f"解析失败: {str(e)}", data=None)
+
+
+import json
+
+
+@publicRouter.post("/recognize-license")
+async def cocoon_license(data: F8ImageRequest):
+    input_data = data.image
+    if "," in input_data:
+        input_data = input_data.split(",")[1]
+    try:
+        img_bytes = base64.b64decode(input_data)
+        json_data = await MyUtils.async_task(
+            process_license_image,
+            img_bytes,
+            f"{data.title}.jpg",
+            data.title,
+            F8_SERVER_USER_ID,
+        )
+        data = json.loads(json_data.get("content"))
+        data["type"] = json_data.get("type")
+        return BaseResponse(data=data)
+    except Exception as e:
+        return BaseResponse(status=False, message=f"解析失败: {str(e)}", data=None)