Commit ac820e3f authored by wuzekai's avatar wuzekai

Update ocr.py

parent d7cdc1a3
Pipeline #576 canceled with stages
...@@ -62,4 +62,26 @@ def ocr_inference(): ...@@ -62,4 +62,26 @@ def ocr_inference():
results.append({"text": text, "confidence": confidence}) results.append({"text": text, "confidence": confidence})
if not results: if not results:
return jsonify({"error": "No valid OCR debug output found"}), 500 return jsonify({"error": "No valid OCR debug output found"}), 500
\ No newline at end of file
# 保存为 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
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment