Update app.py

app.py

@@ -1,8 +1,9 @@
 import openai
 import gradio as gr
+import os
 
 # 设置您的API密钥
-openai.api_key = "your_openai_api_key_here"
+openai.api_key = os.environ[chat_key]
 
 # 存储用户对话历史的字典
 user_dialogue_histories = {}
@@ -20,7 +21,37 @@ def remove_earliest_messages(user_id, tokens_to_remove):
         tokens_to_remove -= len(removed_message["content"])
 
 def chat_with_chatgpt(user_id, user_message):
-    # ... 保持与之前相同的代码 ...
+    # 如果用户不在字典中，添加一个新的对话历史列表
+    if user_id not in user_dialogue_histories:
+        user_dialogue_histories[user_id] = []
+
+    # 将用户的消息添加到对话历史中
+    user_dialogue_histories[user_id].append({"role": "user", "content": user_message})
+
+    # 如果添加用户消息后对话历史中的token数超过限制，删除最早的消息，直到满足限制
+    if get_total_tokens(user_dialogue_histories[user_id]) > max_tokens_per_user:
+        remove_earliest_messages(user_id, get_total_tokens(user_dialogue_histories[user_id]) - max_tokens_per_user)
+
+    # 调用ChatGPT API
+    response = openai.Completion.create(
+        engine="text-davinci-002",
+        prompt=[{"role": "system", "content": "You are a helpful assistant."}] + user_dialogue_histories[user_id],
+        max_tokens=150,
+        n=1,
+        stop=None,
+        temperature=0.5,
+    )
+
+    chatgpt_response = response.choices[0].text.strip()
+
+    # 将ChatGPT的回应添加到对话历史中
+    user_dialogue_histories[user_id].append({"role": "assistant", "content": chatgpt_response})
+
+    # 如果添加ChatGPT的回应后对话历史中的token数超过限制，删除最早的消息，直到满足限制
+    if get_total_tokens(user_dialogue_histories[user_id]) > max_tokens_per_user:
+        remove_earliest_messages(user_id, get_total_tokens(user_dialogue_histories[user_id]) - max_tokens_per_user)
+
+    return chatgpt_response
 
 # Gradio界面
 def gradio_interface(user_id, user_message):