From ac820e3fc7807a1b06d41c85dddfca77f8975f30 Mon Sep 17 00:00:00 2001
From: wuzekai <3025054974@qq.com>
Date: Tue, 8 Jul 2025 05:23:11 +0000
Subject: [PATCH] Update ocr.py

---
 ocr.py | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/ocr.py b/ocr.py
index d45fe64..8725d7b 100644
--- a/ocr.py
+++ b/ocr.py
@@ -62,4 +62,26 @@ def ocr_inference():
                 results.append({"text": text, "confidence": confidence})
 
         if not results:
-            return jsonify({"error": "No valid OCR debug output found"}), 500
\ No newline at end of file
+            return jsonify({"error": "No valid OCR debug output found"}), 500
+            
+        # 保存为 JSON 文件
+        json_filename = f"{uuid_str}.json"
+        json_path = os.path.join(output_dir, json_filename)
+        with open(json_path, "w", encoding="utf-8") as f:
+            json.dump(results, f, ensure_ascii=False, indent=2)
+
+        # 读取 JSON 文件内容并返回
+        with open(json_path, "r", encoding="utf-8") as f:
+            json_content = json.load(f)
+
+        # 返回结果
+        return jsonify({
+            "message": "File processed successfully",
+            "uuid": uuid_str,
+            "content": json_content,
+        }), 200
+
+    except subprocess.CalledProcessError as e:
+        return jsonify({"error": f"OCR subprocess failed: {e}"}), 500
+    except Exception as e:
+        return jsonify({"error": f"Unexpected error: {e}"}), 500
\ No newline at end of file
-- 
2.22.0