Update app.py

app.py

@@ -270,15 +270,17 @@ async def handleTelegramUpdate(update):
                 return
 
             if userMessage.startswith('/settemp ') or userMessage.startswith('/promat ') or command in ['gettemp', 'getpromat', 'resetuser']:
-                await handlePrivateCommand(chatId, userMessage, fromUserId, isGroupChat)
+                # 群组中，私聊指令只用于设置群组的
+                await handleGroupCommand(chatId, userMessage, fromUserId, isGroupChat)
                 return
         else:
             if userMessage.startswith('/settemp ') or userMessage.startswith('/promat ') or command in ['gettemp', 'getpromat', 'resetuser']:
+                 # 私聊时，使用用户的设置
                 await handlePrivateCommand(chatId, userMessage, fromUserId, isGroupChat)
                 return
     if isGroupChat:
         if chatId not in GROUP_SETTINGS:
-            GROUP_SETTINGS[chatId] = {'aiEnabled': True, 'prefix': None}
+            GROUP_SETTINGS[chatId] = {'aiEnabled': True, 'prefix': None, 'temperature': DEFAULT_TEMP, 'prompt_index': CURRENT_PROMPT_INDEX} # 初始化群组设置
             print(f'群组 {chatId} 首次检测到，默认启用 AI。')
 
         groupSettings = GROUP_SETTINGS[chatId]
@@ -290,12 +292,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, isGroupChat) # 传递 isGroupChat
     else:
         if userMessage == UNBAN_PHRASE:
              await unbanUser(chatId, fromUserId)
         else:
-            await processAiMessage(chatId, userMessage, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName)
+            await processAiMessage(chatId, userMessage, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName, isGroupChat) # 传递 isGroupChat
 
 def parseCommand(userMessage):
     command = userMessage.split(' ')[0]
@@ -340,7 +342,44 @@ async def handlePrivateCommand(chatId, userMessage, fromUserId, isGroupChat):
         await sendTelegramMessage(chatId, '已重置您的个人设置。')
         return
 
-async def processAiMessage(chatId, userMessage, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName):
+async def handleGroupCommand(chatId, userMessage, fromUserId, isGroupChat):
+    command = parseCommand(userMessage)
+    if userMessage.startswith('/settemp '):
+        try:
+            temp = float(userMessage[len('/settemp '):].strip())
+            if 0 <= temp <= 2:
+                GROUP_SETTINGS.setdefault(chatId, {}).update({'temperature': temp})
+                await sendTelegramMessage(chatId, f'已设置群组AI回复温度为: {temp}')
+            else:
+                await sendTelegramMessage(chatId, '温度设置无效，请输入0到2之间的数字。')
+        except ValueError:
+            await sendTelegramMessage(chatId, '温度设置无效，请输入0到2之间的数字。')
+        return
+    if command == 'gettemp':
+        temp = GROUP_SETTINGS.get(chatId, {}).get('temperature', DEFAULT_TEMP)
+        await sendTelegramMessage(chatId, f'当前群组AI回复温度为: {temp}')
+        return
+    if userMessage.startswith('/promat '):
+        try:
+            index = int(userMessage[len('/promat '):].strip())
+            if index in PROMPT_TEMPLATES:
+                GROUP_SETTINGS.setdefault(chatId, {}).update({'prompt_index': index})
+                await sendTelegramMessage(chatId, f'已切换到群组提示词 {index}。')
+            else:
+                await sendTelegramMessage(chatId, '提示词索引无效。请使用 /getpromat 查看可用的索引。')
+        except ValueError:
+            await sendTelegramMessage(chatId, '提示词索引无效。请使用 /getpromat 查看可用的索引。')
+        return
+    if command == 'getpromat':
+        index = GROUP_SETTINGS.get(chatId, {}).get('prompt_index', CURRENT_PROMPT_INDEX)
+        await sendTelegramMessage(chatId, f'当前群组使用的提示词索引是: {index}')
+        return
+    if command == 'resetuser':
+       # 群组中不应该重置用户的私聊设置
+        await sendTelegramMessage(chatId, '群组设置中，该指令无效。')
+        return
+
+async def processAiMessage(chatId, userMessage, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName, isGroupChat):
     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)
@@ -354,8 +393,16 @@ async def processAiMessage(chatId, userMessage, fromUserId, message_id, fromUser
             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)
+    
+    if isGroupChat:
+      # 从群组设置获取温度和提示词
+      userTemp = GROUP_SETTINGS.get(chatId, {}).get('temperature', DEFAULT_TEMP)
+      userPromptIndex = GROUP_SETTINGS.get(chatId, {}).get('prompt_index', CURRENT_PROMPT_INDEX)
+    else:
+       # 从用户设置获取温度和提示词
+      userTemp = USER_SETTINGS.get(fromUserId, {}).get('temperature', DEFAULT_TEMP)
+      userPromptIndex = USER_SETTINGS.get(fromUserId, {}).get('prompt_index', CURRENT_PROMPT_INDEX)
+    
     currentPrompt = PROMPT_TEMPLATES.get(userPromptIndex, "")
 
     user_last_active = USER_LAST_ACTIVE.get(fromUserId, None)
@@ -482,8 +529,12 @@ def getHelpMessage():
 /disableai - 在群组中禁用AI回复。
 /setprefix <prefix> - 设置群组中触发AI回复的前缀，例如：/setprefix @bot。
 /getprefix - 获取当前群组的触发前缀。
+/settemp <温度值> - 设置群组AI回复的温度 (0-2)，例如：/settemp 1.0。
+/gettemp - 获取当前群组AI回复的温度。
+/promat <index> - 切换群组提示词，例如: /promat 0, 1, 2。
+/getpromat - 获取当前群组使用的提示词索引。
 
-私聊指令 (在群组中也可以使用)：
+私聊指令 (在群组中也可以使用，但是设置的是私聊的)：
 /settemp <温度值> - 设置AI回复的温度 (0-2)，例如：/settemp 1.0。
 /gettemp - 获取当前AI回复的温度。
 /resetuser - 重置你的个人设置。