132 lines
5.4 KiB
Python
132 lines
5.4 KiB
Python
from toolbox import CatchException, update_ui, gen_time_str
|
|
from .crazy_utils import request_gpt_model_in_new_thread_with_ui_alive
|
|
from .crazy_utils import input_clipping
|
|
|
|
|
|
prompt = """
|
|
I have to achieve some functionalities by calling one of the functions below.
|
|
Your job is to find the correct funtion to use to satisfy my requirement,
|
|
and then write python code to call this function with correct parameters.
|
|
|
|
These are functions you are allowed to choose from:
|
|
1.
|
|
功能描述: 总结音视频内容
|
|
调用函数: ConcludeAudioContent(txt, llm_kwargs)
|
|
参数说明:
|
|
txt: 音频文件的路径
|
|
llm_kwargs: 模型参数, 永远给定None
|
|
2.
|
|
功能描述: 将每次对话记录写入Markdown格式的文件中
|
|
调用函数: WriteMarkdown()
|
|
3.
|
|
功能描述: 将指定目录下的PDF文件从英文翻译成中文
|
|
调用函数: BatchTranslatePDFDocuments_MultiThreaded(txt, llm_kwargs)
|
|
参数说明:
|
|
txt: PDF文件所在的路径
|
|
llm_kwargs: 模型参数, 永远给定None
|
|
4.
|
|
功能描述: 根据文本使用GPT模型生成相应的图像
|
|
调用函数: ImageGeneration(txt, llm_kwargs)
|
|
参数说明:
|
|
txt: 图像生成所用到的提示文本
|
|
llm_kwargs: 模型参数, 永远给定None
|
|
5.
|
|
功能描述: 对输入的word文档进行摘要生成
|
|
调用函数: SummarizingWordDocuments(input_path, output_path)
|
|
参数说明:
|
|
input_path: 待处理的word文档路径
|
|
output_path: 摘要生成后的文档路径
|
|
|
|
|
|
You should always anwser with following format:
|
|
----------------
|
|
Code:
|
|
```
|
|
class AutoAcademic(object):
|
|
def __init__(self):
|
|
self.selected_function = "FILL_CORRECT_FUNCTION_HERE" # e.g., "GenerateImage"
|
|
self.txt = "FILL_MAIN_PARAMETER_HERE" # e.g., "荷叶上的蜻蜓"
|
|
self.llm_kwargs = None
|
|
```
|
|
Explanation:
|
|
只有GenerateImage和生成图像相关, 因此选择GenerateImage函数。
|
|
----------------
|
|
|
|
Now, this is my requirement:
|
|
|
|
"""
|
|
def get_fn_lib():
|
|
return {
|
|
"BatchTranslatePDFDocuments_MultiThreaded": ("crazy_functions.批量翻译PDF文档_多线程", "批量翻译PDF文档"),
|
|
"SummarizingWordDocuments": ("crazy_functions.总结word文档", "总结word文档"),
|
|
"ImageGeneration": ("crazy_functions.图片生成", "图片生成"),
|
|
"TranslateMarkdownFromEnglishToChinese": ("crazy_functions.批量Markdown翻译", "Markdown中译英"),
|
|
"SummaryAudioVideo": ("crazy_functions.总结音视频", "总结音视频"),
|
|
}
|
|
|
|
def inspect_dependency(chatbot, history):
|
|
return True
|
|
|
|
def eval_code(code, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, web_port):
|
|
import subprocess, sys, os, shutil, importlib
|
|
|
|
with open('gpt_log/void_terminal_runtime.py', 'w', encoding='utf8') as f:
|
|
f.write(code)
|
|
|
|
try:
|
|
AutoAcademic = getattr(importlib.import_module('gpt_log.void_terminal_runtime', 'AutoAcademic'), 'AutoAcademic')
|
|
# importlib.reload(AutoAcademic)
|
|
auto_dict = AutoAcademic()
|
|
selected_function = auto_dict.selected_function
|
|
txt = auto_dict.txt
|
|
fp, fn = get_fn_lib()[selected_function]
|
|
fn_plugin = getattr(importlib.import_module(fp, fn), fn)
|
|
yield from fn_plugin(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, web_port)
|
|
except:
|
|
from toolbox import trimmed_format_exc
|
|
chatbot.append(["执行错误", f"\n```\n{trimmed_format_exc()}\n```\n"])
|
|
yield from update_ui(chatbot=chatbot, history=history) # 刷新界面
|
|
|
|
def get_code_block(reply):
|
|
import re
|
|
pattern = r"```([\s\S]*?)```" # regex pattern to match code blocks
|
|
matches = re.findall(pattern, reply) # find all code blocks in text
|
|
if len(matches) != 1:
|
|
raise RuntimeError("GPT is not generating proper code.")
|
|
return matches[0].strip('python') # code block
|
|
|
|
@CatchException
|
|
def 终端(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, web_port):
|
|
"""
|
|
txt 输入栏用户输入的文本, 例如需要翻译的一段话, 再例如一个包含了待处理文件的路径
|
|
llm_kwargs gpt模型参数, 如温度和top_p等, 一般原样传递下去就行
|
|
plugin_kwargs 插件模型的参数, 暂时没有用武之地
|
|
chatbot 聊天显示框的句柄, 用于显示给用户
|
|
history 聊天历史, 前情提要
|
|
system_prompt 给gpt的静默提醒
|
|
web_port 当前软件运行的端口号
|
|
"""
|
|
# 清空历史, 以免输入溢出
|
|
history = []
|
|
|
|
# 基本信息：功能、贡献者
|
|
chatbot.append(["函数插件功能？", "根据自然语言执行插件命令, 作者: binary-husky, 插件初始化中 ..."])
|
|
yield from update_ui(chatbot=chatbot, history=history) # 刷新界面
|
|
|
|
# # 尝试导入依赖, 如果缺少依赖, 则给出安装建议
|
|
# dep_ok = yield from inspect_dependency(chatbot=chatbot, history=history) # 刷新界面
|
|
# if not dep_ok: return
|
|
|
|
# 输入
|
|
i_say = prompt + txt
|
|
# 开始
|
|
gpt_say = yield from request_gpt_model_in_new_thread_with_ui_alive(
|
|
inputs=i_say, inputs_show_user=txt,
|
|
llm_kwargs=llm_kwargs, chatbot=chatbot, history=[],
|
|
sys_prompt=""
|
|
)
|
|
|
|
# 将代码转为动画
|
|
code = get_code_block(gpt_say)
|
|
yield from eval_code(code, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, web_port)
|