mkm/TaskSystem
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新ocr图片识别

Browse Source
This commit is contained in:
yaooo 2023-10-11 15:04:34 +08:00
parent 1c0482a76a
commit 7b9296039c
1 changed files with 91 additions and 93 deletions

184
app/api/controller/XunFeiController.php

@ -166,97 +166,89 @@ class XunFeiController extends BaseApiController
$iatHostUrl = "wss://iat-api.xfyun.cn/v2/iat";
$iat = new IatClient($this->app_id, $this->api_key, $this->api_secret);
$client = new Client($iat->assembleAuthUrl($iatHostUrl));
if ($client) {
$frameSize = 1280; //每一帧的音频大小
$intervel = 20 * 1000; //发送音频间隔
$status = 0;
while (true) {
$len = fread($audioFile, $frameSize);
if ($len === false) {
break;
}
if ($len === '') { //文件读取完了
$status = 2;
}
switch ($status) {
case 0: //发送第一帧音频，带business 参数
$frameData = array(
'common' => array(
'app_id' => $this->app_id //appid 必须带上，只需第一帧发送
),
'business' => array( //business 参数，只需一帧发送
'language' => 'zh_cn',
'domain' => 'iat',
'accent' => 'mandarin'
),
'data' => array(
'status' => 0,
'format' => 'audio/L16;rate=16000',
'audio' => base64_encode($len),
'encoding' => $encoding
)
);
$client->send(json_encode($frameData));
$status = 1;
break;
case 1:
$frameData = array(
'data' => array(
'status' => 1,
'format' => 'audio/L16;rate=16000',
'audio' => base64_encode($len),
'encoding' => $encoding
)
);
$client->send(json_encode($frameData));
break;
case 2:
$frameData = array(
'data' => array(
'status' => 2,
'format' => 'audio/L16;rate=16000',
'audio' => base64_encode($len),
'encoding' => $encoding
)
);
$client->send(json_encode($frameData));
break 2;
}
//模拟音频采样间隔
usleep($intervel);
$frameSize = 1280; //每一帧的音频大小
$intervel = 20 * 1000; //发送音频间隔
$status = 0;
while (true) {
$len = fread($audioFile, $frameSize);
if ($len === false) {
break;
}
while (true) {
$response = $client->receive();
if ($response === null) {
break;
}
$resp = json_decode($response, true);
$code = $resp['code'];
if ($code != 0) {
break;
}
$message = $resp['message'];
$data = $resp['data'];
$result = $data['result'];
$status = $data['status'];
foreach($result['ws'] as $v) {
$words .= $v['cw'][0]['w'] ?? '';
}
if ($status === 2) {
break;
}
if ($len === '') { //文件读取完了
$status = 2;
}
} else {
// 删除临时音频文件
if (file_exists($file)) {
//unlink($file);
switch ($status) {
case 0: //发送第一帧音频，带business 参数
$frameData = array(
'common' => array(
'app_id' => $this->app_id //appid 必须带上，只需第一帧发送
),
'business' => array( //business 参数，只需一帧发送
'language' => 'zh_cn',
'domain' => 'iat',
'accent' => 'mandarin'
),
'data' => array(
'status' => 0,
'format' => 'audio/L16;rate=16000',
'audio' => base64_encode($len),
'encoding' => $encoding
)
);
$client->send(json_encode($frameData));
$status = 1;
break;
case 1:
$frameData = array(
'data' => array(
'status' => 1,
'format' => 'audio/L16;rate=16000',
'audio' => base64_encode($len),
'encoding' => $encoding
)
);
$client->send(json_encode($frameData));
break;
case 2:
$frameData = array(
'data' => array(
'status' => 2,
'format' => 'audio/L16;rate=16000',
'audio' => base64_encode($len),
'encoding' => $encoding
)
);
$client->send(json_encode($frameData));
break 2;
}
//模拟音频采样间隔
usleep($intervel);
}
while (true) {
$response = $client->receive();
if ($response === null) {
break;
}
$resp = json_decode($response, true);
$code = $resp['code'];
if ($code != 0) {
break;
}
$message = $resp['message'];
$data = $resp['data'];
$result = $data['result'];
$status = $data['status'];
foreach($result['ws'] as $v) {
$words .= $v['cw'][0]['w'] ?? '';
}
if ($status === 2) {
break;
}
return $this->fail('无法连接到 WebSocket 服务器');
}
fclose($audioFile);
// 删除临时音频文件
if (file_exists($file)) {
//unlink($file);
unlink($file);
}
} catch (\Exception $e) {
if (file_exists($file)) {
@ -334,6 +326,10 @@ class XunFeiController extends BaseApiController
}
$ext = pathinfo($file, PATHINFO_EXTENSION);
$base64_image = base64_encode(file_get_contents($file));
if (file_exists($file)) {
// 删除临时文件
unlink($file);
}
$ocr = new OcrClient($this->app_id, $this->api_key, $this->api_secret);
$ocrHostUrl = $ocr->assembleAuthUrl($ocrHostUrl);
$requestBody = [
@ -359,7 +355,7 @@ class XunFeiController extends BaseApiController
]
]
];
$responseData = '';
$text = [];
try {
$client = new GzClient(['timeout' => 2]);
$response = $client->request('POST', $ocrHostUrl, [
@ -367,18 +363,20 @@ class XunFeiController extends BaseApiController
'verify' => false
]);
$responseData = $response->getBody()->getContents();
if (file_exists($file)) {
// 删除临时文件
unlink($file);
$responseArray = json_decode($responseData, true);
if (empty($responseArray['payload']['result']['text'])) {
return $this->fail('json解析错误');
}
$encodeText = $responseArray['payload']['result']['text'];
$textArray = json_decode(base64_decode($encodeText), true);
$lineArray = $textArray['pages'][0]['lines'] ?? [];
foreach($lineArray as $item) {
$text[] = $item['words'][0]['content'] ?? '';
}
} catch (GuzzleException $e) {
if (file_exists($file)) {
// 删除临时文件
unlink($file);
}
return $this->fail($e->getMessage());
}
return $this->data(['words' => (string)$responseData]);
return $this->data(['words' => $text]);
}
}