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}日"),
|
("颁发日期", 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(
|
@app.post(
|
||||||
"/extract",
|
"/extract",
|
||||||
summary="提取专利/标准等文件里的信息",
|
summary="提取专利/标准等文件里的信息",
|
||||||
|
@ -304,6 +314,9 @@ async def extract_info(
|
||||||
elif file_type == "province":
|
elif file_type == "province":
|
||||||
info = extract_social_info(ocr_text, province_patterns)
|
info = extract_social_info(ocr_text, province_patterns)
|
||||||
clean_info(info)
|
clean_info(info)
|
||||||
|
elif file_type == "software":
|
||||||
|
info = extract_social_info(ocr_text, software_patterns)
|
||||||
|
clean_info(info)
|
||||||
else:
|
else:
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
400, detail="Invalid file type. Please choose 'standard' or 'patent'."
|
400, detail="Invalid file type. Please choose 'standard' or 'patent'."
|
||||||
|
|
Loading…
Reference in New Issue