import config from '@/config'
|
import { getToken } from '@/utils/auth'
|
|
const baseUrl = config.baseUrl
|
|
/**
|
* OCR识别API
|
*/
|
|
/**
|
* 单图OCR识别（通用接口）
|
* @param {String} filePath 图片临时路径
|
* @param {String} type 识别类型：HandWriting/IDCard/BankCard/General
|
* @param {String} provider OCR服务提供商：tencent/baidu
|
* @param {Array} itemNames 需要提取的字段名称数组（手写体识别时使用）
|
* @returns {Promise}
|
*/
|
export function recognizeImage(filePath, type = 'HandWriting', provider = 'tencent', itemNames = []) {
|
return new Promise((resolve, reject) => {
|
const token = getToken()
|
|
// 构建formData
|
const formData = {
|
type: type,
|
provider: provider
|
}
|
|
// 如果有itemNames，添加到formData
|
if (itemNames && itemNames.length > 0) {
|
itemNames.forEach((itemName, index) => {
|
formData[`itemNames[${index}]`] = itemName
|
})
|
}
|
|
uni.uploadFile({
|
url: `${baseUrl}/system/ocr/recognize`,
|
filePath: filePath,
|
name: 'file',
|
header: {
|
'Authorization': `Bearer ${token}`
|
},
|
formData: formData,
|
success: (res) => {
|
try {
|
const response = JSON.parse(res.data)
|
if (response.code === 200) {
|
resolve(response)
|
} else {
|
reject(response)
|
}
|
} catch (e) {
|
reject({ msg: '解析识别结果失败', error: e })
|
}
|
},
|
fail: (err) => {
|
reject({ msg: 'OCR请求失败', error: err })
|
}
|
})
|
})
|
}
|
|
/**
|
* 腾讯云手写体识别（单图）
|
* @param {String} filePath 图片临时路径
|
* @param {Array} itemNames 需要提取的字段名称数组
|
* @returns {Promise}
|
*/
|
export function tencentHandwritingRecognize(filePath, itemNames = []) {
|
return new Promise((resolve, reject) => {
|
const token = getToken()
|
|
// 构建formData
|
const formData = {}
|
if (itemNames && itemNames.length > 0) {
|
itemNames.forEach((itemName, index) => {
|
formData[`itemNames[${index}]`] = itemName
|
})
|
}
|
|
uni.uploadFile({
|
url: `${baseUrl}/system/ocr/tencent/handwriting`,
|
filePath: filePath,
|
name: 'file',
|
header: {
|
'Authorization': `Bearer ${token}`
|
},
|
formData: formData,
|
success: (res) => {
|
try {
|
const response = JSON.parse(res.data)
|
if (response.code === 200) {
|
resolve(response)
|
} else {
|
reject(response)
|
}
|
} catch (e) {
|
reject({ msg: '解析识别结果失败', error: e })
|
}
|
},
|
fail: (err) => {
|
reject({ msg: 'OCR请求失败', error: err })
|
}
|
})
|
})
|
}
|
|
/**
|
* 批量OCR识别（多图）
|
* @param {Array} filePaths 图片临时路径数组
|
* @param {Array} itemNames 需要提取的字段名称数组
|
* @returns {Promise} 返回合并后的字段Map
|
*/
|
export function batchRecognizeImages(filePaths, itemNames = []) {
|
if (!filePaths || filePaths.length === 0) {
|
return Promise.reject({ msg: '图片列表不能为空' })
|
}
|
|
// 创建上传任务队列
|
const uploadPromises = filePaths.map((filePath) => {
|
return tencentHandwritingRecognize(filePath, itemNames)
|
})
|
|
// 等待所有上传完成并合并结果
|
return Promise.all(uploadPromises)
|
.then(results => {
|
// 合并所有结果
|
const mergedFields = {}
|
let successCount = 0
|
let failCount = 0
|
|
results.forEach(response => {
|
if (response.code === 200 && response.data && response.data.fields) {
|
const fields = response.data.fields
|
// 合并字段（如果key已存在且不为空，不覆盖）
|
Object.keys(fields).forEach(key => {
|
if (!mergedFields[key] || mergedFields[key].trim() === '') {
|
mergedFields[key] = fields[key]
|
}
|
})
|
successCount++
|
} else {
|
failCount++
|
}
|
})
|
|
// 过滤结果：只返回itemNames中指定的字段
|
const filteredFields = {}
|
if (itemNames && itemNames.length > 0) {
|
itemNames.forEach(itemName => {
|
if (mergedFields.hasOwnProperty(itemName)) {
|
filteredFields[itemName] = mergedFields[itemName]
|
}
|
})
|
} else {
|
// 如果没有指定itemNames，返回所有字段
|
Object.assign(filteredFields, mergedFields)
|
}
|
|
return {
|
success: true,
|
successCount: successCount,
|
failCount: failCount,
|
fields: filteredFields
|
}
|
})
|
.catch(error => {
|
return Promise.reject({
|
success: false,
|
msg: '批量识别失败',
|
error: error
|
})
|
})
|
}
|
|
/**
|
* 默认的转运单字段列表
|
*/
|
export const DEFAULT_TRANSFER_ITEM_NAMES = [
|
"患者姓名", "性别", "年龄", "身份证号", "诊断", "需支付转运费用",
|
"行程", "开始时间", "结束时间", "家属签名", "患者签名（手印）",
|
"签字人身份证号码", "日期", "联系电话", "本人", "签字人与患者关系"
|
]
|
