chunzhi/chatgpt_academic
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修复关于abstract的搜索

Browse Source
This commit is contained in:
OverKit 2023-06-17 19:27:21 +08:00
parent cbcb98ef6a
commit 9a5a509dd9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
crazy_functions/latex_utils.py
View File

@ -424,7 +424,9 @@ class LatexPaperSplit():
if mode == 'translate_zh': if mode == 'translate_zh':
pattern = re.compile(r'\\begin\{abstract\}.*\n') pattern = re.compile(r'\\begin\{abstract\}.*\n')
match = pattern.search(result_string) match = pattern.search(result_string)
assert match is not None, "Cannot find paper abstract section!" if not match:
pattern = re.compile(r'\\abstract\{')
match = pattern.search(result_string)
position = match.end() position = match.end()
result_string = result_string[:position] + self.msg + msg + self.msg_declare + result_string[position:] result_string = result_string[:position] + self.msg + msg + self.msg_declare + result_string[position:]
return result_string return result_string