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

declare resp

Browse Source
This commit is contained in:
qingxu fu 2023-06-05 23:24:41 +08:00
parent cea2144f34
commit 66018840da
1 changed files with 10 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
crazy_functions/latex_utils.py
View File

@ -189,6 +189,10 @@ class LatexPaperSplit():
root是链表的根节点 root是链表的根节点
""" """
self.root = None self.root = None
self.msg = "{\\scriptsize\\textbf{警告：该PDF由GPT-Academic开源项目调用大语言模型+Latex翻译插件一键生成，" + \
"版权归原文作者所有。翻译内容可靠性无任何保障，请仔细鉴别并以原文为准。" + \
"项目Github地址 \\url{https://github.com/binary-husky/gpt_academic/}。"
self.msg_declare = "为了防止大语言模型的意外谬误产生扩散影响，禁止移除或修改此警告。}}\\\\"
def merge_result(self, arr, mode, msg): def merge_result(self, arr, mode, msg):
""" """
@ -206,18 +210,10 @@ class LatexPaperSplit():
node = node.next node = node.next
if node is None: break if node is None: break
if mode == 'translate_zh': if mode == 'translate_zh':
try: pattern = re.compile(r'\\begin\{abstract\}.*\n')
pattern = re.compile(r'\\begin\{abstract\}.*\n') match = pattern.search(result_string)
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] + \
"{\\scriptsize\\textbf{警告：该PDF由GPT-Academic开源项目调用大语言模型+Latex翻译插件一键生成，版权归原文作者所有。翻译内容可靠性无任何保障，请仔细鉴别并以原文为准。" + \
"项目Github地址 \\url{https://github.com/binary-husky/gpt_academic/}。" + \
msg + \
"为了防止大语言模型的意外谬误产生扩散影响，禁止移除或修改此警告。}}\\\\" + \
result_string[position:]
except:
pass
return result_string return result_string
def split(self, txt, project_folder): def split(self, txt, project_folder):
@ -487,7 +483,8 @@ def Latex精细分解与转化(file_manifest, project_folder, llm_kwargs, plugin
msg = f"当前大语言模型: {llm_kwargs['llm_model']}，当前语言模型温度设定: {llm_kwargs['temperature']}。" msg = f"当前大语言模型: {llm_kwargs['llm_model']}，当前语言模型温度设定: {llm_kwargs['temperature']}。"
final_tex = lps.merge_result(pfg.file_result, mode, msg) final_tex = lps.merge_result(pfg.file_result, mode, msg)
with open(project_folder + f'/merge_{mode}.tex', 'w', encoding='utf-8', errors='replace') as f: with open(project_folder + f'/merge_{mode}.tex', 'w', encoding='utf-8', errors='replace') as f:
f.write(final_tex) if mode != 'translate_zh' or "binary" in final_tex: f.write(final_tex)
# <-------- 整理结果, 退出 ----------> # <-------- 整理结果, 退出 ---------->
chatbot.append((f"完成了吗？", 'GPT结果已输出, 正在编译PDF')) chatbot.append((f"完成了吗？", 'GPT结果已输出, 正在编译PDF'))