package jnpf.base.model.ocr;
|
|
import com.alibaba.fastjson.JSONArray;
|
import com.alibaba.fastjson.JSONObject;
|
import com.fasterxml.jackson.annotation.JsonIgnore;
|
import io.swagger.v3.oas.annotations.media.Schema;
|
import jnpf.base.model.ocr.model.*;
|
import jnpf.exception.DataException;
|
import jnpf.util.wxutil.HttpUtil;
|
import lombok.Data;
|
import lombok.NoArgsConstructor;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* ocr解析模型父类
|
*
|
* @author JNPF开发平台组
|
* @version v6.1.0
|
* @copyright 引迈信息技术有限公司
|
* @date 2025/7/29 16:58:47
|
*/
|
@Data
|
@Schema(description = "ocr解析模型父类")
|
@NoArgsConstructor
|
public class OcrModel {
|
@JsonIgnore
|
private String type;
|
@JsonIgnore
|
private List<String> strList;
|
private String content;
|
|
public static OcrModel get(String type) {
|
OcrModel ocrModel = new OcrModel();
|
if (OcrConstant.TEXT_OCR.equals(type)) {
|
return ocrModel;
|
}
|
if (OcrConstant.BANK_CARD.equals(type)) {
|
ocrModel = new BankCardModel();
|
}
|
if (OcrConstant.IDCARD_FRONT.equals(type) || OcrConstant.IDCARD_BACK.equals(type)) {
|
ocrModel = new IdCardModel();
|
ocrModel.setType(type);
|
return ocrModel;
|
}
|
if (OcrConstant.BUSINESS_LICENSE.equals(type)) {
|
ocrModel = new BusinessLicModel();
|
}
|
if (OcrConstant.INVOICE.equals(type)) {
|
ocrModel = new InvoiceLicModel();
|
}
|
if (OcrConstant.DRIVING_LICENSE.equals(type)) {
|
ocrModel = new DrivingLicModel();
|
}
|
if (OcrConstant.VEHICLE_LICENSE.equals(type)) {
|
ocrModel = new VehicleLicModel();
|
}
|
if (OcrConstant.TRAIN_TICKET.equals(type)) {
|
ocrModel = new TrainTicketModel();
|
}
|
return ocrModel;
|
}
|
|
public void getStrList(JSONObject obj) {
|
List<String> list = new ArrayList<>();
|
if (obj != null && obj.containsKey("status")) {
|
JSONArray results = obj.getJSONArray("results");
|
for (Object result : results) {
|
JSONArray m = (JSONArray) result;
|
for (Object n : m) {
|
JSONObject o = (JSONObject) n;
|
list.add(o.getString("text"));
|
}
|
}
|
}
|
strList = list;
|
}
|
|
public void extract(String url, JSONObject body) {
|
JSONObject systemJson = this.getSystemJson(url, body);
|
getStrList(systemJson);
|
this.content = String.join("", strList);
|
}
|
|
public JSONObject getSystemJson(String url, JSONObject body) {
|
JSONObject jsonObject = HttpUtil.httpRequest(url + "/predict/ocr_system", "POST", body.toJSONString());
|
if (jsonObject == null) {
|
throw new DataException("OCR服务器连接失败！");
|
}
|
return jsonObject;
|
}
|
|
public JSONObject getTableJson(String url, JSONObject body) {
|
//structure_layout
|
return HttpUtil.httpRequest(url + "/predict/structure_table", "POST", body.toJSONString());
|
}
|
}
|
