From bac3ae7fe37230641e9453c3daa61368d6190fbf Mon Sep 17 00:00:00 2001 From: zty Date: Tue, 29 Oct 2024 16:39:59 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E6=94=B9=E6=A8=A1=E5=9E=8B?= =?UTF-8?q?=E7=9A=84=E8=AF=AD=E5=8F=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index e8d80a7..e0d805c 100644 --- a/main.py +++ b/main.py @@ -314,7 +314,14 @@ async def extract_info( prompt = f"我有以下文本,是一个专利的内容。请按专利名称，发明人，专利号，专利申请日， 专利权人，授权公告号，授权公告日为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}" elif file_type == "standard": prompt = f"我有以下文本,是一个标准的内容。请按标准号，中文名称，英文名称，发布部门，发布日期，实施日期，提出单位，归口单位，起草单位，起草人为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}" - else: + elif file_type == "country": + prompt = f"我有以下文本,是一个国家标准的内容。请按项目名称，奖励等级，获奖者，证书号，颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}" + elif file_type == "province": + prompt = f"我有以下文本,是一个省级标准的内容。请按项目名称，奖励等级，获奖者，证书编号，颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}" + elif file_type == "social": + prompt = f"我有以下文本,是一个社会标准的内容。请按项目名称，获奖单位，奖励等级，获奖人，奖励年度,证书编号,颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}" + elif file_type=="building": + prompt = f"我有以下文本,是一个建筑标准的内容。请按项目名称，奖励等级，获奖单位，证书编号，颁发日期为key的json格式返回数据,注意只返回json数据。文本如下：{ocr_text}" raise HTTPException( 400, detail="Invalid file type. Please choose 'standard' or 'patent'." )