Comprehensive fix for AI response display issues - ensure proper error handling and message consistency

app.py

@@ -294,12 +294,25 @@ def respond(
         token_to_use = os.environ.get("HF_TOKEN", "")
         
     if not token_to_use:
-        yield "错误：需要Hugging Face访问令牌。请在环境变量中设置HF_TOKEN。"
+        error_msg = "错误：需要Hugging Face访问令牌。请在环境变量中设置HF_TOKEN。"
+        # 保存错误信息到历史记录
+        conversation_data = {
+            "timestamp": datetime.now().isoformat(),
+            "message": message,
+            "history": history,
+            "response": error_msg
+        }
+        history_list = load_history()
+        history_list.append(conversation_data)
+        save_history(history_list)
+        
         # 发送webhook通知
         send_webhook_notification("error", {
             "type": "missing_hf_token",
-            "message": "未提供HF_TOKEN"
+            "message": "未提供HF_TOKEN",
+            "user_message": message
         })
+        yield error_msg
         return
     
     # 保存对话历史
@@ -346,12 +359,19 @@ def respond(
         
         if client is None:
             error_msg = "错误：无法连接到任何可用的模型。请检查您的Hugging Face访问令牌权限。"
-            yield error_msg
+            # 保存错误信息到历史记录
+            conversation_data["response"] = error_msg
+            history_list = load_history()
+            history_list.append(conversation_data)
+            save_history(history_list)
+            
             # 发送webhook通知
             send_webhook_notification("error", {
                 "type": "model_connection_failed",
-                "message": "无法连接到任何可用模型"
+                "message": "无法连接到任何可用模型",
+                "user_message": message
             })
+            yield error_msg
             return
             
         messages = [{"role": "system", "content": effective_system_message}]
@@ -384,7 +404,7 @@ def respond(
         # 发送webhook通知（仅在有重要对话时）
         if len(response) > 50:  # 只有较长的回复才发送通知
             send_webhook_notification("new_conversation", {
-                "user_message": conversation_data["message"],  # 使用conversation_data中的原始消息
+                "user_message": message,
                 "ai_response": response,
                 "model_used": model_used,
                 "character_role": character_role
@@ -403,7 +423,7 @@ def respond(
         send_webhook_notification("error", {
             "type": "processing_error",
             "message": str(e),
-            "user_message": conversation_data["message"]  # 使用conversation_data中的原始消息
+            "user_message": message
         })