Update ocr.py

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/<uuid_str>", 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
+