🤹 AI IMG RECOGNITION БЛЯТЬ

This commit is contained in:
Wind-Explorer
2024-08-12 22:42:34 +08:00
parent 6b7b4bba11
commit 878e854be8
7 changed files with 527 additions and 239 deletions

View File

@@ -1,5 +1,8 @@
const express = require("express");
const { openAiChatCompletion } = require("../connections/openai");
const {
openAiChatCompletion,
openAiHomeBillVerification,
} = require("../connections/openai");
const { validateToken } = require("../middlewares/auth");
const router = express.Router();
@@ -37,4 +40,40 @@ router.get("/nls/:query", validateToken, async (req, res) => {
}
});
async function resolveBillPayableAmount(base64Data) {
return await openAiHomeBillVerification(base64Data);
}
let base64Chunks = [];
router.post(
"/resolve-home-bill-payable-amount",
validateToken,
async (req, res) => {
const { chunk, chunkIndex, totalChunks } = req.body;
// 存储接收到的块
base64Chunks[chunkIndex] = chunk;
// 检查是否接收到所有块
if (base64Chunks.length === parseInt(totalChunks)) {
const completeBase64String = base64Chunks.join("");
base64Chunks = []; // 清空数组以便下次上传使用
try {
console.log("starting actual resolve");
let verificationResponse = await resolveBillPayableAmount(
completeBase64String
);
res.json({ response: verificationResponse });
} catch (error) {
console.error("Error with AI:", error);
res.status(500).json({ message: "Internal Server Error: " + error });
}
} else {
res.status(200).json({ message: "Chunk received" });
}
}
);
module.exports = router;