Update webapp.py

This commit is contained in:
Joseph Cheung 2023-02-19 21:09:40 +08:00
parent 91ee8a3d71
commit 5471cf8f0f

View file

@ -712,7 +712,6 @@ def search():
tmp_prompt = new_url +'\n'+ res['title'] +'\n'+ res['content']
if len(prompt)+len(tmp_prompt)<=1500:
prompt += tmp_prompt +'\n'
print("prompt:"+prompt+"\n以上是搜索引擎对关键词 " + search_query.query + " 的结果用简体中文分条总结简报在文中用markdown脚注指向内容来源链接")
if prompt != "":
gpt = ""
gpt_url = "https://api.openai.com/v1/engines/text-davinci-003/completions"
@ -733,16 +732,15 @@ def search():
"logprobs": 0,
"stream": False
}
print(gpt_response)
gpt_response = requests.post(gpt_url, headers=gpt_headers, data=json.dumps(gpt_data))
gpt_json = gpt_response.json()
if 'choices' in gpt_json:
gpt = gpt_json['choices'][0]['text']
for urls in url_pair.keys():
gpt.replace(urls,url_pair[urls])
gpt = gpt.replace(urls,url_pair[urls])
if gpt and gpt!="":
gptbox = {
'infobox': 'New Search',
'infobox': 'GPT3',
'id': 'gpt'+str(len(prompt)),
'content': gpt,
}