115 lines
4.7 KiB
Python
115 lines
4.7 KiB
Python
# """
|
||
# 'primary' for main call-to-action,
|
||
# 'secondary' for a more subdued style,
|
||
# 'stop' for a stop button.
|
||
# """
|
||
def 自我程序解构简单案例(txt, top_p, temperature, chatbot, history, systemPromptTxt, WEB_PORT):
|
||
import time
|
||
from predict import predict_no_ui_no_history
|
||
for i in range(5):
|
||
i_say = f'我给出一个数字，你给出该数字的平方。我给出数字：{i}'
|
||
gpt_say = predict_no_ui_no_history(inputs=i_say, top_p=top_p, temperature=temperature)
|
||
chatbot.append((i_say, gpt_say))
|
||
history.append(i_say)
|
||
history.append(gpt_say)
|
||
yield chatbot, history, '正常'
|
||
time.sleep(10)
|
||
|
||
def 解析项目本身(txt, top_p, temperature, chatbot, history, systemPromptTxt, WEB_PORT):
|
||
import time, glob, os
|
||
from predict import predict_no_ui
|
||
file_manifest = [f for f in glob.glob('*.py')]
|
||
|
||
for index, fp in enumerate(file_manifest):
|
||
with open(fp, 'r', encoding='utf-8') as f:
|
||
file_content = f.read()
|
||
|
||
前言 = "接下来请你分析自己的程序构成，别紧张，" if index==0 else ""
|
||
i_say = f'请对下面的程序文件做一个概述: ```{file_content}```'
|
||
i_say_show_user = 前言 + f'请对下面的程序文件做一个概述: {os.path.abspath(fp)}'
|
||
chatbot.append((i_say_show_user, "[waiting gpt response]"))
|
||
yield chatbot, history, '正常'
|
||
|
||
# ** gpt request **
|
||
gpt_say = predict_no_ui(inputs=i_say, top_p=top_p, temperature=temperature)
|
||
|
||
chatbot[-1] = (i_say_show_user, gpt_say)
|
||
history.append(i_say_show_user); history.append(gpt_say)
|
||
yield chatbot, history, '正常'
|
||
time.sleep(2)
|
||
|
||
i_say = f'根据以上你自己的分析，对程序的整体功能和构架做出概括。然后用一张markdown表格整理每个文件的功能（包括{file_manifest}）。'
|
||
chatbot.append((i_say, "[waiting gpt response]"))
|
||
yield chatbot, history, '正常'
|
||
|
||
# ** gpt request **
|
||
gpt_say = predict_no_ui(inputs=i_say, top_p=top_p, temperature=temperature, history=history)
|
||
|
||
chatbot[-1] = (i_say, gpt_say)
|
||
history.append(i_say); history.append(gpt_say)
|
||
yield chatbot, history, '正常'
|
||
|
||
|
||
def 解析一个Python项目(txt, top_p, temperature, chatbot, history, systemPromptTxt, WEB_PORT):
|
||
import time, glob, os
|
||
from predict import predict_no_ui
|
||
if os.path.exists(txt):
|
||
project_folder = txt
|
||
else:
|
||
if txt == "": txt = '空空如也的输入栏'
|
||
chatbot.append((f"解析项目: {txt}", f"找不到本地项目: {txt}"))
|
||
history.append(f"解析项目: {txt}"); history.append(f"找不到本地项目: {txt}")
|
||
yield chatbot, history, '正常'
|
||
return
|
||
|
||
file_manifest = [f for f in glob.glob(f'{project_folder}/*.py')]
|
||
print('begin analysis on:', file_manifest)
|
||
for index, fp in enumerate(file_manifest):
|
||
with open(fp, 'r', encoding='utf-8') as f:
|
||
file_content = f.read()
|
||
|
||
前言 = "接下来请你逐文件分析下面的Python工程" if index==0 else ""
|
||
i_say = f'请对下面的程序文件做一个概述: ```{file_content}```'
|
||
i_say_show_user = 前言 + f'[{index}/{len(file_manifest)}] 请对下面的程序文件做一个概述: {os.path.abspath(fp)}'
|
||
chatbot.append((i_say_show_user, "[waiting gpt response]"))
|
||
print('[1] yield chatbot, history')
|
||
yield chatbot, history, '正常'
|
||
|
||
# ** gpt request **
|
||
gpt_say = predict_no_ui(inputs=i_say, top_p=top_p, temperature=temperature)
|
||
|
||
print('[2] end gpt req')
|
||
chatbot[-1] = (i_say_show_user, gpt_say)
|
||
history.append(i_say_show_user); history.append(gpt_say)
|
||
print('[3] yield chatbot, history')
|
||
yield chatbot, history, '正常'
|
||
print('[4] next')
|
||
time.sleep(2)
|
||
|
||
i_say = f'根据以上你自己的分析，对程序的整体功能和构架做出概括。然后用一张markdown表格整理每个文件的功能（包括{file_manifest}）。'
|
||
chatbot.append((i_say, "[waiting gpt response]"))
|
||
yield chatbot, history, '正常'
|
||
|
||
# ** gpt request **
|
||
gpt_say = predict_no_ui(inputs=i_say, top_p=top_p, temperature=temperature, history=history)
|
||
|
||
chatbot[-1] = (i_say, gpt_say)
|
||
history.append(i_say); history.append(gpt_say)
|
||
yield chatbot, history, '正常'
|
||
|
||
|
||
def get_crazy_functionals():
|
||
return {
|
||
"程序解构简单案例": {
|
||
"Function": 自我程序解构简单案例
|
||
},
|
||
"请解析并解构此项目本身": {
|
||
"Function": 解析项目本身
|
||
},
|
||
"解析一整个Python项目（输入栏给定项目完整目录）": {
|
||
"Function": 解析一个Python项目
|
||
},
|
||
}
|
||
|
||
|