商务技术评分能处理多评分表情况，规范响应返回接口，判断上传文件是否为招标文件

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):
+def judge_zbfile(file_path):
-    # reader = PdfReader(pdf_path)
+
-    # num_pages = len(reader.pages)
+    try:
-    # if num_pages <= 5:
+        # 检查文件是否存在且是pdf格式（不区分大小写）
-    #     return False
+        if file_path.lower().endswith(('.pdf', '.PDF')):
-    user_query="""该文件是否属于招标文件？如果是的话，请返回'是',如果不是的话，返回'否'。请不要返回其他解释或内容。
+            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:
+        """
+        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")