From d3425adcfb76648b2e20c6b5e6831fc4af4b0af4 Mon Sep 17 00:00:00 2001 From: zy123 <646228430@qq.com> Date: Thu, 9 Jan 2025 16:12:31 +0800 Subject: [PATCH] =?UTF-8?q?=E5=95=86=E5=8A=A1=E6=8A=80=E6=9C=AF=E8=AF=84?= =?UTF-8?q?=E5=88=86=E8=83=BD=E5=A4=84=E7=90=86=E5=A4=9A=E8=AF=84=E5=88=86?= =?UTF-8?q?=E8=A1=A8=E6=83=85=E5=86=B5=EF=BC=8C=E8=A7=84=E8=8C=83=E5=93=8D?= =?UTF-8?q?=E5=BA=94=E8=BF=94=E5=9B=9E=E6=8E=A5=E5=8F=A3=EF=BC=8C=E5=88=A4?= =?UTF-8?q?=E6=96=AD=E4=B8=8A=E4=BC=A0=E6=96=87=E4=BB=B6=E6=98=AF=E5=90=A6?= =?UTF-8?q?=E4=B8=BA=E6=8B=9B=E6=A0=87=E6=96=87=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- flask_app/general/判断是否是招标文件.py | 42 +++++++++++++++---------- 1 file changed, 25 insertions(+), 17 deletions(-) diff --git a/flask_app/general/判断是否是招标文件.py b/flask_app/general/判断是否是招标文件.py index 1b66766..c1b90d1 100644 --- a/flask_app/general/判断是否是招标文件.py +++ b/flask_app/general/判断是否是招标文件.py @@ -3,26 +3,34 @@ from PyPDF2 import PdfReader from flask_app.general.通义千问long import upload_file, qianwen_long -def judge_zbfile(pdf_path): - # reader = PdfReader(pdf_path) - # num_pages = len(reader.pages) - # if num_pages <= 5: - # return False - user_query="""该文件是否属于招标文件？如果是的话，请返回'是',如果不是的话，返回'否'。请不要返回其他解释或内容。 -以下是常见的招标文件类型： - 公开招标文件、邀请招标文件、竞争性谈判文件、竞争性磋商文件、询价文件、问询文件、货物类招标文件、工程类招标文件、施工类招标文件、服务类招标文件、比选文件。 -若有未涵盖的类型，但其内容明确表达了项目需求、采购或招标信息，且包含指导投标人参与的关键要素，则可视为招标文件。 -请基于上述内容判断文件是否属于招标文件。 - """ - file_id=upload_file(pdf_path) - model_res=qianwen_long(file_id,user_query) - print(f"判断是否属于招标文件：{model_res}") - if '否' in model_res: +def judge_zbfile(file_path): + + try: + # 检查文件是否存在且是pdf格式（不区分大小写） + if file_path.lower().endswith(('.pdf', '.PDF')): + reader = PdfReader(file_path) + num_pages = len(reader.pages) + if num_pages <= 5: + return False + user_query = """该文件是否属于招标文件？如果是的话，请返回'是',如果不是的话，返回'否'。请不要返回其他解释或内容。 + 以下是常见的招标文件类型： + 公开招标文件、邀请招标文件、竞争性谈判文件、竞争性磋商文件、询价文件、问询文件、货物类招标文件、工程类招标文件、施工类招标文件、服务类招标文件、比选文件。 + 若有未涵盖的类型，但其内容明确表达了项目需求、采购或招标信息，且包含指导投标人参与的关键要素，则可视为招标文件。 + 请基于上述内容判断文件是否属于招标文件。 + """ + file_id = upload_file(pdf_path) + model_res = qianwen_long(file_id, user_query) + print(f"判断是否属于招标文件：{model_res}") + if '否' in model_res: + return False + return True + + except Exception as e: + print(f"处理PDF文件时出错: {e}") return False - return True if __name__ == '__main__': - pdf_path=r"C:\Users\Administrator\Desktop\货物标\zbfiles\zbtest4_evaluation_method.pdf" + pdf_path=r"C:\Users\Administrator\Desktop\测试信号测试信号.docx" res=judge_zbfile(pdf_path) if res: print("yes")