package foundation.ai;
|
|
import java.io.File;
|
|
import com.alibaba.fastjson.JSONObject;
|
|
import foundation.action.ActionProvider;
|
import foundation.ai.dao.VatInvoiceVerifyDao;
|
import foundation.ai.logic.VatInvoiceVerifyLogic;
|
import foundation.io.FileCenter;
|
import foundation.io.object.FileRecord;
|
import foundation.util.Util;
|
|
public class OCRHandler extends ActionProvider {
|
|
@Override
|
protected void publishMethod() {
|
//1. bizLicense
|
addMethod("bizLicense");
|
|
//2. enterpriseLicense
|
addMethod("enterpriseLicense");
|
|
//3. orgCodeCert
|
addMethod("orgCodeCert");
|
|
//4. institution
|
addMethod("institution");
|
|
//5. idCard
|
addMethod("idCard");
|
|
//6. textAccurate
|
addMethod("textAccurate");
|
|
//7. textCommon //不行
|
addMethod("textCommon");
|
|
//8. textFast //不行
|
addMethod("textFast");
|
|
//9. invoiceValidate
|
addMethod("invoiceIdentify2");
|
|
//10. BankSlip 银行回单识别
|
addMethod("bankSlip");
|
|
//11. Invoice verification--发票验真
|
addMethod("invoiceVerification2");
|
|
//12. Invoice Identify--发票识别
|
addMethod("invoiceIdentify");
|
|
//13. getInvoiceIdentifyByCode--获取发票信息
|
addMethod("getInvoiceIdentifyByCode");
|
|
//14. Invoice verification--发票验真
|
addMethod("invoiceVerification");
|
|
//15. invoiceIdentifyAndVerification--发票识别及核验
|
addMethod("invoiceIdentifyAndVerification");
|
}
|
|
public void bizLicense() throws Exception {
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
dataWriter.addValue("upload", fileRecord);
|
|
//2. OCR识别
|
File file = fileRecord.getFile();
|
OCRResult ocrResult = AICenter.execOCR(Operator.BizLicense, file);
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
|
public void enterpriseLicense() throws Exception {
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
dataWriter.addValue("upload", fileRecord);
|
|
//2. OCR识别
|
File file = fileRecord.getFile();
|
OCRResult ocrResult = AICenter.execOCR(Operator.EnterpriseLicense, file);
|
if (!"fail".equalsIgnoreCase(ocrResult.getJson())) {
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
}
|
|
public void orgCodeCert() throws Exception {
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
dataWriter.addValue("upload", fileRecord);
|
|
//2. OCR识别
|
File file = fileRecord.getFile();
|
OCRResult ocrResult = AICenter.execOCR(Operator.OrgCodeCert, file);
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
|
public void institution() throws Exception {
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
dataWriter.addValue("upload", fileRecord);
|
|
//2. OCR识别
|
File file = fileRecord.getFile();
|
OCRResult ocrResult = AICenter.execOCR(Operator.Institution, file);
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
|
public void idCard() throws Exception {
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
dataWriter.addValue("upload", fileRecord);
|
|
//2. OCR识别
|
File file = fileRecord.getFile();
|
OCRResult ocrResult = AICenter.execOCR(Operator.IDCard, file);
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
|
public void textAccurate() throws Exception {
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
dataWriter.addValue("upload", fileRecord);
|
|
//2. OCR识别
|
File file = fileRecord.getFile();
|
OCRResult ocrResult = AICenter.execOCR(Operator.TextAccurate, file);
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
|
public void textCommon() throws Exception {
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
dataWriter.addValue("upload", fileRecord);
|
|
//2. OCR识别
|
File file = fileRecord.getFile();
|
OCRResult ocrResult = AICenter.execOCR(Operator.TextCommon, file);
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
|
public void textFast() throws Exception {
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
dataWriter.addValue("upload", fileRecord);
|
|
//2. OCR识别
|
File file = fileRecord.getFile();
|
OCRResult ocrResult = AICenter.execOCR(Operator.TextFast, file);
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
|
public void invoiceIdentify2() throws Exception {
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
dataWriter.addValue("upload", fileRecord);
|
|
//2. OCR识别
|
File file = fileRecord.getFile();
|
OCRResult ocrResult = AICenter.execOCR(Operator.InvoiceIdentify, file);
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
|
public void invoiceVerification2() throws Exception {
|
String fileId = null;
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
dataWriter.addValue("upload", fileRecord);
|
|
fileId = fileRecord.getIndexId();
|
//2. OCR识别
|
File file = fileRecord.getFile();
|
OCRResult ocrResult = AICenter.execOCR(Operator.InvoiceIdentify, file);
|
|
if ("fail".equalsIgnoreCase(ocrResult.getJson())) {
|
dataWriter.addValue("ocr", ocrResult);
|
return;
|
}
|
|
VatInvoiceVerifyLogic vatInvoiceVerifyLogic = new VatInvoiceVerifyLogic();
|
|
//3. 解析发票内容
|
VatInvoiceVerifyDao vatInvoiceVerify = vatInvoiceVerifyLogic.parseInvoiceInfo(ocrResult);
|
|
if (Util.isEmpty(vatInvoiceVerify.getInvoiceCode()) && Util.isEmpty(vatInvoiceVerify.getInvoiceNo()) &&
|
Util.isEmpty(vatInvoiceVerify.getInvoiceDate()) && Util.isEmpty(vatInvoiceVerify.getAdditional()) &&
|
Util.isEmpty(vatInvoiceVerify.getAmount())) {
|
ocrResult.setIs_bill(false);
|
ocrResult.setJson(null);
|
dataWriter.addValue("ocr", ocrResult);
|
return;
|
}
|
|
//4.发票验真
|
ocrResult = AICenter.execOCR(Operator.InvoiceVerification, vatInvoiceVerify);
|
|
if (!ocrResult.isIs_real()) {
|
//将假发票删除
|
//vatInvoiceVerifyLogic.deleteBillImage(fileId);
|
ocrResult.setIs_real(false);
|
ocrResult.setIs_bill(true);
|
dataWriter.addValue("ocr", ocrResult);
|
return;
|
}
|
|
//5.将发票明细保存到表中
|
vatInvoiceVerifyLogic.saveVatInvoiceVerify(ocrResult);
|
ocrResult.setIs_real(true);
|
ocrResult.setJson(null);
|
ocrResult.setIs_bill(true);
|
|
//6.更新file_index是is_bill
|
vatInvoiceVerifyLogic.markBill(fileId);
|
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
|
//银行回单识别
|
public void bankSlip() throws Exception {
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
dataWriter.addValue("upload", fileRecord);
|
|
//2. OCR识别
|
File file = fileRecord.getFile();
|
OCRResult ocrResult = AICenter.execOCR(Operator.BankSlip, file);
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
|
public void invoiceIdentify() throws Exception {
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
File file = fileRecord.getFile();
|
dataWriter.addValue("upload", fileRecord);
|
|
//2. OCR识别
|
OCRResult ocrResult = AICenter.execOCR(Operator.InvoiceIdentify, file);
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
|
|
public void getInvoiceIdentifyByCode() throws Exception {
|
Object paramObject = null;
|
//1. 收取参数
|
if(dataReader.getBody() != null ) {
|
paramObject = JSONObject.parseObject(dataReader.getBody());
|
}
|
|
//2. OCR识别
|
OCRResult ocrResult = AICenter.execOCR(Operator.GetInvoiceByCode, paramObject);
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
|
public void invoiceVerification() throws Exception {
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
File file = fileRecord.getFile();
|
dataWriter.addValue("upload", fileRecord);
|
|
//2. OCR识别
|
OCRResult ocrResult = AICenter.execOCR(Operator.InvoiceVerification, file);
|
dataWriter.addValue("ocr", ocrResult);
|
}
|
|
public void invoiceIdentifyAndVerification() throws Exception {
|
//1. 收取上载文件
|
FileRecord fileRecord = FileCenter.uploadOneFile(dataReader);
|
File file = fileRecord.getFile();
|
dataWriter.addValue("upload", fileRecord);
|
|
//2. OCR识别
|
OCRResult ocrResult = AICenter.execOCR(Operator.InvoiceIdentifyAndVerification, file);
|
dataWriter.addValue("ocr", ocrResult);
|
|
|
}
|
|
}
|
