From 96e3991e43e835554bc7fa7f1a4be6dbea9a211a Mon Sep 17 00:00:00 2001 From: wuzekai <3025054974@qq.com> Date: Tue, 8 Jul 2025 05:24:13 +0000 Subject: [PATCH] Update ocr.py --- ocr.py | 20 +++++++++++++++++++- 1 file changed, 19 insertions(+), 1 deletion(-) diff --git a/ocr.py b/ocr.py index e43a505..bdac9b9 100644 --- a/ocr.py +++ b/ocr.py @@ -106,4 +106,22 @@ def get_text(uuid_str): "content": text_list }), 200 except Exception as e: - return jsonify({"error": f"Failed to read JSON file: {str(e)}"}), 500 \ No newline at end of file + return jsonify({"error": f"Failed to read JSON file: {str(e)}"}), 500 + +@ocr.route("/download/", methods=["GET"]) +def download_image(uuid_str): + if not is_valid_uuid(uuid_str): + return jsonify({"error": "Invalid UUID format"}), 400 + # 构建图片文件路径 + image_file = os.path.join(result_dir, f"{uuid_str}.jpg") + if not os.path.exists(image_file): + return jsonify({"error": "Image file not found"}), 404 + try: + return send_file( + image_file, + as_attachment=True, + download_name=f"{uuid_str}.jpg" + ) + except Exception as e: + return jsonify({"error": f"Failed to send image: {str(e)}"}), 500 + -- 2.22.0