Update app.py

app.py

@@ -105,6 +105,7 @@ async def handleTelegramUpdate(update):
     userMessage = update['message'].get('text', '')
     isGroupChat = update['message']['chat']['type'] in ['group', 'supergroup']
     fromUserId = update['message']['from']['id']
+    message_id = update['message'].get('message_id')
 
     if not userMessage:
         return
@@ -171,9 +172,9 @@ async def handleTelegramUpdate(update):
 
             messageContent = userMessage[len(prefix):].strip() if prefix else userMessage
             if messageContent:
-                 await processAiMessage(chatId, messageContent, fromUserId)
+                 await processAiMessage(chatId, messageContent, fromUserId, message_id)
     else:
-        await processAiMessage(chatId, userMessage, fromUserId)
+        await processAiMessage(chatId, userMessage, fromUserId, message_id)
 
 def parseCommand(userMessage):
     command = userMessage.split(' ')[0]
@@ -219,7 +220,7 @@ async def handlePrivateCommand(chatId, userMessage, fromUserId, isGroupChat):
         await sendTelegramMessage(chatId, '已重置您的个人设置。')
         return
 
-async def processAiMessage(chatId, userMessage, fromUserId):
+async def processAiMessage(chatId, userMessage, fromUserId, message_id):
     history = chatHistories.get(chatId, [])
     userTemp = USER_SETTINGS.get(fromUserId, {}).get('temperature', DEFAULT_TEMP)
     userPromptIndex = USER_SETTINGS.get(fromUserId, {}).get('prompt_index', CURRENT_PROMPT_INDEX)
@@ -233,6 +234,9 @@ async def processAiMessage(chatId, userMessage, fromUserId):
         {'role': 'system', 'content': currentPrompt},
         *history
     ]
+    
+    thinking_message = await sendTelegramMessage(chatId, "正在思考，请等待...", reply_to_message_id=message_id)
+    thinking_message_id = thinking_message.get('result', {}).get('message_id')
 
     try:
         ai_response = requests.post(AI_API_ENDPOINT, headers=AI_API_HEADERS, json={
@@ -248,13 +252,14 @@ async def processAiMessage(chatId, userMessage, fromUserId):
         history.append({'role': 'assistant', 'content': ai_reply})
         chatHistories[chatId] = history
 
-        await sendTelegramMessage(chatId, ai_reply)
+        await editTelegramMessage(chatId, thinking_message_id, ai_reply)
     except requests.exceptions.RequestException as e:
         print(f'AI API 响应失败: {e}')
-        await sendTelegramMessage(chatId, 'AI API 响应失败，请稍后再试')
+        await editTelegramMessage(chatId, thinking_message_id, 'AI API 响应失败，请稍后再试')
     except Exception as error:
         print(f'处理消息时发生错误: {error}')
-        await sendTelegramMessage(chatId, '处理消息时发生错误，请稍后再试')
+        await editTelegramMessage(chatId, thinking_message_id, '处理消息时发生错误，请稍后再试')
+
 
 async def handleAiResponse(ai_data, chatId, history):
     if ai_data and ai_data.get('choices') and len(ai_data['choices']) > 0:
@@ -290,8 +295,28 @@ async def sendTelegramMessage(chatId, text, options={}):
     try:
         response = requests.post(url, headers={'Content-Type': 'application/json'}, json=data)
         response.raise_for_status()
+        return response.json()
     except requests.exceptions.RequestException as e:
         print(f'发送 Telegram 消息失败: {e}')
+        return {}
+
+async def editTelegramMessage(chatId, message_id, text, options={}):
+    url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/editMessageText"
+    if PHP_PROXY_URL:
+        method_name = url.split('/')[-1]
+        url = f"{PHP_PROXY_URL}{method_name}"
+    data = {
+        'chat_id': chatId,
+        'message_id': message_id,
+        'text': text,
+        **options
+    }
+    try:
+        response = requests.post(url, headers={'Content-Type': 'application/json'}, json=data)
+        response.raise_for_status()
+    except requests.exceptions.RequestException as e:
+        print(f'编辑 Telegram 消息失败: {e}')
+
 
 def getHelpMessage():
     return f"""