diff --git a/main.py b/main.py index e0d805c..0f4d14a 100644 --- a/main.py +++ b/main.py @@ -284,7 +284,7 @@ async def extract_info( ocr_text = "\n".join( [line[1][0] for res in result if res is not None for line in res] ) - # print("ocr_text", ocr_text) + print("ocr_text", ocr_text) # 提取信息 if extract_method == "re": if file_type == "patent": @@ -321,10 +321,11 @@ async def extract_info( elif file_type == "social": prompt = f"我有以下文本,是一个社会标准的内容。请按项目名称,获奖单位,奖励等级,获奖人,奖励年度,证书编号,颁发日期为key的json格式返回数据,注意只返回json数据。文本如下:{ocr_text}" elif file_type=="building": - prompt = f"我有以下文本,是一个建筑标准的内容。请按项目名称,奖励等级,获奖单位,证书编号,颁发日期为key的json格式返回数据,注意只返回json数据。文本如下:{ocr_text}" + prompt = f"我有以下文本,是一个建筑标准的内容。请按项目名称,奖励等级,获奖单位,证书编号,颁发日期为key的json格式返回数据,注意只返回json数据。文本如下:{ocr_text}" + else: raise HTTPException( - 400, detail="Invalid file type. Please choose 'standard' or 'patent'." - ) + 400, detail="Invalid file type. Please choose 'standard' or 'patent'." + ) r = requests.post( conf.CHAT_API, json={"model": conf.CHAT_MODEL, "prompt": prompt, "stream": False},