Update app.py

app.py

@@ -5,6 +5,8 @@ import requests
 API_BASE_URL = "http://key.aistory.uk/v1/chat/completions"  # 替换为你自己的API URL
 API_KEY = "sk-HfD4NYIN6bq2DkSfIiUcciRvo9MkgMdFCsahP9NWEOUPHe8H"  # 替换为你自己的 API 密钥
 
+
+
 # 定义 AI 响应函数，调用第三方 API
 def ai_response(message, chat_history):
     # 定义系统提示词
@@ -13,7 +15,7 @@ def ai_response(message, chat_history):
     # 组合历史聊天记录和用户输入的信息
     conversation = [{"role": "system", "content": system_prompt}]
     for msg in chat_history:
-        conversation.append({"role": msg['role'], "content": msg['message']})
+        conversation.append({"role": msg[0], "content": msg[1]})
     conversation.append({"role": "user", "content": message})
 
     # 构建请求体
@@ -37,14 +39,14 @@ def ai_response(message, chat_history):
         response_data = response.json()
         assistant_message = response_data['choices'][0]['message']['content']
         
-        # 返回新的聊天记录
-        chat_history.append({"role": "user", "message": message})
-        chat_history.append({"role": "assistant", "message": assistant_message})
+        # 返回新的聊天记录，转换为符合 gr.Chatbot 期望的元组格式
+        chat_history.append(("user", message))
+        chat_history.append(("assistant", assistant_message))
         
-        return assistant_message, chat_history
+        return chat_history
     else:
         # 如果请求失败，返回错误消息
-        return "Sorry, something went wrong with the API.", chat_history
+        return chat_history + [("assistant", "Sorry, something went wrong with the API.")]
 
 # 创建 Gradio 应用
 def create_interface():