feat: 增加软著

This commit is contained in:
zty 2024-11-15 09:36:48 +08:00
parent a334d5aac1
commit 442187a80d
1 changed files with 13 additions and 0 deletions

13
main.py
View File

@ -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'."