From 1aa68c14eac9a1006bca42d0736ffdb9ee9bfe4e Mon Sep 17 00:00:00 2001 From: w_xiaolizu Date: Thu, 18 May 2023 14:45:08 +0800 Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3=E5=88=B7=E6=96=B0dataset?= =?UTF-8?q?=E7=9A=84bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- func_box.py | 9 +++++---- prompt_generator.py | 6 ++---- 2 files changed, 7 insertions(+), 8 deletions(-) diff --git a/func_box.py b/func_box.py index a3a3edb..cd7f4de 100644 --- a/func_box.py +++ b/func_box.py @@ -55,6 +55,7 @@ class Shell(object): else: sysout = self.subp.stdout.read() syserr = self.subp.stderr.read() + self.subp.stdin if sysout: logger.debug(f"{self.args} \n{sysout}") return 1, sysout @@ -218,7 +219,7 @@ def json_convert_dict(): def draw_results(txt, prompt: gr.Dataset, percent, switch, ipaddr: gr.Request): data = diff_list(txt, percent=percent, switch=switch, hosts=ipaddr.client.host) prompt.samples = data - return prompt.update(samples=data, samples_per_page=10), prompt + return prompt.update(samples=data, samples_per_page=10, visible=True), prompt def diff_list(txt='', percent=0.70, switch: list = None, lst: list = None, sp=15, hosts=''): @@ -330,13 +331,13 @@ def prompt_save(txt, name, checkbox, prompt: gr.Dataset, ipaddr: gr.Request): if txt and name: yaml_obj = SqliteHandle(f'prompt_{ipaddr.client.host}') yaml_obj.inset_prompt({name: txt}) - result = prompt_retrieval(is_all=checkbox, hosts=ipaddr.client.host) + result = prompt_retrieval(is_all=checkbox+['个人'], hosts=ipaddr.client.host) prompt.samples = result return "", "", ['个人'], prompt.update(samples=result, samples_per_page=10, visible=True), prompt - if not txt or not name: + elif not txt or not name: result = [[f'{html_tag_color("编辑框 or 名称不能为空!!!!!", color="red")}', '']] prompt.samples = [[f'{html_tag_color("编辑框 or 名称不能为空!!!!!", color="red")}', '']] - return txt, name, checkbox, prompt.update(samples=result, samples_per_page=10, visible=True), prompt + return txt, name, [], prompt.update(samples=result, visible=True), prompt def prompt_input(txt, index, data: gr.Dataset): data_str = str(data.samples[index][1]) diff --git a/prompt_generator.py b/prompt_generator.py index f04dccd..a53f6e9 100644 --- a/prompt_generator.py +++ b/prompt_generator.py @@ -68,8 +68,6 @@ class SqliteHandle: sqlite_handle = SqliteHandle if __name__ == '__main__': + test = func_box.YamlHandle('/Users/kilig/Job/Python-project/academic_gpt/prompt_users/prompt_127.0.0.1.yaml').load() - - # print(sqlite_handle('ai_common').inset_prompt(test)) - # sqlite_handle('ai_common').delete_prompt() - print(sqlite_handle('ai_common').get_prompt_value()) + sqlite_handle('prompt_127.0.0.1').inset_prompt(test) \ No newline at end of file