Update worker.js
This commit is contained in:
programmerWsy
GitHub
parent
3da4260d99
commit
a6a5241c26
277
worker.js
277
worker.js
@ -16,162 +16,19 @@ async function handleRequest(request) {
|
||||
});
|
||||
}
|
||||
|
||||
// API路由处理
|
||||
switch (url.pathname) {
|
||||
// 1. 通过图片URL识别
|
||||
case '/api/recognize/url':
|
||||
if (request.method === 'POST') {
|
||||
return handleImageUrlRecognition(request);
|
||||
}
|
||||
break;
|
||||
|
||||
// 2. 通过Base64识别
|
||||
case '/api/recognize/base64':
|
||||
if (request.method === 'POST') {
|
||||
return handleBase64Recognition(request);
|
||||
}
|
||||
break;
|
||||
|
||||
// 3. 通过图片文件识别 (原有的/recognize端点)
|
||||
case '/recognize':
|
||||
if (request.method === 'POST') {
|
||||
return handleFileRecognition(request);
|
||||
}
|
||||
break;
|
||||
|
||||
// 返回前端界面
|
||||
case '/':
|
||||
return new Response(getHTML(), {
|
||||
headers: { 'Content-Type': 'text/html' },
|
||||
});
|
||||
}
|
||||
|
||||
return new Response('Not Found', { status: 404 });
|
||||
}
|
||||
|
||||
// 处理图片URL识别
|
||||
async function handleImageUrlRecognition(request) {
|
||||
try {
|
||||
const { token, imageUrl } = await request.json();
|
||||
|
||||
if (!token || !imageUrl) {
|
||||
return new Response(JSON.stringify({
|
||||
error: 'Missing token or imageUrl'
|
||||
}), {
|
||||
status: 400,
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
});
|
||||
}
|
||||
|
||||
// 下载图片
|
||||
const imageResponse = await fetch(imageUrl);
|
||||
const imageBlob = await imageResponse.blob();
|
||||
|
||||
// 上传到QwenLM
|
||||
const formData = new FormData();
|
||||
formData.append('file', imageBlob);
|
||||
|
||||
const uploadResponse = await fetch('https://chat.qwenlm.ai/api/v1/files/', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'accept': 'application/json',
|
||||
'authorization': `Bearer ${token}`,
|
||||
},
|
||||
body: formData,
|
||||
});
|
||||
|
||||
const uploadData = await uploadResponse.json();
|
||||
if (!uploadData.id) throw new Error('File upload failed');
|
||||
|
||||
// 调用识别API
|
||||
return await recognizeImage(token, uploadData.id);
|
||||
} catch (error) {
|
||||
return new Response(JSON.stringify({
|
||||
error: error.message || 'Internal Server Error'
|
||||
}), {
|
||||
status: 500,
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
// 处理Base64识别
|
||||
async function handleBase64Recognition(request) {
|
||||
try {
|
||||
const { token, base64Image } = await request.json();
|
||||
|
||||
if (!token || !base64Image) {
|
||||
return new Response(JSON.stringify({
|
||||
error: 'Missing token or base64Image'
|
||||
}), {
|
||||
status: 400,
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
});
|
||||
}
|
||||
|
||||
// 转换Base64为Blob
|
||||
const imageData = base64Image.startsWith('data:') ?
|
||||
base64Image :
|
||||
'data:image/png;base64,' + base64Image;
|
||||
|
||||
const response = await fetch(imageData);
|
||||
const blob = await response.blob();
|
||||
|
||||
// 上传到QwenLM
|
||||
const formData = new FormData();
|
||||
formData.append('file', blob);
|
||||
|
||||
const uploadResponse = await fetch('https://chat.qwenlm.ai/api/v1/files/', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'accept': 'application/json',
|
||||
'authorization': `Bearer ${token}`,
|
||||
},
|
||||
body: formData,
|
||||
});
|
||||
|
||||
const uploadData = await uploadResponse.json();
|
||||
if (!uploadData.id) throw new Error('File upload failed');
|
||||
|
||||
// 调用识别API
|
||||
return await recognizeImage(token, uploadData.id);
|
||||
} catch (error) {
|
||||
return new Response(JSON.stringify({
|
||||
error: error.message || 'Internal Server Error'
|
||||
}), {
|
||||
status: 500,
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
// 处理文件识别 (原有功能)
|
||||
async function handleFileRecognition(request) {
|
||||
// 处理 POST 请求
|
||||
if (request.method === 'POST' && url.pathname === '/recognize') {
|
||||
try {
|
||||
const { token, imageId } = await request.json();
|
||||
|
||||
if (!token || !imageId) {
|
||||
return new Response(JSON.stringify({
|
||||
error: 'Missing token or imageId'
|
||||
}), {
|
||||
return new Response(JSON.stringify({ error: 'Missing token or imageId' }), {
|
||||
status: 400,
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
});
|
||||
}
|
||||
|
||||
return await recognizeImage(token, imageId);
|
||||
} catch (error) {
|
||||
return new Response(JSON.stringify({
|
||||
error: error.message || 'Internal Server Error'
|
||||
}), {
|
||||
status: 500,
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
// 通用的识别函数
|
||||
async function recognizeImage(token, imageId) {
|
||||
// 调用 QwenLM API
|
||||
const response = await fetch('https://chat.qwenlm.ai/api/chat/completions', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
@ -195,9 +52,10 @@ async function recognizeImage(token, imageId) {
|
||||
'3. 对于行内公式使用$单个符号$\n' +
|
||||
'4. 对于独立公式块使用$$公式$$\n' +
|
||||
'5. 严格保持原文的段落格式和换行\n' +
|
||||
'6. 当文本明显换行时，使用\\n进行换行处理'
|
||||
'6. 当文本明显换行时，使用\\n进行换行处理\n' +
|
||||
'请尽可能精确地转换每个数学符号并保持原始排版格式。'
|
||||
},
|
||||
{ type: 'image', image: imageId },
|
||||
{ type: 'image', image: imageId }, // 使用上传后的图片 ID
|
||||
],
|
||||
},
|
||||
],
|
||||
@ -209,24 +67,36 @@ async function recognizeImage(token, imageId) {
|
||||
|
||||
const data = await response.json();
|
||||
|
||||
// 处理识别结果
|
||||
// 对识别结果进行后处理，确保LaTeX格式正确并保持换行
|
||||
let result = data.choices[0]?.message?.content || '识别失败';
|
||||
result = result
|
||||
// 修复可能的LaTeX格式问题
|
||||
.replace(/\\（/g, '\\(')
|
||||
.replace(/\\）/g, '\\)')
|
||||
// 确保连续的换行符被保留（2个以上的换行符表示段落分隔）
|
||||
.replace(/\n{3,}/g, '\n\n')
|
||||
// 保留单个换行符，不合并
|
||||
.replace(/([^\n])\n([^\n])/g, '$1\n$2')
|
||||
.trim();
|
||||
|
||||
return new Response(JSON.stringify({
|
||||
success: true,
|
||||
result: result
|
||||
}), {
|
||||
return new Response(JSON.stringify({ ...data, choices: [{ message: { content: result } }] }), {
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'Access-Control-Allow-Origin': '*',
|
||||
},
|
||||
});
|
||||
} catch (error) {
|
||||
return new Response(JSON.stringify({ error: 'Internal Server Error' }), {
|
||||
status: 500,
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
// 返回前端界面
|
||||
return new Response(getHTML(), {
|
||||
headers: { 'Content-Type': 'text/html' },
|
||||
});
|
||||
}
|
||||
|
||||
function getHTML() {
|
||||
@ -686,30 +556,6 @@ function getHTML() {
|
||||
' .copy-btn.copied {',
|
||||
' background: #27ae60;',
|
||||
' }',
|
||||
|
||||
' /* Base64输入相关样式 */',
|
||||
' #base64Input {',
|
||||
' width: 100%;',
|
||||
' height: 100px;',
|
||||
' padding: 10px;',
|
||||
' margin-top: 10px;',
|
||||
' border: 1px solid #dcdde1;',
|
||||
' border-radius: 8px;',
|
||||
' resize: vertical;',
|
||||
' }',
|
||||
' .toggle-btn {',
|
||||
' background: #3498db;',
|
||||
' color: white;',
|
||||
' border: none;',
|
||||
' padding: 8px 15px;',
|
||||
' border-radius: 5px;',
|
||||
' cursor: pointer;',
|
||||
' margin-top: 10px;',
|
||||
' transition: background 0.3s ease;',
|
||||
' }',
|
||||
' .toggle-btn:hover {',
|
||||
' background: #2980b9;',
|
||||
' }',
|
||||
'</style>',
|
||||
'</head>',
|
||||
'<body>',
|
||||
@ -732,11 +578,9 @@ function getHTML() {
|
||||
'<div class="upload-area" id="uploadArea">',
|
||||
'<i>📸</i>',
|
||||
'<div class="upload-text">',
|
||||
'拖拽图片到这里，点击上传，或粘贴Base64图片内容<br>',
|
||||
'拖拽图片到这里，或点击上传<br>',
|
||||
'支持复制粘贴图片',
|
||||
'</div>',
|
||||
'<textarea id="base64Input" placeholder="在此输入Base64格式的图片内容..." style="display: none; width: 100%; height: 100px; margin-top: 10px;"></textarea>',
|
||||
'<button id="toggleBase64" class="toggle-btn" style="margin-top: 10px;">切换Base64输入</button>',
|
||||
'<img id="previewImage" class="preview-image">',
|
||||
'</div>',
|
||||
'<div class="loading" id="loading"></div>',
|
||||
@ -976,15 +820,7 @@ function getHTML() {
|
||||
' });',
|
||||
|
||||
' // 点击上传',
|
||||
' uploadArea.addEventListener(\'click\', (e) => {',
|
||||
' // 如果点击的是 base64Input 或 toggleBase64 按钮，不触发文件上传',
|
||||
' if (e.target.id === \'base64Input\' || ',
|
||||
' e.target.id === \'toggleBase64\' || ',
|
||||
' e.target.closest(\'#base64Input\') || ',
|
||||
' e.target.closest(\'#toggleBase64\')) {',
|
||||
' return;',
|
||||
' }',
|
||||
|
||||
' uploadArea.addEventListener(\'click\', () => {',
|
||||
' const input = document.createElement(\'input\');',
|
||||
' input.type = \'file\';',
|
||||
' input.accept = \'image/*\';',
|
||||
@ -1039,7 +875,6 @@ function getHTML() {
|
||||
' // 复制结果功能',
|
||||
' copyBtn.addEventListener(\'click\', async () => {',
|
||||
' const result = resultDiv.textContent;',
|
||||
' toggleBase64.textContent = \'隐藏Base64输入\';',
|
||||
' try {',
|
||||
' await navigator.clipboard.writeText(result);',
|
||||
' copyBtn.textContent = \'已复制\';',
|
||||
@ -1056,66 +891,6 @@ function getHTML() {
|
||||
' document.getElementById("closeSidebar").addEventListener("click", () => {',
|
||||
' sidebar.classList.remove("open");',
|
||||
' });',
|
||||
' // Base64 输入相关功能',
|
||||
' const base64Input = document.getElementById(\'base64Input\');',
|
||||
' const toggleBase64 = document.getElementById(\'toggleBase64\');',
|
||||
|
||||
' // 切换 Base64 输入框显示',
|
||||
' toggleBase64.addEventListener(\'click\', (e) => {',
|
||||
' e.stopPropagation(); // 阻止事件冒泡到 uploadArea',
|
||||
' if (base64Input.style.display === \'none\') {',
|
||||
' base64Input.style.display = \'block\';',
|
||||
' toggleBase64.textContent = \'隐藏Base64输入\';',
|
||||
' } else {',
|
||||
' base64Input.style.display = \'none\';',
|
||||
' toggleBase64.textContent = \'切换Base64输入\';',
|
||||
' }',
|
||||
' });',
|
||||
|
||||
' // 为 base64Input 添加阻止事件冒泡',
|
||||
' document.getElementById(\'base64Input\').addEventListener(\'click\', (e) => {',
|
||||
' e.stopPropagation(); // 阻止事件冒泡到 uploadArea',
|
||||
' });',
|
||||
|
||||
' // base64Input 的 input 事件处理也需要阻止冒泡',
|
||||
' base64Input.addEventListener(\'input\', async (e) => {',
|
||||
' e.stopPropagation();',
|
||||
' const base64Content = base64Input.value.trim();',
|
||||
' if (base64Content) {',
|
||||
' try {',
|
||||
' // 尝试转换Base64为Blob',
|
||||
' let imageData;',
|
||||
' if (base64Content.startsWith(\'data:image\')) {',
|
||||
' imageData = base64Content;',
|
||||
' } else {',
|
||||
' imageData = \'data:image/png;base64,\' + base64Content;',
|
||||
' }',
|
||||
|
||||
' // 验证Base64是否为有效图片',
|
||||
' const img = new Image();',
|
||||
' img.src = imageData;',
|
||||
' await new Promise((resolve, reject) => {',
|
||||
' img.onload = resolve;',
|
||||
' img.onerror = reject;',
|
||||
' });',
|
||||
|
||||
' // 转换Base64为Blob',
|
||||
' const response = await fetch(imageData);',
|
||||
' const blob = await response.blob();',
|
||||
' const file = new File([blob], "image.png", { type: "image/png" });',
|
||||
|
||||
' // 显示预览',
|
||||
' previewImage.src = imageData;',
|
||||
' previewImage.style.display = \'block\';',
|
||||
|
||||
' // 处理图片',
|
||||
' processImage(file);',
|
||||
' } catch (error) {',
|
||||
' alert(\'无效的Base64图片内容\');',
|
||||
' console.error(\'Base64处理错误:\', error);',
|
||||
' }',
|
||||
' }',
|
||||
' });',
|
||||
'</script>',
|
||||
'</body>',
|
||||
'</html>'
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user