Spaces:
Build error
Build error
import threading | |
from predict import predict_no_ui_long_connection | |
from toolbox import CatchException, write_results_to_file | |
def 全项目切换英文(txt, top_p, temperature, chatbot, history, sys_prompt, WEB_PORT): | |
history = [] # 清空历史,以免输入溢出 | |
# 集合文件 | |
import time, glob, os | |
file_manifest = [f for f in glob.glob('./**/*.py', recursive=True) if ('test_project' not in f) and ('gpt_log' not in f)] | |
i_say_show_user_buffer = [] | |
# 随便显示点什么防止卡顿 | |
for index, fp in enumerate(file_manifest): | |
# if 'test_project' in fp: continue | |
with open(fp, 'r', encoding='utf-8') as f: | |
file_content = f.read() | |
i_say_show_user =f'[{index}/{len(file_manifest)}] 接下来请将以下代码中包含的所有中文转化为英文,只输出代码: {os.path.abspath(fp)}' | |
i_say_show_user_buffer.append(i_say_show_user) | |
chatbot.append((i_say_show_user, "[Local Message] 等待多线程操作,中间过程不予显示.")) | |
yield chatbot, history, '正常' | |
# 任务函数 | |
mutable_return = [None for _ in file_manifest] | |
def thread_worker(fp,index): | |
with open(fp, 'r', encoding='utf-8') as f: | |
file_content = f.read() | |
i_say = f'接下来请将以下代码中包含的所有中文转化为英文,只输出代码,文件名是{fp},文件代码是 ```{file_content}```' | |
# ** gpt request ** | |
gpt_say = predict_no_ui_long_connection(inputs=i_say, top_p=top_p, temperature=temperature, history=history, sys_prompt=sys_prompt) | |
mutable_return[index] = gpt_say | |
# 所有线程同时开始执行任务函数 | |
handles = [threading.Thread(target=thread_worker, args=(fp,index)) for index, fp in enumerate(file_manifest)] | |
for h in handles: | |
h.daemon = True | |
h.start() | |
# 等待各个线程逐一完成 | |
for index, h in enumerate(handles): | |
h.join() | |
fp = file_manifest[index] | |
gpt_say = mutable_return[index] | |
i_say_show_user = i_say_show_user_buffer[index] | |
os.makedirs('gpt_log/generated_english_version', exist_ok=True) | |
os.makedirs('gpt_log/generated_english_version/crazy_functions', exist_ok=True) | |
where_to_relocate = f'gpt_log/generated_english_version/{fp}' | |
with open(where_to_relocate, 'w+', encoding='utf-8') as f: f.write(gpt_say.lstrip('```').rstrip('```')) | |
chatbot.append((i_say_show_user, f'[Local Message] 已完成{os.path.abspath(fp)}的转化,\n\n存入{os.path.abspath(where_to_relocate)}')) | |
history.append(i_say_show_user); history.append(gpt_say) | |
yield chatbot, history, '正常' | |
time.sleep(2) | |
# 结束 | |
res = write_results_to_file(history) | |
chatbot.append(("完成了吗?", res)) | |
yield chatbot, history, '正常' | |