diff --git a/apps/riic_report_analysis.py b/apps/riic_report_analysis.py index f99ebfb..f55fdb9 100644 --- a/apps/riic_report_analysis.py +++ b/apps/riic_report_analysis.py @@ -1,7 +1,19 @@ -from pepperbot.core.message.chain import MessageChain, Text +from pepperbot.core.message.chain import MessageChain, Image, Text +from paddleocr import PaddleOCR +import os class RIICReportAnalysis: async def group_message(self, chain: MessageChain): - if chain.pure_text == "芜湖": - await chain.onebot_reply(Text("起飞")) + if len(chain) != 1: + return + img_seg: Image = chain[0] + img_path = await img_seg.download() + print(f"Image saved to {img_path}") + + ocr = PaddleOCR(use_angle_cls=False, lang="ch", use_gpu=False) + ocr_text = ocr.ocr(img_path, det=False) + print(ocr_text) + + os.remove(img_path) + print(f"file removed.")