Update README.md

2025-02-15 09:49:03 +08:00 · 2025-02-13 13:48:59 +08:00 · 2025-02-13 13:47:19 +08:00 · 2025-02-13 13:15:20 +08:00 · 2025-02-13 09:20:15 +08:00 · 2025-02-10 12:15:24 +08:00
2 changed files with 151 additions and 66 deletions
--- a/README.md
+++ b/README.md
@@ -1,6 +1,10 @@
-# 🖼️ QwenLM OCR 
+<div align="center"> <img src="https://github.com/Cunninger/ocr-based-qwen/blob/main/ocr.png?raw=true" width="120"> </div>

-本项目是对 [QwenLM](https://chat.qwenlm.ai/) 的 OCR 功能进行逆向工程的实现。通过调用 QwenLM 的 API，你可以从图片中提取文字内容，并且该项目支持一键部署到 **Cloudflare Workers** (CF) 上。
+
+#               QwenLM OCR 
+
+
+本项目基于 [QwenLM](https://chat.qwenlm.ai/) 。通过调用 QwenLM 的 API，你可以从图片中提取文字内容，并且该项目支持一键部署到 **Cloudflare Workers** (CF) 上。

 ## 项目展示
 ![image](https://github.com/user-attachments/assets/9e794686-f914-4fb7-9931-1ada7c22bb2d)
@@ -17,7 +21,7 @@ cna=LI8HIAirQyACAWrg3cNOjH8F; _gcl_au=1.1.1200556497.1736728490; xlly_s=1; acw_t
 - **历史记录**：保存每次识别的结果和图片，方便查看。
 - **一键复制**：轻松复制识别结果到剪贴板。
 - **数学公式识别**：特别优化了对数学公式的提取，支持 LaTeX 格式输出。
- **API 支持**：提供 `curl` 接口调用，支持 base64 和图片 URL 两种方式。
+- **API 支持**：提供 `curl` 接口调用，支持通过图片文件、base64 和图片 URL 3种方式。(Apifox调用文档示例(仅作为代码示例，尝试本地调试)：https://we4q3dzb3r.apifox.cn/)
 - **验证码识别**：新增验证码识别功能，支持常见类型的验证码（如数字、字母、混合字符等），提升自动化处理能力。
 ## 🛠️ 部署指南

@@ -57,20 +61,22 @@ cna=LI8HIAirQyACAWrg3cNOjH8F; _gcl_au=1.1.1200556497.1736728490; xlly_s=1; acw_t
 5. **API 调用**：
   - **支持 base64**：
     ```bash
-     curl --location 'https://ocr.doublefenzhuan.me/api/recognize/base64' \
-     --header 'Content-Type: application/json' \
-     --data '{
-         "token": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZCI6IjUzZTk0Nzg4LWMwM2QtNDY4Mi05OTNhLWE0ZDNjNGUyZDY0OSIsImV4cCI6MTczOTA3NTE0MX0.FtwG6xDLYd2rngWUhuldg56WXCiLSTL0RI6xJJQ4vHM",
-         "base64Image": "xxx"
-     }'
+      curl -X POST \
+        'https://test-qwen-cor.aughumes8.workers.dev/api/recognize/base64' \
+        -H 'Content-Type: application/json' \
+        -H 'x-custom-cookie: YOUR_COOKIE_STRING' \
+        -d '{
+          "base64Image": "YOUR_BASE64_IMAGE_STRING"
+        }'
     ```
   - **支持图片 URL**:
     ```bash
-     curl --location 'https://ocr.doublefenzhuan.me/api/recognize/url' \
-     --header 'Content-Type: application/json' \
-     --data '{
-         "token": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZCI6IjUzZTk0Nzg4LWMwM2QtNDY4Mi05OTNhLWE0ZDNjNGUyZDY0OSIsImV4cCI6MTczOTA3NTE0MX0.FtwG6xDLYd2rngWUhuldg56WXCiLSTL0RI6xJJQ4vHM",
-         "imageUrl": "xxxx"
+      curl -X POST \
+        'https://test-qwen-cor.aughumes8.workers.dev/api/recognize/url' \
+        -H 'Content-Type: application/json' \
+        -H 'x-custom-cookie: YOUR_COOKIE_STRING' \
+        -d '{
+          "imageUrl": "YOUR_IMAGE_URL"
     }'
     ```
 6. **验证码识别**
@@ -115,29 +121,48 @@ cna=LI8HIAirQyACAWrg3cNOjH8F; _gcl_au=1.1.1200556497.1736728490; xlly_s=1; acw_t

 ### 2025/01/13 18点34分 支持`curl`接口调用
 - **支持base64**：
-```
-curl --location 'https://ocr.doublefenzhuan.me/api/recognize/base64' \
--header 'Content-Type: application/json' \
--data '{
-    "token": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZCI6IjUzZTk0Nzg4LWMwM2QtNDY4Mi05OTNhLWE0ZDNjNGUyZDY0OSIsImV4cCI6MTczOTA3NTE0MX0.FtwG6xDLYd2rngWUhuldg56WXCiLSTL0RI6xJJQ4vHM",
-    "base64Image": "xxx"
-}'
+```bash
+curl -X POST \
+  'https://test-qwen-cor.aughumes8.workers.dev/api/recognize/base64' \
+  -H 'Content-Type: application/json' \
+  -H 'x-custom-cookie: YOUR_COOKIE_STRING' \
+  -d '{
+    "base64Image": "YOUR_BASE64_IMAGE_STRING"
+  }'
 ```
 - 效果图：
-![image](https://github.com/user-attachments/assets/ef160aae-e741-49d3-96f0-a0969b883f1a)
+![image](https://github.com/user-attachments/assets/363afb39-444b-4308-a2bd-55831df81b6f)

 - **支持图片URL**:
 ```bash
-curl --location 'https://ocr.doublefenzhuan.me/api/recognize/url' \
--header 'Content-Type: application/json' \
--data '{
-    "token": "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZCI6IjUzZTk0Nzg4LWMwM2QtNDY4Mi05OTNhLWE0ZDNjNGUyZDY0OSIsImV4cCI6MTczOTA3NTE0MX0.FtwG6xDLYd2rngWUhuldg56WXCiLSTL0RI6xJJQ4vHM",
-    
-     "imageUrl": "xxxx"
-}'
+curl -X POST \
+  'https://test-qwen-cor.aughumes8.workers.dev/api/recognize/url' \
+  -H 'Content-Type: application/json' \
+  -H 'x-custom-cookie: YOUR_COOKIE_STRING' \
+  -d '{
+    "imageUrl": "YOUR_IMAGE_URL"
+  }'
 ```
 - 效果图：
-![image](https://github.com/user-attachments/assets/db0c89f9-96f1-45b1-b1e9-88ac3d01e196)
+![image](https://github.com/user-attachments/assets/a816085e-8a52-4425-b02c-94ea543bf95b)
+
+- **通过图片文件识别（需要先上传获取imageId）**
+```bash
+# 1. 先上传文件
+curl -X POST \
+  'https://test-qwen-cor.aughumes8.workers.dev/proxy/upload' \
+  -H 'x-custom-cookie: YOUR_COOKIE_STRING' \
+  -F 'file=@/path/to/your/image.jpg'
+
+# 2. 使用返回的imageId进行识别
+curl -X POST \
+  'https://test-qwen-cor.aughumes8.workers.dev/recognize' \
+  -H 'Content-Type: application/json' \
+  -H 'x-custom-cookie: YOUR_COOKIE_STRING' \
+  -d '{
+    "imageId": "RETURNED_IMAGE_ID"
+  }'
+```
 ## Cloudflare访问数据
 ![image](https://github.com/user-attachments/assets/bb456075-6107-47ee-a361-a0edba532c38)

--- a/worker.js
+++ b/worker.js
@@ -60,17 +60,30 @@ async function handleRequest(request) {
 // 处理图片URL识别
 async function handleImageUrlRecognition(request) {
  try {
-    const { token, imageUrl } = await request.json();
+    const { imageUrl } = await request.json();
+    const cookie = request.headers.get('x-custom-cookie');

-    if (!token || !imageUrl) {
+    if (!cookie || !imageUrl) {
      return new Response(JSON.stringify({
-        error: 'Missing token or imageUrl'
+        error: 'Missing cookie or imageUrl'
      }), {
        status: 400,
        headers: { 'Content-Type': 'application/json' },
      });
    }

+    // 从cookie中提取token
+    const tokenMatch = cookie.match(/token=([^;]+)/);
+    if (!tokenMatch) {
+      return new Response(JSON.stringify({
+        error: 'Invalid cookie format: missing token'
+      }), {
+        status: 400,
+        headers: { 'Content-Type': 'application/json' },
+      });
+    }
+    const token = tokenMatch[1];
+
    // 下载图片
    const imageResponse = await fetch(imageUrl);
    const imageBlob = await imageResponse.blob();
@@ -84,6 +97,7 @@ async function handleImageUrlRecognition(request) {
      headers: {
        'accept': 'application/json',
        'authorization': `Bearer ${token}`,
+        'cookie': cookie
      },
      body: formData,
    });
@@ -91,8 +105,8 @@ async function handleImageUrlRecognition(request) {
    const uploadData = await uploadResponse.json();
    if (!uploadData.id) throw new Error('File upload failed');

-    // 调用识别API
-    return await recognizeImage(token, uploadData.id);
+    // 调用通用识别函数
+    return await recognizeImage(token, uploadData.id, request);
  } catch (error) {
    return new Response(JSON.stringify({
      error: error.message || 'Internal Server Error'
@@ -106,22 +120,32 @@ async function handleImageUrlRecognition(request) {
 // 处理Base64识别
 async function handleBase64Recognition(request) {
  try {
-    const { token, base64Image } = await request.json();
+    const { base64Image } = await request.json();
+    const cookie = request.headers.get('x-custom-cookie');

-    if (!token || !base64Image) {
+    if (!cookie || !base64Image) {
      return new Response(JSON.stringify({
-        error: 'Missing token or base64Image'
+        error: 'Missing cookie or base64Image'
      }), {
        status: 400,
        headers: { 'Content-Type': 'application/json' },
      });
    }

-    // 转换Base64为Blob
-    const imageData = base64Image.startsWith('data:') ?
-      base64Image :
-      'data:image/png;base64,' + base64Image;
+    // 从cookie中提取token
+    const tokenMatch = cookie.match(/token=([^;]+)/);
+    if (!tokenMatch) {
+      return new Response(JSON.stringify({
+        error: 'Invalid cookie format: missing token'
+      }), {
+        status: 400,
+        headers: { 'Content-Type': 'application/json' },
+      });
+    }
+    const token = tokenMatch[1];

+    // 转换Base64为Blob
+    const imageData = base64Image.startsWith('data:') ? base64Image : 'data:image/png;base64,' + base64Image;
    const response = await fetch(imageData);
    const blob = await response.blob();

@@ -134,6 +158,7 @@ async function handleBase64Recognition(request) {
      headers: {
        'accept': 'application/json',
        'authorization': `Bearer ${token}`,
+        'cookie': cookie
      },
      body: formData,
    });
@@ -141,8 +166,8 @@ async function handleBase64Recognition(request) {
    const uploadData = await uploadResponse.json();
    if (!uploadData.id) throw new Error('File upload failed');

-    // 调用识别API
-    return await recognizeImage(token, uploadData.id);
+    // 调用通用识别函数
+    return await recognizeImage(token, uploadData.id, request);
  } catch (error) {
    return new Response(JSON.stringify({
      error: error.message || 'Internal Server Error'
@@ -156,18 +181,23 @@ async function handleBase64Recognition(request) {
 // 处理文件识别 (原有功能)
 async function handleFileRecognition(request) {
  try {
-    const { token, imageId } = await request.json();
+    const { imageId } = await request.json();
+    const cookie = request.headers.get('x-custom-cookie') || '';

-    if (!token || !imageId) {
+    if (!cookie || !imageId) {
      return new Response(JSON.stringify({
-        error: 'Missing token or imageId'
+        error: 'Missing cookie or imageId'
      }), {
        status: 400,
        headers: { 'Content-Type': 'application/json' },
      });
    }

-    return await recognizeImage(token, imageId, request); // 传递 request 对象
+    // 从cookie中提取token
+    const tokenMatch = cookie.match(/token=([^;]+)/);
+    const token = tokenMatch ? tokenMatch[1] : '';
+
+    return await recognizeImage(token, imageId, request);
  } catch (error) {
    return new Response(JSON.stringify({
      error: error.message || 'Internal Server Error'
@@ -182,13 +212,18 @@ async function handleFileRecognition(request) {
 async function handleProxyUpload(request) {
  try {
    const formData = await request.formData();
-    const token = request.headers.get('Authorization').replace('Bearer ', '');
+    const cookie = request.headers.get('x-custom-cookie') || '';
+    
+    // 从cookie中提取token
+    const tokenMatch = cookie.match(/token=([^;]+)/);
+    const token = tokenMatch ? tokenMatch[1] : '';
    
    const response = await fetch('https://chat.qwenlm.ai/api/v1/files/', {
      method: 'POST',
      headers: {
        'accept': 'application/json',
        'authorization': `Bearer ${token}`,
+        'cookie': cookie
      },
      body: formData,
    });
@@ -216,7 +251,7 @@ async function handleProxyUpload(request) {

 // 通用的识别函数
 async function recognizeImage(token, imageId, request) {
-  // 从请求头获取 cookie
+  const cookie = request.headers.get('x-custom-cookie') || '';
  const response = await fetch('https://chat.qwenlm.ai/api/chat/completions', {
    method: 'POST',
    headers: {
@@ -224,7 +259,7 @@ async function recognizeImage(token, imageId, request) {
      'User-Agent': 'PostmanRuntime/7.43.0', 
      'accept': '*/*',
      'authorization': `Bearer ${token}`,
-      'cookie': request.headers.get('x-custom-cookie') || '', // 使用自定义请求头中的 cookie
+      'cookie': cookie,
    },
    body: JSON.stringify({
      stream: false,
@@ -1015,7 +1050,6 @@ function getHTML() {
    '      background: rgba(255, 255, 255, 0.9);',
    '      backdrop-filter: blur(5px);',
    '      z-index: 900;',
-    '      border-top: 1px solid rgba(0, 0, 0, 0.1);',
    ' display: flex;',           
    '      justify-content: center;', 
    '      align-items: center;',     
@@ -1080,13 +1114,13 @@ function getHTML() {

 '    .github-link {',
 '      position: fixed;',
-'      right: 150px;', // 放在 token 设置按钮左边',
+'      right: 150px;',
 '      top: 20px;',
-'      background: #333;', // GitHub 的深色背景',
+'      background: #333;',
 '      color: white;',
 '      border: none;',
 '      padding: 10px;',
-'      border-radius: 50%;', // 圆形按钮',
+'      border-radius: 50%;',
 '      cursor: pointer;',
 '      z-index: 1001;',
 '      width: 40px;',
@@ -1098,7 +1132,7 @@ function getHTML() {
 '    }',

 '    .github-link:hover {',
-'      background: #24292e;', // GitHub hover 颜色',
+'      background: #24292e;',
 '    }',

 '    .github-icon {',
@@ -1144,10 +1178,34 @@ function getHTML() {
 '      color: #3498db;',
 '      font-family: monospace;',
 '      word-break: break-all;',
+'    }',
+
+// 添加获取cookie按钮样式
+'    .get-cookie-link {',
+'      position: fixed;', 
+'      right: 200px;', // 放在github图标左边
+'      top: 20px;',
+'      background: #2ecc71;', // 使用不同的颜色区分
+'      color: white;',
+'      border: none;',
+'      padding: 8px 15px;',
+'      border-radius: 5px;',
+'      cursor: pointer;',
+'      z-index: 1001;',
+'      text-decoration: none;',
+'      font-size: 14px;',
+'      transition: background 0.3s ease;',
+'    }',
+
+'    .get-cookie-link:hover {',
+'      background: #27ae60;',
 '    }',
    '</style>',
    '</head>',
    '<body>',
+    '<a href="https://chat.qwenlm.ai/" target="_blank" rel="noopener noreferrer" class="get-cookie-link">',
+    '  获取Cookie',
+    '</a>',
    '<a href="https://github.com/Cunninger/ocr-based-qwen" target="_blank" rel="noopener noreferrer" class="github-link" title="View on GitHub">',
    '  <svg class="github-icon" viewBox="0 0 16 16" fill="currentColor">',
    '    <path fill-rule="evenodd" d="M8 0C3.58 0 0 3.58 0 8c0 3.54 2.29 6.53 5.47 7.59.4.07.55-.17.55-.38 0-.19-.01-.82-.01-1.49-2.01.37-2.53-.49-2.69-.94-.09-.23-.48-.94-.82-1.13-.28-.15-.68-.52-.01-.53.63-.01 1.08.58 1.23.82.72 1.21 1.87.87 2.33.66.07-.52.28-.87.51-1.07-1.78-.2-3.64-.89-3.64-3.95 0-.87.31-1.59.82-2.15-.08-.2-.36-1.02.08-2.12 0 0 .67-.21 2.2.82.64-.18 1.32-.27 2-.27.68 0 1.36.09 2 .27 1.53-1.04 2.2-.82 2.2-.82.44 1.1.16 1.92.08 2.12.51.56.82 1.27.82 2.15 0 3.07-1.87 3.75-3.65 3.95.29.25.54.73.54 1.48 0 1.07-.01 1.93-.01 2.2 0 .21.15.46.55.38A8.013 8.013 0 0016 8c0-4.42-3.58-8-8-8z"></path>',
@@ -1421,6 +1479,12 @@ function getHTML() {
    '      }',

    '      const savedCookie = localStorage.getItem(\'imageRecognitionCookie\');',
+    '      if (!savedCookie) {',
+    '        alert(\'请先设置Cookie\');',
+    '        sidebar.classList.add(\'open\');',
+    '        return;',
+    '      }',
+
    '      // 显示图片预览',
    '      const reader = new FileReader();',
    '      let imageData;',
@@ -1443,8 +1507,7 @@ function getHTML() {
    '        const uploadResponse = await fetch(\'/proxy/upload\', {',
    '          method: \'POST\',',
    '          headers: {',
-    '            \'Authorization\': \'Bearer \' + currentToken,',
-    '            \'x-custom-cookie\': savedCookie || \'\', // 添加 cookie 到请求头',
+    '            \'x-custom-cookie\': savedCookie,',
    '          },',
    '          body: formData,',
    '        });',
@@ -1455,22 +1518,19 @@ function getHTML() {
    '        // 识别图片',
    '        const recognizeResponse = await fetch(\'/recognize\', {',
    '          method: \'POST\',',
-    '          headers: { \'Content-Type\': \'application/json\' },',
-    '          body: JSON.stringify({ ',
-    '            token: currentToken, ',
-    '            imageId: uploadData.id ',
-    '          }),',
+    '          headers: { ',
+    '            \'Content-Type\': \'application/json\',',
+    '            \'x-custom-cookie\': savedCookie,',
+    '          },',
+    '          body: JSON.stringify({ imageId: uploadData.id }),',
    '        });',

    '        const recognizeData = await recognizeResponse.json();',
-
-    '        // 修改这里：使用新的响应格式',
    '        if (!recognizeData.success) {',
    '          throw new Error(recognizeData.error || \'识别失败\');',
    '        }',

    '        const result = recognizeData.result || \'识别失败\';',
-    '        // 保存原始文本到属性中，确保 LaTeX 格式完整',
    '        resultDiv.setAttribute(\'data-original-text\', result);',
    '        resultDiv.innerHTML = result;',
    '        waitForMathJax(() => {',
Author	SHA1	Message	Date
programmerWsy	2637226c23	Update README.md	2025-02-15 09:49:03 +08:00
programmerWsy	f638c5d65e	Update README.md	2025-02-13 13:48:59 +08:00
programmerWsy	6bd52ea74a	Update README.md	2025-02-13 13:47:19 +08:00
cunninger	4e55b325e0	feat: add "Get Cookie" button next to GitHub icon - Introduced a new green "Get Cookie" button to the UI. - Positioned the button to the right of the history button for easy access. - Implemented functionality to open the URL `https://chat.qwenlm.ai` in a new tab when the button is clicked. - Enhanced user experience by providing a direct link for users to obtain their cookie.	2025-02-13 13:15:20 +08:00
programmerWsy	2f8f57ade7	Update README.md	2025-02-13 09:20:15 +08:00
programmerWsy	ad532bd994	docs: update API endpoint examples	2025-02-10 12:15:24 +08:00
programmerWsy	9c6745fcf5	docs: update API endpoint examples and usage notes	2025-02-10 12:02:12 +08:00
programmerWsy	d7e0bf4106	docs: update API endpoint examples and usage notes	2025-02-10 11:54:33 +08:00
cunninger	bee8754e24	feat: enhance API implementation and add documentation - Added detailed comments to API endpoint implementations for clarity. - Ensured consistent error handling across all endpoints with standardized JSON responses. - Implemented unit tests for API endpoints to verify functionality. - Updated constants and added comments for better understanding of parameters. - Ensured code logic aligns with the updated README documentation.	2025-02-10 11:48:05 +08:00