feat: 增加软著
This commit is contained in:
parent
a334d5aac1
commit
442187a80d
13
main.py
13
main.py
|
@ -221,6 +221,16 @@ province_patterns = [
|
|||
("颁发日期", r"\d{4}年\d{1,2}月\d{1,2}日"),
|
||||
]
|
||||
|
||||
# 软件著作权奖
|
||||
software_patterns = [
|
||||
("软件名称", r"软件名称[::](.*?)"),
|
||||
("著作权人", r"著[\s]*作[\s]*权[\s]*人[\s]*[::](.+?)\n",),
|
||||
("开发完成日期", r"\d{4}年\d{1,2}月\d{1,2}日"),
|
||||
("首次发表日期", r"[::](.+?)"),
|
||||
("权力取得方式", r"[::](.+?)"),
|
||||
("登记号", r"登[\s]*记[\s]*号[::](.+?)"),
|
||||
]
|
||||
|
||||
@app.post(
|
||||
"/extract",
|
||||
summary="提取专利/标准等文件里的信息",
|
||||
|
@ -304,6 +314,9 @@ async def extract_info(
|
|||
elif file_type == "province":
|
||||
info = extract_social_info(ocr_text, province_patterns)
|
||||
clean_info(info)
|
||||
elif file_type == "software":
|
||||
info = extract_social_info(ocr_text, software_patterns)
|
||||
clean_info(info)
|
||||
else:
|
||||
raise HTTPException(
|
||||
400, detail="Invalid file type. Please choose 'standard' or 'patent'."
|
||||
|
|
Loading…
Reference in New Issue