# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve. # # 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. import copy from pathlib import Path from .base import BaseResult from .utils.mixin import Base64Mixin class LayoutParsingResult(BaseResult): """LayoutParsingResult""" pass class VisualInfoResult(BaseResult): """VisualInfoResult""" pass class VisualResult(BaseResult): """VisualInfoResult""" def _to_str(self): return str({"layout_parsing_result": self["layout_parsing_result"]}) def save_to_html(self, save_path): if not save_path.lower().endswith(("html")): input_path = self["input_path"] save_path = Path(save_path) / f"{Path(input_path).stem}" else: save_path = Path(save_path).stem for table_result in self["table_result"]: table_result.save_to_html(save_path) def save_to_xlsx(self, save_path): if not save_path.lower().endswith(("xlsx")): input_path = self["input_path"] save_path = Path(save_path) / f"{Path(input_path).stem}" else: save_path = Path(save_path).stem for table_result in self["table_result"]: table_result.save_to_xlsx(save_path) def save_to_img(self, save_path): if not save_path.lower().endswith((".jpg", ".png")): input_path = self["input_path"] save_path = Path(save_path) / f"{Path(input_path).stem}" else: save_path = Path(save_path).stem oricls_save_path = f"{save_path}_oricls.jpg" oricls_result = self["oricls_result"] if oricls_result: oricls_result._HARD_FLAG = True oricls_result.save_to_img(oricls_save_path) uvdoc_save_path = f"{save_path}_uvdoc.jpg" unwarp_result = self["unwarp_result"] if unwarp_result: # unwarp_result._HARD_FLAG = True unwarp_result.save_to_img(uvdoc_save_path) curve_save_path = f"{save_path}_curve" curve_results = self["curve_result"] # TODO(): support list of result if isinstance(curve_results, dict): curve_results = [curve_results] for idx, curve_result in enumerate(curve_results): curve_result._HARD_FLAG = True if not unwarp_result else False curve_result.save_to_img(f"{curve_save_path}_{idx}.jpg") layout_save_path = f"{save_path}_layout.jpg" layout_result = self["layout_result"] if layout_result: layout_result._HARD_FLAG = True if not unwarp_result else False layout_result.save_to_img(layout_save_path) ocr_save_path = f"{save_path}_ocr.jpg" table_save_path = f"{save_path}_table" ocr_result = self["ocr_result"] if ocr_result: ocr_result._HARD_FLAG = True if not unwarp_result else False ocr_result.save_to_img(ocr_save_path) for idx, table_result in enumerate(self["table_result"]): table_result._HARD_FLAG = True if not unwarp_result else False table_result.save_to_img(f"{table_save_path}_{idx}.jpg") class VectorResult(BaseResult, Base64Mixin): """VisualInfoResult""" def _to_base64(self): return self["vector"] class RetrievalResult(BaseResult): """VisualInfoResult""" pass class ChatResult(BaseResult): """VisualInfoResult""" pass