CC/gpt_academic
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

pre

Browse Source
This commit is contained in:
binary-husky
2024-10-18 14:16:24 +00:00
parent 2b299cf579
commit a9c86a7fb8
3 changed files with 45 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
crazy_functions/Latex_Function.py
View File

@@ -364,6 +364,24 @@ def Latex翻译中文并重新编译PDF(txt, llm_kwargs, plugin_kwargs, chatbot,
chatbot=chatbot, history=history)
return
# allow_cloud_io = True
# arxiv_id = "2203.01927"
# if allow_cloud_io and arxiv_id:
# # 如果用户允许，我们将arxiv论文PDF上传到云端
# for file_path in chatbot._cookies.get("files_to_promote", []):
# if file_path.endswith('comparison.pdf'):
# def compute_hash(file_path):
# return map_file_to_sha256(file_path)
# with open(file_path, 'rb') as f:
# import requests
# url = 'https://cloud-2.agent-matrix.com/upload'
# files = {'file': (file_path, f, 'application/octet-stream')}
# data = {
# 'arxiv_id': arxiv_id,
# 'file_hash': compute_hash(file_path),
# }
# resp = requests.get(url=url, files=files, data=data, timeout=10)
if txt.endswith('.pdf'):
report_exception(chatbot, history, a=f"解析项目: {txt}", b=f"发现已经存在翻译好的PDF文档")
yield from update_ui(chatbot=chatbot, history=history) # 刷新界面
@@ -406,14 +424,34 @@ def Latex翻译中文并重新编译PDF(txt, llm_kwargs, plugin_kwargs, chatbot,
# <-------------- zip PDF ------------->
zip_res = zip_result(project_folder)
if success:
allow_cloud_io = True
arxiv_id = "2203.01927"
if allow_cloud_io and arxiv_id:
# 如果用户允许，我们将arxiv论文PDF上传到云端
for file_path in chatbot._cookies.get("files_to_promote", []):
if file_path.endswith('translate_zh.pdf') or file_path.endswith('comparison.pdf'):
def compute_hash(file_path):
return map_file_to_sha256(file_path)
with open(file_path, 'rb') as f:
import requests
url = 'https://cloud-2.agent-matrix.com/upload'
files = {'file': (file_path, f, 'application/octet-stream')}
data = {
'arxiv_id': arxiv_id,
'file_hash': compute_hash(file_path),
}
resp = requests.post(url=url, files=files, data=data, timeout=10)
chatbot.append((f"成功啦", '请查收结果（压缩包）...'))
yield from update_ui(chatbot=chatbot, history=history);
yield from update_ui(chatbot=chatbot, history=history)
time.sleep(1) # 刷新界面
promote_file_to_downloadzone(file=zip_res, chatbot=chatbot)
else:
chatbot.append((f"失败了",
'虽然PDF生成失败了, 但请查收结果（压缩包）, 内含已经翻译的Tex文档, 您可以到Github Issue区, 用该压缩包进行反馈。如系统是Linux，请检查系统字体（见Github wiki） ...'))
yield from update_ui(chatbot=chatbot, history=history);
yield from update_ui(chatbot=chatbot, history=history)
time.sleep(1) # 刷新界面
promote_file_to_downloadzone(file=zip_res, chatbot=chatbot)

3
crazy_functions/latex_fns/latex_actions.py
View File

@@ -423,6 +423,9 @@ def 编译Latex(chatbot, history, main_file_original, main_file_modified, work_f
except Exception as e:
logger.error(e)
pass
return True # 成功啦
else:
if n_fix>=max_try: break

3
tests/test_latex_auto_correct.py
View File

@@ -19,4 +19,5 @@ if __name__ == "__main__":
plugin_test = importlib.import_module('test_utils').plugin_test
plugin_test(plugin='crazy_functions.Latex_Function->Latex翻译中文并重新编译PDF', main_input="2203.01927")
# plugin_test(plugin='crazy_functions.Latex_Function->Latex翻译中文并重新编译PDF', main_input="2203.01927")
plugin_test(plugin='crazy_functions.Latex_Function->Latex翻译中文并重新编译PDF', main_input="gpt_log/arxiv_cache/2203.01927/workfolder")