zbparse/flask_app/general/判断是否是招标文件.py

30 lines
1.3 KiB
Python
Raw Normal View History

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:
return False
return True
if __name__ == '__main__':
pdf_path=r"C:\Users\Administrator\Desktop\货物标\zbfiles\zbtest4_evaluation_method.pdf"
res=judge_zbfile(pdf_path)
if res:
print("yes")
else:
print("no")