Update app.py

app.py

@@ -147,40 +147,15 @@ async def handleTelegramUpdate(update):
                 prefix = GROUP_SETTINGS.get(chatId, {}).get('prefix', '无')
                 await sendTelegramMessage(chatId, f'当前群组触发前缀为: {prefix}')
                 return
-        else:
-            if userMessage.startswith('/settemp '):
-                try:
-                  temp = float(userMessage[len('/settemp '):].strip())
-                  if 0 <= temp <= 2:
-                    USER_SETTINGS.setdefault(fromUserId, {}).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 = USER_SETTINGS.get(fromUserId, {}).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:
-                     global CURRENT_PROMPT_INDEX
-                     CURRENT_PROMPT_INDEX = index
-                     await sendTelegramMessage(chatId, f'已切换到提示词 {index}。')
-                   else:
-                    await sendTelegramMessage(chatId, '提示词索引无效。请使用 /getpromat 查看可用的索引。')
-                except ValueError:
-                    await sendTelegramMessage(chatId, '提示词索引无效。请使用 /getpromat 查看可用的索引。')
-                return
-            if command == 'getpromat':
-                await sendTelegramMessage(chatId, f'当前使用的提示词索引是: {CURRENT_PROMPT_INDEX}')
+
+            # 在群组中处理私聊指令
+            if userMessage.startswith('/settemp ') or userMessage.startswith('/promat ') or command in ['gettemp', 'getpromat', 'resetuser']:
+                await handlePrivateCommand(chatId, userMessage, fromUserId, isGroupChat)
                 return
-            if command == 'resetuser':
-                USER_SETTINGS.pop(fromUserId, None)
-                await sendTelegramMessage(chatId, '已重置您的个人设置。')
+        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:
@@ -206,10 +181,49 @@ def parseCommand(userMessage):
         command = command.split('@')[0]
     return command[1:]
 
+
+async def handlePrivateCommand(chatId, userMessage, fromUserId, isGroupChat):
+    command = parseCommand(userMessage)
+    if userMessage.startswith('/settemp '):
+        try:
+            temp = float(userMessage[len('/settemp '):].strip())
+            if 0 <= temp <= 2:
+                USER_SETTINGS.setdefault(fromUserId, {}).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 = USER_SETTINGS.get(fromUserId, {}).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:
+                USER_SETTINGS.setdefault(fromUserId, {}).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 = USER_SETTINGS.get(fromUserId, {}).get('prompt_index', CURRENT_PROMPT_INDEX)
+        await sendTelegramMessage(chatId, f'当前使用的提示词索引是: {index}')
+        return
+    if command == 'resetuser':
+        USER_SETTINGS.pop(fromUserId, None)
+        await sendTelegramMessage(chatId, '已重置您的个人设置。')
+        return
+
 async def processAiMessage(chatId, userMessage, fromUserId):
     history = chatHistories.get(chatId, [])
     userTemp = USER_SETTINGS.get(fromUserId, {}).get('temperature', DEFAULT_TEMP)
-    currentPrompt = PROMPT_TEMPLATES.get(CURRENT_PROMPT_INDEX, "")
+    userPromptIndex = USER_SETTINGS.get(fromUserId, {}).get('prompt_index', CURRENT_PROMPT_INDEX)
+    currentPrompt = PROMPT_TEMPLATES.get(userPromptIndex, "")
     history.append({'role': 'user', 'content': userMessage})
 
     if len(history) > MAX_HISTORY_LENGTH:
@@ -292,7 +306,7 @@ def getHelpMessage():
 /setprefix <prefix> - 设置群组中触发AI回复的前缀，例如：/setprefix @bot。
 /getprefix - 获取当前群组的触发前缀。
     
-私聊指令:
+私聊指令 (在群组中也可以使用)：
 /settemp <温度值> - 设置AI回复的温度 (0-2)，例如：/settemp 1.0。
 /gettemp - 获取当前AI回复的温度。
 /resetuser - 重置你的个人设置。