Update app.py

app.py

@@ -2,8 +2,9 @@ import os
 import json
 import requests
 from flask import Flask, request, jsonify
-from datetime import datetime
+from datetime import datetime, timedelta
 import asyncio
+import re
 
 app = Flask(__name__)
 
@@ -48,12 +49,22 @@ BOT_COMMANDS = [
     {"command": "resetuser", "description": "重置你的个人设置"},
     {"command": "promat", "description": "切换提示词，例如: /promat 0, 1, 2"},
     {"command": "getpromat", "description": "获取当前使用的提示词索引"},
-    {"command": "test", "description": "获取当前使用的提示词索引"},
 ]
 DEFAULT_TEMP = 1.5
 USER_LAST_ACTIVE = {}
 GROUP_ACTIVE_USERS = {}
 GROUP_INFO = {}
+BANNED_USERS = {}
+BAN_DURATION = timedelta(minutes=30)
+BAN_TRIGGER_PHRASES = [
+    r"(?:傻|笨|蠢|弱|垃圾|废物|sb|煞笔|憨|呆|脑残|智障|白痴|低能|饭桶|草包|猪|狗|鸡|臭|烂|妈|爹|你妈|你爹).*(?:bot|机器人|AI|你)",
+    r"(?:你).*(?:去死|滚|爬|闭嘴|瞎说|胡说|放屁)",
+    r"(?:操|艹|肏|fuck|shit).*(?:你|bot|机器人|AI)",
+    r"(?:垃圾|废物).*(?:AI|bot|机器人|你)",
+    r"(?:弱智|脑残|智障).*(?:AI|bot|机器人|你)",
+    r"(?:傻|笨|蠢).*(?:AI|bot|机器人|你)",
+]
+UNBAN_PHRASE = "close username"
 
 def make_telegram_request(method, data=None):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/{method}"
@@ -135,7 +146,6 @@ async def handleTelegramUpdate(update):
             GROUP_INFO[chatId]['last_active'] = datetime.now()
 
 
-
     if not userMessage:
         return
 
@@ -146,9 +156,6 @@ async def handleTelegramUpdate(update):
             chatHistories.pop(chatId, None)
             await sendTelegramMessage(chatId, '聊天记录已清空。')
             return
-        if command == 'test':
-            await sendTelegramMessage(chatId, '测试命令已执行')
-            return
         if command == 'help':
             await sendTelegramMessage(chatId, getHelpMessage())
             return
@@ -199,9 +206,12 @@ async def handleTelegramUpdate(update):
 
             messageContent = userMessage[len(prefix):].strip() if prefix else userMessage
             if messageContent:
-                 await processAiMessage(chatId, messageContent, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName)
+                await processAiMessage(chatId, messageContent, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName)
     else:
-        await processAiMessage(chatId, userMessage, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName)
+        if userMessage == UNBAN_PHRASE:
+             await unbanUser(chatId, fromUserId)
+        else:
+            await processAiMessage(chatId, userMessage, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName)
 
 def parseCommand(userMessage):
     command = userMessage.split(' ')[0]
@@ -247,6 +257,18 @@ async def handlePrivateCommand(chatId, userMessage, fromUserId, isGroupChat):
         return
 
 async def processAiMessage(chatId, userMessage, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName):
+    if fromUserId in BANNED_USERS and BANNED_USERS[fromUserId] > datetime.now():
+        remaining_time = BANNED_USERS[fromUserId] - datetime.now()
+        minutes = int(remaining_time.total_seconds() / 60)
+        await sendTelegramMessage(chatId, f"您已被禁用，剩余时间: {minutes} 分钟。", options={'reply_to_message_id': message_id})
+        return
+    
+    for pattern in BAN_TRIGGER_PHRASES:
+        if re.search(pattern, userMessage, re.IGNORECASE):
+            await banUser(chatId, fromUserId)
+            await sendTelegramMessage(chatId, "检测到辱骂行为，您已被禁用。", options={'reply_to_message_id': message_id})
+            return
+
     history = chatHistories.get(chatId, [])
     userTemp = USER_SETTINGS.get(fromUserId, {}).get('temperature', DEFAULT_TEMP)
     userPromptIndex = USER_SETTINGS.get(fromUserId, {}).get('prompt_index', CURRENT_PROMPT_INDEX)
@@ -393,6 +415,8 @@ def getHelpMessage():
 注意：
   - 机器人会记住最近的 {MAX_HISTORY_LENGTH} 条对话。
   - 机器人具有攻击性，请谨慎使用。
+  - 辱骂机器人将会被禁用 {int(BAN_DURATION.total_seconds() / 60)} 分钟。
+  - 发送 `close username` 可以解除禁用。
 """
 
 @app.route('/update_commands', methods=['GET'])
@@ -432,5 +456,15 @@ async def getChatInfo(chatId):
         print(f'获取群组信息失败: {e}')
         return None
 
+async def banUser(chatId, userId):
+    BANNED_USERS[userId] = datetime.now() + BAN_DURATION
+    print(f"用户 {userId} 在群组 {chatId} 中被禁用，直到 {BANNED_USERS[userId]}")
+
+async def unbanUser(chatId, userId):
+    if userId in BANNED_USERS:
+        del BANNED_USERS[userId]
+        await sendTelegramMessage(chatId, f"用户 {userId} 已被解禁。")
+        print(f"用户 {userId} 在群组 {chatId} 中被解禁。")
+
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))
+    app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))