PaddleOCR MCP Server

// Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. #pragma once #include <yaml-cpp/yaml.h> #include <string> #include <unordered_map> #include "absl/status/status.h" #include "absl/status/statusor.h" #include "utility.h" enum VectorType { VECTOR_INT, VECTOR_FLOAT, VECTOR_STRING, VECTOR_UNKNOWN }; struct VectorVariant { VectorType type; std::vector<int> vec_int; std::vector<float> vec_float; std::vector<std::string> vec_string; }; class YamlConfig { public: YamlConfig(const std::unordered_map<std::string, std::string> &data) : data_(data) {} YamlConfig(const std::string &model_dir); YamlConfig() = default; ~YamlConfig() = default; void Init(); std::unordered_map<std::string, std::string> PreProcessOpInfo() { return pre_process_op_info_; }; std::unordered_map<std::string, std::string> PostProcessOpInfo() { return post_process_op_info_; }; absl::StatusOr<std::string> GetString(const std::string &key, const std::string &default_value = "") const; absl::StatusOr<int> GetInt(const std::string &key, int default_value) const; absl::StatusOr<float> GetFloat(const std::string &key, float default_value) const; absl::StatusOr<double> GetDouble(const std::string &key) const; absl::StatusOr<bool> GetBool(const std::string &key, bool default_value) const; absl::StatusOr<std::unordered_map<std::string, std::string>> GetSubModule(const std::string &key) const; absl::Status HasKey(const std::string &key) const; absl::Status PrintAll() const; absl::Status PrintWithPrefix(const std::string &prefix) const; absl::Status FindPreProcessOp( const std::string &prefix = "PreProcess.transform_ops[0]") const; std::unordered_map<std::string, std::string> &Data() { return data_; }; std::string ConfigYamlPath() { return config_yaml_path_; }; absl::Status GetConfigYamlPaths(const std::string &model_dir); absl::Status LoadYamlFile(); absl::StatusOr<std::pair<std::string, std::string>> FindKey(const std::string &key); static VectorVariant SmartParseVector(const std::string &input); private: std::string config_yaml_path_; void ParseNode(const YAML::Node &node, const std::string &prefix = ""); std::unordered_map<std::string, std::string> data_; std::unordered_map<std::string, std::string> pre_process_op_info_; std::unordered_map<std::string, std::string> post_process_op_info_; };