Update app.py

app.py

@@ -2,34 +2,43 @@ import gradio as gr
 from openai import OpenAI
 import os
 
-# 简化版本的预测函数
+# 修复格式问题的预测函数
 def predict(message, history):
     try:
         # 初始化OpenAI客户端
         api_key = os.environ.get("API_TOKEN")
         if not api_key:
-            return "错误：API密钥未设置，请在Hugging Face Space设置中添加名为API_TOKEN的Secret。"
+            return history + [["错误：API密钥未设置", ""]]
             
         client = OpenAI(api_key=api_key)
         
-        # 构建消息
-        messages = [
-            {"role": "system", "content": "你是一个有用的AI助手。"},
-            {"role": "user", "content": message}
-        ]
+        # 构建消息列表，包含历史记录
+        messages = [{"role": "system", "content": "你是一个有用的AI助手。"}]
         
-        # 非流式调用，更简单更可靠
+        # 添加历史消息
+        for user_msg, bot_msg in history:
+            messages.append({"role": "user", "content": user_msg})
+            messages.append({"role": "assistant", "content": bot_msg})
+        
+        # 添加当前用户消息
+        messages.append({"role": "user", "content": message})
+        
+        # 非流式调用
         response = client.chat.completions.create(
             model="gpt-4o",
             messages=messages,
             max_tokens=1000,
             temperature=0.7,
-            stream=False  # 关闭流式响应
+            stream=False
         )
         
-        return response.choices[0].message.content
+        bot_response = response.choices[0].message.content
+        
+        # 返回新的历史记录，添加当前交互
+        return history + [[message, bot_response]]
     except Exception as e:
-        return f"发生错误: {str(e)}"
+        # 出错时也保持正确的格式
+        return history + [[message, f"发生错误: {str(e)}"]]
 
 # 创建简化版界面
 with gr.Blocks(css="""
@@ -53,9 +62,15 @@ with gr.Blocks(css="""
     
     chatbot = gr.Chatbot()
     msg = gr.Textbox(placeholder="输入你的问题...", show_label=False)
+    clear = gr.Button("清除对话")
+    
     send_btn = gr.Button("发送", elem_id="custom-send")
     
-    send_btn.click(predict, [msg, chatbot], [chatbot])
-    msg.submit(predict, [msg, chatbot], [chatbot])
+    # 重要：正确处理输入和状态
+    msg.submit(predict, [msg, chatbot], [chatbot]).then(
+        lambda: "", None, msg)
+    send_btn.click(predict, [msg, chatbot], [chatbot]).then(
+        lambda: "", None, msg)
+    clear.click(lambda: None, None, chatbot)
 
 demo.launch()