zaclys/searxng
1
0
Fork 1
mirror of https://github.com/searxng/searxng synced 2024-01-01 19:24:07 +01:00
Code Issues Projects Releases Packages Wiki Activity

Update webapp.py

Browse source
This commit is contained in:
Joseph Cheung 2023-02-23 23:17:49 +08:00
parent 39e81d0b11
commit 21005b30b3
1 changed files with 82 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

83
searx/webapp.py
View file

@ -916,6 +916,87 @@ button.btn_more {
var word_last="";
var lock_chat=1;
function send_webchat(elem)
{
if(lock_chat!=0) return;
lock_chat = 1;
knowledge = document.querySelector("#chat").innerHTML.replace(/<a.*?>.*?<\/a.*?>/g, '').replace(/<hr.*/gs, '').replace(/<[^>]+>/g,"").replace(/\n\n/g,"\n")
if(knowledge.length>400)knowledge.slice(400)
knowledge += "\n以上是关键词 ''' + search_query.query + r''' 的搜索结果\n"
let word = document.querySelector("#chat_input").value;
if(elem){word = elem.textContent;elem.remove()}
if(word.length==0 || word.length > 140) return;
fetch('https://search.kg/search?q='+encodeURIComponent(word)+'&language=zh-CN&time_range=&safesearch=0&categories=general&format=json')
.then(response => response.json())
.then(data => {
prompt = JSON.parse(atob( (/<div id="prompt" style="display:none">(.*?)<\/div>/).exec(data.infoboxes[0].content)[1] ) )
prompt.data.max_tokens -= parseInt(knowledge.length*1.2)
prompt.data.prompt = knowledge + prompt.data.prompt
optionsweb = {
method: "POST",
headers: headers,
body: JSON.stringify(prompt.data)
};
document.querySelector("#prompt").innerHTML="";
markdownToHtml(beautify(word), document.querySelector("#prompt"))
chatTextRaw = "提问：" + word + "\n回答：";
chatTemp = ""
text_offset = -1;
prev_chat = document.getElementById('chat').innerHTML;
prev_chat = prev_chat+'<div class="chat_question">'+document.querySelector("#prompt").innerHTML+"</div>";
fetch("https://api.openai.com/v1/engines/text-davinci-003/completions", optionsweb)
.then((response) => {
const reader = response.body.getReader();
let result = '';
reader.read().then(function processText({ done, value }) {
if (done) return;
const text = new TextDecoder('utf-8').decode(value);
text.trim().split('\n').forEach(function(v) {
if(v.length>6) result = v.slice(6);
if(result == "[DONE]")
{
word_last += chatTextRaw + chatTemp
lock_chat=0
document.querySelector("#chat_input").value="";
return;
}
const { choices } = JSON.parse(result);
if(choices[0].logprobs.text_offset[0] > text_offset)
{
chatTemp+=choices[0].text
text_offset = choices[0].logprobs.text_offset[choices[0].logprobs.text_offset.length - 1]
}
chatTemp=chatTemp.replaceAll("\n\n","\n").replaceAll("\n\n","\n")
document.querySelector("#prompt").innerHTML="";
markdownToHtml(beautify(chatTemp), document.querySelector("#prompt"))
document.getElementById('chat').innerHTML = prev_chat+'<div class="chat_answer">'+document.querySelector("#prompt").innerHTML+"</div>";
})
return reader.read().then(processText);
});
})
.catch((error) => {
console.error('Error:', error);
});
})
}
function send_chat(elem)
{
if(lock_chat!=0) return;
@ -1134,7 +1215,7 @@ fetch("https://api.openai.com/v1/engines/text-davinci-003/completions", optionsI
.then(response => response.json())
.then(data => {
JSON.parse(data.choices[0].text.replaceAll("\n","")).forEach(item => {
document.querySelector("#chat_more").innerHTML += '<button class="btn_more" onclick="send_chat(this)">'+ String(item) +'</button>'
document.querySelector("#chat_more").innerHTML += '<button class="btn_more" onclick="send_webchat(this)">'+ String(item) +'</button>'
});
})
.catch(error => console.error(error));