Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -91,6 +91,14 @@ blacklist_patterns = [
     r"\bdeath\s*threat\b",
 ]
 
+
+def is_command_disabled(text):
+  text_lower = text.lower()
+  if text_lower.startswith("/") and not (text_lower.startswith("/setmodel") or text_lower.startswith("/start")):
+    return True
+  else:
+    return False
+
 def is_blacklisted_english(text: str) -> bool:
     for pattern in blacklist_patterns:
         if re.search(pattern, text, re.IGNORECASE):
@@ -1331,6 +1339,10 @@ async def chatbot_talk(client: Client, message: Message):
                 await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
                 return
 
+            if is_command_disabled(query_base):
+                await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
+                return
+
             if regex_all_blacklist(query_base) or is_blacklisted_english(query_base):
                 unfreeze_at = dt.now() + timedelta(days=30)
                 await db.user_blacklists.update_one(