diff --git a/flask_app/routes/judge_zbfile.py b/flask_app/routes/judge_zbfile.py index ba43a31..b34df9b 100644 --- a/flask_app/routes/judge_zbfile.py +++ b/flask_app/routes/judge_zbfile.py @@ -36,7 +36,7 @@ def judge_zbfile() -> Any: #判断是否是招标文件 # ) result = pool.apply( judge_zbfile_exec, # 你的实际执行函数 - args=(file_path,) + args=(file_path,logger,) ) return result def wrapper() -> None: diff --git a/flask_app/routes/判断是否是招标文件.py b/flask_app/routes/判断是否是招标文件.py index d608b40..8437fb7 100644 --- a/flask_app/routes/判断是否是招标文件.py +++ b/flask_app/routes/判断是否是招标文件.py @@ -4,7 +4,7 @@ from docx import Document import fitz from flask_app.general.llm.通义千问long import upload_file, qianwen_long -def judge_zbfile_exec(file_path): +def judge_zbfile_exec(file_path,logger): """ 判断文件是否属于招标文件,并返回结果。 """ @@ -45,7 +45,7 @@ def judge_zbfile_exec(file_path): print("DOCX 文件读取失败,可能为乱码文件") return False # 解析失败直接返回 False pre_endtime=time.time() - print(f"judge_zbfile_exec预处理耗时:{pre_endtime - start_time:.2f} 秒") + logger.info(f"judge_zbfile_exec预处理耗时:{pre_endtime - start_time:.2f} 秒") # 使用大模型进行判断 user_query = """该文件是否属于招标文件?如果是的话,请返回'是',如果不是的话,返回'否'。请不要返回其他解释或内容。 以下是常见的招标文件类型: @@ -59,8 +59,8 @@ def judge_zbfile_exec(file_path): file_id = upload_file(file_path) model_res = qianwen_long(file_id, user_query) end_time = time.time() - print(f"judge_zbfile_exec实际耗时:{end_time - start_time:.2f} 秒") - print(f"判断是否属于招标文件:{model_res}") + logger.info(f"judge_zbfile_exec实际耗时:{end_time - start_time:.2f} 秒") + print(f"判断是否属于招标文件:{model_res} 实际耗时:{end_time - start_time:.2f} 秒") return '否' not in model_res diff --git a/flask_app/run_serve.py b/flask_app/run_serve.py index f6d8d55..233e441 100644 --- a/flask_app/run_serve.py +++ b/flask_app/run_serve.py @@ -20,5 +20,5 @@ def warmup_request(): if __name__ == "__main__": app = create_app() - # threading.Thread(target=warmup_request, daemon=True).start() + threading.Thread(target=warmup_request, daemon=True).start() serve(app, host='0.0.0.0', port=5000) \ No newline at end of file diff --git a/flask_app/start_up.py b/flask_app/start_up.py index 23b0fcf..75ae428 100644 --- a/flask_app/start_up.py +++ b/flask_app/start_up.py @@ -16,7 +16,7 @@ from flask_app.routes.test_preprocess import test_process_bp def create_app(): # 创建全局日志记录器 app = Flask(__name__) - app.process_pool = Pool(processes=5, maxtasksperchild=1) + app.process_pool = Pool(processes=3, maxtasksperchild=1) app.global_logger = create_logger_main('model_log') # 全局日志记录器 # 注册蓝图 app.register_blueprint(get_deviation_bp)