Fix poe server crash

This commit is contained in:
SillyLossy
2023-04-04 12:19:18 +03:00
parent 67ac6a07a2
commit 3eb9fa975c
2 changed files with 40 additions and 20 deletions

View File

@@ -50,7 +50,7 @@ function loadSettings() {
selectBot();
const autoConnect = localStorage.getItem('AutoConnectEnabled') == "true";
if (autoConnect && token) {
if (autoConnect && token && token.length) {
onConnectClick();
}
}
@@ -69,12 +69,12 @@ function saveSettings() {
}
function onTokenInput() {
token = $(this).val();
token = $('#poe_token').val();
saveSettings();
}
function onBotChange() {
bot = $(this).find(":selected").val();
bot = $('#poe_bots').find(":selected").val();
saveSettings();
}
@@ -194,6 +194,10 @@ async function sendMessage(prompt) {
}
async function onConnectClick() {
if (!token || !token.length) {
return;
}
const body = JSON.stringify({ token: token });
const response = await fetch('/status_poe', {
headers: {

View File

@@ -1629,10 +1629,16 @@ app.post('/status_poe', jsonParser, async (request, response) => {
return response.sendStatus(400);
}
try {
const client = await getPoeClient(request.body.token);
const botNames = client.get_bot_names();
client.disconnect_ws();
return response.send({'bot_names': botNames});
}
catch {
return response.sendStatus(500);
}
});
app.post('/purge_poe', jsonParser, async (request, response) => {
@@ -1644,11 +1650,16 @@ app.post('/purge_poe', jsonParser, async (request, response) => {
const bot = request.body.bot ?? POE_DEFAULT_BOT;
const count = request.body.count ?? -1;
try {
const client = await getPoeClient(token);
await client.purge_conversation(bot, count)
client.disconnect_ws();
return response.send({"ok" : true});
}
catch {
return response.sendStatus(500);
}
});
app.post('/generate_poe', jsonParser, async (request, response) => {
@@ -1660,6 +1671,7 @@ app.post('/generate_poe', jsonParser, async (request, response) => {
const prompt = request.body.prompt;
const bot = request.body.bot ?? POE_DEFAULT_BOT;
try {
const client = await getPoeClient(token);
let reply;
@@ -1670,6 +1682,10 @@ app.post('/generate_poe', jsonParser, async (request, response) => {
client.disconnect_ws();
return response.send({'reply': reply});
}
catch {
return response.sendStatus(500);
}
});
function getThumbnailFolder(type) {