def chatbot(bot: Bot, update: Update):
global api_client
msg = update.effective_message
chat_id = update.effective_chat.id
is_chat = sql.is_chat(chat_id)
if not is_chat:
return
if msg.text and not msg.document:
if not check_message(bot, msg):
return
sesh, exp = sql.get_ses(chat_id)
query = msg.text
try:
if int(exp) < time():
ses = api_client.create_session()
ses_id = str(ses.id)
expires = str(ses.expires)
sql.set_ses(chat_id, ses_id, expires)
sesh, exp = sql.get_ses(chat_id)
except ValueError:
pass
try:
bot.send_chat_action(chat_id, action='typing')
rep = api_client.think_thought(sesh, query)
sleep(0.3)
msg.reply_text(rep, timeout=60)
except CFError as e:
bot.send_message(OWNER_ID,
f"Chatbot error: {e} occurred in {chat_id}!")
def remove_chat(_bot: Bot, update: Update):
msg = update.effective_message
chat_id = update.effective_chat.id
is_chat = sql.is_chat(chat_id)
if not is_chat:
msg.reply_text("AI isn't enabled here in the first place!")
else:
sql.rem_chat(chat_id)
msg.reply_text("AI disabled successfully!")
def add_chat(_bot: Bot, update: Update):
global api_client
chat_id = update.effective_chat.id
msg = update.effective_message
is_chat = sql.is_chat(chat_id)
if not is_chat:
ses = api_client.create_session()
ses_id = str(ses.id)
expires = str(ses.expires)
sql.set_ses(chat_id, ses_id, expires)
msg.reply_text("AI successfully enabled for this chat!")
else:
msg.reply_text("AI is already enabled for this chat!")
