Bladeren bron

lydia fix

TheHamkerCat 5 jaren geleden
bovenliggende
commit
a73e8cd628
1 gewijzigde bestanden met toevoegingen van 12 en 8 verwijderingen
  1. 12 8
      wbb/modules/lydia.py

+ 12 - 8
wbb/modules/lydia.py

@@ -10,15 +10,15 @@ __HELP__ = "Lydia AI Chatbot"
 
 CoffeeHouseAPI = API(str(LYDIA_API))
 api_ = LydiaAI(CoffeeHouseAPI)
-is_chat = False
+chats = []
 ses = None
 
 @app.on_message(filters.command('addchatbot'))
 async def add_chat(_, message):
-    global is_chat, api_, ses
-    if not is_chat:
+    global chats, api_, ses
+    if message.chat.id not in chats:
         ses = api_.create_session()
-        is_chat = True
+        chats.append(message.chat.id)
         await message.reply_text('ChatBot enabled!')
     else:
         await message.reply_text('ChatBot Is Already Enabled')
@@ -26,22 +26,26 @@ async def add_chat(_, message):
 
 @app.on_message(filters.command('rmchatbot'))
 async def remove_chat(_, message):
-    global is_chat
-    if not is_chat:
+    global chats
+    if message.chats.id not in chats:
         await message.reply_text("ChatBot Is Already Off")
     else:
+        chats.remove(message.chat.id)
         await message.reply_text('ChatBot Turned Off!')
 
 
 @app.on_message(~filters.edited & ~filters.private)
 async def chat_bot(_, message):
-    global api_, is_chat, ses
-    if not is_chat:
+    global api_, chats, ses
+    if message.chat.id not in chats:
         return
     if not message.reply_to_message:
         return
     if not message.reply_to_message.from_user.id == BOT_ID:
         return
+    if not message.text:
+        return
+    await app.send_chat_action(message.chat.id, "typing")
     query = message.text
     rep = api_.think_thought(text=query, session_id=ses.id)
     await message.reply_text(rep)