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

markdown translation handle github index page

Browse Source
This commit is contained in:
qingxu fu 2023-07-28 20:20:30 +08:00
parent f446dbb62d
commit 2a362cec84
1 changed files with 17 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
crazy_functions/批量Markdown翻译.py
View File

@ -109,21 +109,30 @@ def 多文件翻译(file_manifest, project_folder, llm_kwargs, plugin_kwargs, ch
yield from update_ui(chatbot=chatbot, history=history) # 刷新界面 yield from update_ui(chatbot=chatbot, history=history) # 刷新界面
def get_files_from_everything(txt): def get_files_from_everything(txt, preference=''):
import glob, os import glob, os
success = True success = True
if txt.startswith('http'): if txt.startswith('http'):
# 网络的远程文件
txt = txt.replace("https://github.com/", "https://raw.githubusercontent.com/")
txt = txt.replace("/blob/", "/")
import requests import requests
from toolbox import get_conf from toolbox import get_conf
proxies, = get_conf('proxies') proxies, = get_conf('proxies')
# 网络的远程文件
if preference == 'Github':
print('正在从github下载资源 ...')
if not txt.endswith('.md'):
# Make a request to the GitHub API to retrieve the repository information
url = txt.replace("https://github.com/", "https://api.github.com/repos/") + '/readme'
response = requests.get(url, proxies=proxies)
txt = response.json()['download_url']
else:
txt = txt.replace("https://github.com/", "https://raw.githubusercontent.com/")
txt = txt.replace("/blob/", "/")
r = requests.get(txt, proxies=proxies) r = requests.get(txt, proxies=proxies)
with open('./gpt_log/temp.md', 'wb+') as f: f.write(r.content) download_local = f'./gpt_log/temp_{gen_time_str()}.md'
with open(download_local, 'wb+') as f: f.write(r.content)
project_folder = './gpt_log/' project_folder = './gpt_log/'
file_manifest = ['./gpt_log/temp.md'] file_manifest = [download_local]
elif txt.endswith('.md'): elif txt.endswith('.md'):
# 直接给定文件 # 直接给定文件
file_manifest = [txt] file_manifest = [txt]
@ -158,7 +167,7 @@ def Markdown英译中(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_p
return return
history = [] # 清空历史，以免输入溢出 history = [] # 清空历史，以免输入溢出
success, file_manifest, project_folder = get_files_from_everything(txt) success, file_manifest, project_folder = get_files_from_everything(txt, preference="Github")
if not success: if not success:
# 什么都没有 # 什么都没有