diff --git a/ocr.py b/ocr.py index e43a5052f8618ba8b8477ad94d196c86f7f16e23..bdac9b982429a8e452b7c7f1bf9e15126c70bb09 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 +