Update app.py

app.py

@@ -51,6 +51,10 @@ BOT_COMMANDS = [
     {"command": "test", "description": "获取当前使用的提示词索引"},
 ]
 DEFAULT_TEMP = 1.5
+# 新增的用户和群组信息存储
+USER_LAST_ACTIVE = {}
+GROUP_ACTIVE_USERS = {}
+GROUP_INFO = {}
 
 def make_telegram_request(method, data=None):
     url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/{method}"
@@ -106,6 +110,34 @@ async def handleTelegramUpdate(update):
     isGroupChat = update['message']['chat']['type'] in ['group', 'supergroup']
     fromUserId = update['message']['from']['id']
     message_id = update['message'].get('message_id')
+    fromUserFirstName = update['message']['from'].get('first_name', '用户')
+    fromUserLastName = update['message']['from'].get('last_name', '')
+    fromUserName = update['message']['from'].get('username', '')
+    
+    # 更新用户最后活跃时间
+    USER_LAST_ACTIVE[fromUserId] = datetime.now()
+
+    if isGroupChat:
+         # 更新群组活跃用户
+        GROUP_ACTIVE_USERS.setdefault(chatId, set()).add(fromUserId)
+        if chatId not in GROUP_INFO:
+             chat_info = await getChatInfo(chatId)
+             if chat_info:
+                  GROUP_INFO[chatId] = {
+                      'name': chat_info.get('title', '未知群组'),
+                      'description': chat_info.get('description', '无描述'),
+                      'last_active': datetime.now()
+                  }
+             else:
+                  GROUP_INFO[chatId] = {
+                      'name': '未知群组',
+                      'description': '无描述',
+                      'last_active': datetime.now()
+                   }
+        else:
+            GROUP_INFO[chatId]['last_active'] = datetime.now()
+
+
 
     if not userMessage:
         return
@@ -172,9 +204,9 @@ async def handleTelegramUpdate(update):
 
             messageContent = userMessage[len(prefix):].strip() if prefix else userMessage
             if messageContent:
-                 await processAiMessage(chatId, messageContent, fromUserId, message_id)
+                 await processAiMessage(chatId, messageContent, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName)
     else:
-        await processAiMessage(chatId, userMessage, fromUserId, message_id)
+        await processAiMessage(chatId, userMessage, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName)
 
 def parseCommand(userMessage):
     command = userMessage.split(' ')[0]
@@ -220,21 +252,45 @@ async def handlePrivateCommand(chatId, userMessage, fromUserId, isGroupChat):
         await sendTelegramMessage(chatId, '已重置您的个人设置。')
         return
 
-async def processAiMessage(chatId, userMessage, fromUserId, message_id):
+async def processAiMessage(chatId, userMessage, fromUserId, message_id, fromUserFirstName, fromUserLastName, fromUserName):
     history = chatHistories.get(chatId, [])
     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)
+    group_info = GROUP_INFO.get(chatId, None)
+    group_active_users = GROUP_ACTIVE_USERS.get(chatId, None)
+
+    # 构建更丰富的系统提示
+    system_prompt = f"""
+{currentPrompt}
+当前时间: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
+
+用户信息:
+  - 用户ID: {fromUserId}
+  - 用户名: {fromUserFirstName} {fromUserLastName}
+  - 用户账号: {fromUserName}
+  - 最后活跃时间: {user_last_active.strftime("%Y-%m-%d %H:%M:%S") if user_last_active else "未知"}
+
+群组信息:
+  - 群组ID: {chatId}
+  - 群组名称: {group_info['name'] if group_info else "未知"}
+  - 群组描述: {group_info['description'] if group_info else "无"}
+  - 群组最后活跃时间: {group_info['last_active'].strftime("%Y-%m-%d %H:%M:%S") if group_info and group_info.get('last_active') else "未知"}
+  - 群组活跃用户数: {len(group_active_users) if group_active_users else "未知"}
+"""
     history.append({'role': 'user', 'content': userMessage})
 
     if len(history) > MAX_HISTORY_LENGTH:
         history = history[-MAX_HISTORY_LENGTH:]
 
     messages = [
-        {'role': 'system', 'content': currentPrompt},
+        {'role': 'system', 'content': system_prompt},
         *history
     ]
-    
+  
     thinking_message = await sendTelegramMessage(chatId, "正在思考，请等待...", options={'reply_to_message_id': message_id})
     thinking_message_id = thinking_message.get('result', {}).get('message_id')
 
@@ -274,7 +330,7 @@ async def handleCallbackQuery(callbackQuery):
     if data == "clearall":
         chatHistories.pop(chatId, None)
         await sendTelegramMessage(chatId, "聊天记录已清空。")
-    
+  
     await sendTelegramMessage(chatId, '请选择操作：', {
         'reply_markup': {
           'inline_keyboard': [[{'text': "清空���天记录", 'callback_data': "clearall"}]],
@@ -286,13 +342,13 @@ async def sendTelegramMessage(chatId, text, options={}):
     if PHP_PROXY_URL:
         method_name = url.split('/')[-1]
         url = f"{PHP_PROXY_URL}{method_name}"
-    
+  
     data = {
         'chat_id': chatId,
         'text': text,
         **options
     }
-    
+  
     try:
         response = requests.post(url, headers={'Content-Type': 'application/json'}, json=data)
         response.raise_for_status()
@@ -330,18 +386,18 @@ def getHelpMessage():
 /disableai - 在群组中禁用AI回复。
 /setprefix <prefix> - 设置群组中触发AI回复的前缀，例如：/setprefix @bot。
 /getprefix - 获取当前群组的触发前缀。
-    
+  
 私聊指令 (在群组中也可以使用)：
 /settemp <温度值> - 设置AI回复的温度 (0-2)，例如：/settemp 1.0。
 /gettemp - 获取当前AI回复的温度。
 /resetuser - 重置你的个人设置。
 /promat <index> - 切换提示词，例如: /promat 0, 1, 2。
 /getpromat - 获取当前使用的提示词索引。
-    
+  
 直接发送文本消息与AI对话 (私聊)。
 
 群组中，需要使用前缀触发AI回复，如果设置了前缀的话。
-    
+  
 注意：
   - 机器人会记住最近的 {MAX_HISTORY_LENGTH} 条对话。
   - 机器人具有攻击性，请谨慎使用。
@@ -368,5 +424,21 @@ async def handle_webhook():
 def health_check():
     return 'OK'
 
+async def getChatInfo(chatId):
+    url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/getChat"
+    if PHP_PROXY_URL:
+        method_name = url.split('/')[-1]
+        url = f"{PHP_PROXY_URL}{method_name}"
+    data = {
+        'chat_id': chatId,
+    }
+    try:
+        response = requests.post(url, headers={'Content-Type': 'application/json'}, json=data)
+        response.raise_for_status()
+        return response.json().get('result', {})
+    except requests.exceptions.RequestException as e:
+        print(f'获取群组信息失败: {e}')
+        return None
+
 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)))