10kilo/MeowNex
1
0
Fork 0
forked from swee/MeowNex
Code Pull requests Activity

Update sweebot.py

Browse source
This commit is contained in:
Nova Cat 2024-12-30 16:06:48 -08:00
parent 78445c2d1f
commit 159f884a21
1 changed files with 27 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
sweebot.py
View file

@ -15,6 +15,7 @@ from requests import get, exceptions as rex
from bs4 import BeautifulSoup
from googleapiclient.discovery import build
from characterai import pycai
import uuid
CAItoken = environ["CAItoken"]
CAImodel = environ["CAIchara"]
try:
@ -28,8 +29,9 @@ block = 0
parsed = 0
lastquery = "None yet."
# Dashboard thread
from flask import Flask, send_file, redirect as redir
from flask import Flask, send_file, abort as abrt, redirect as redir
app = Flask(__name__)
responses_ai = {} # {uuid: response}
@app.route('/')
def index_html():
@ -52,6 +54,13 @@ def lastquery_txt():
@app.route('/docs/config')
def config_docs():
return redir("https://git.swee.codes/swee/MeowNex/src/branch/main/docs/config.md")
@app.route('/response/<uuid>')
def ai_response_truncate(uuidd):
if uuidd in responses_ai:
res = responses_ai[uuidd]
return f'<title>AI response {uuidd}</title>\n<meta name="viewport" content="width=device-width, initial-scale=1.0">\n<link rel="stylesheet" href="https://swee.codes/style.css" type="text/css">\n<body><center><h1>AI response {uuidd}</h1><br><br>' + f"<textarea class=code width=100% spellcheck=\"false\" autocapitalize=\"off\" autocomplete=\"off\" autocorrect=\"off\" readonly=\"true\">{res}</textarea>"
else:
abrt(404)
threading.Thread(target=app.run, daemon=True, kwargs={"port": 2005}).start()
# YouTube API
@ -543,8 +552,23 @@ while True:
)
aimsg = aiconnector.send_message(CAImodel, new.chat_id, chatquery)
ai_storage[channel] = new.chat_id
multiline(aimsg.text, channel)
lastquery = f"<p>Channel: {channel}</p><p>Request:</p><br><br>{chatquery}<hr><p>Response</p><br><br>{aimsg.text}"
randomname = str(uuid.uuid4())
responses_ai[randomname] = aimsg.text
mtline = ""
lastquery = f"<p>Channel: {channel}</p><p>Request:</p><br><br><textarea class=code width=100% spellcheck=\"false\" autocapitalize=\"off\" autocomplete=\"off\" autocorrect=\"off\" readonly=\"true\">{chatquery}</textarea><hr><p>Response</p><br><br><textarea class=code width=100% spellcheck=\"false\" autocapitalize=\"off\" autocomplete=\"off\" autocorrect=\"off\" readonly=\"true\">{aimsg.text}</textarea>"
broke = False
for num, i in enumerate(aimsg.text.split("\n")):
if num == 5:
broke = True
break
elif len(i) > 75:
mtline += "\n" + i[:75] + "..."
broke = True
else:
mtline += "\n" + i
if broke:
mtline += f"\nView full response: https://irc-bot.swee.codes/response/{randomname}"
multiline(mtline, channel)
except:
multiline("Something went wrong while connecting to Character.AI.", channel)
print(traceback.format_exc())