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

解决刷新dataset的bug

Browse Source
This commit is contained in:
w_xiaolizu
2023-05-18 14:45:08 +08:00
parent 1e28a4feea
commit 1aa68c14ea
2 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
func_box.py
View File

@ -55,6 +55,7 @@ class Shell(object):
else: else:
sysout = self.subp.stdout.read() sysout = self.subp.stdout.read()
syserr = self.subp.stderr.read() syserr = self.subp.stderr.read()
self.subp.stdin
if sysout: if sysout:
logger.debug(f"{self.args} \n{sysout}") logger.debug(f"{self.args} \n{sysout}")
return 1, sysout return 1, sysout
@ -218,7 +219,7 @@ def json_convert_dict():
def draw_results(txt, prompt: gr.Dataset, percent, switch, ipaddr: gr.Request): def draw_results(txt, prompt: gr.Dataset, percent, switch, ipaddr: gr.Request):
data = diff_list(txt, percent=percent, switch=switch, hosts=ipaddr.client.host) data = diff_list(txt, percent=percent, switch=switch, hosts=ipaddr.client.host)
prompt.samples = data 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=''): 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: if txt and name:
yaml_obj = SqliteHandle(f'prompt_{ipaddr.client.host}') yaml_obj = SqliteHandle(f'prompt_{ipaddr.client.host}')
yaml_obj.inset_prompt({name: txt}) 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 prompt.samples = result
return "", "", ['个人'], prompt.update(samples=result, samples_per_page=10, visible=True), prompt 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")}', '']] result = [[f'{html_tag_color("编辑框 or 名称不能为空!!!!!", color="red")}', '']]
prompt.samples = [[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): def prompt_input(txt, index, data: gr.Dataset):
data_str = str(data.samples[index][1]) data_str = str(data.samples[index][1])

6
prompt_generator.py
View File

@ -68,8 +68,6 @@ class SqliteHandle:
sqlite_handle = SqliteHandle sqlite_handle = SqliteHandle
if __name__ == '__main__': if __name__ == '__main__':
test = func_box.YamlHandle('/Users/kilig/Job/Python-project/academic_gpt/prompt_users/prompt_127.0.0.1.yaml').load()
sqlite_handle('prompt_127.0.0.1').inset_prompt(test)
# print(sqlite_handle('ai_common').inset_prompt(test))
# sqlite_handle('ai_common').delete_prompt()
print(sqlite_handle('ai_common').get_prompt_value())