Update app.py

app.py

@@ -4,11 +4,7 @@ import random
 
 openai.api_key = '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'
 
-context = [
-    {"role": "system", "content": "你是一个聪明的AI助手。"},
-    {"role": "user", "content": "你会说中文吗？"},
-    {"role": "assistant", "content": "是的，我可以说中文。"}
-]
+user_contexts = {}
 
 def get_assistant_response(user_question, context):
     context.append({"role": "user", "content": user_question})
@@ -21,14 +17,24 @@ def get_assistant_response(user_question, context):
     context.append({"role": "assistant", "content": assistant_response})
     return assistant_response
 
-def greet(user_question, clear_history):
-    global context
+def greet(user_id, user_question, clear_history):
+    global user_contexts
+    if user_id not in user_contexts:
+        user_contexts[user_id] = [
+            {"role": "system", "content": "你是一个聪明的AI助手。"},
+            {"role": "user", "content": "你会说中文吗？"},
+            {"role": "assistant", "content": "是的，我可以说中文。"}
+        ]
+
+    context = user_contexts[user_id]
+
     if clear_history:
         context = [
             {"role": "system", "content": "你是一个聪明的AI助手。"},
             {"role": "user", "content": "你会说中文吗？"},
             {"role": "assistant", "content": "是的，我可以说中文。"}
         ]
+        user_contexts[user_id] = context
         return '清空成功', '保持聊天记录'
     else:
         get_assistant_response(user_question, context)
@@ -41,16 +47,17 @@ def greet(user_question, clear_history):
 demo = gr.Interface(
     fn=greet,
     inputs=[
-        gr.Textbox(lines=20, label='请输入问题', placeholder='请输入您的问题'),
+        gr.Textbox(lines=1, label='请输入用户ID', placeholder='请输入用户ID'),
+        gr.Textbox(lines=15, label='请输入问题', placeholder='请输入您的问题'),
         gr.Checkbox(label='清空聊天记录', default=False)
     ],
     outputs=[
         gr.Textbox(lines=1, label='聊天记录状态', placeholder='等待清空聊天记录'),
-        gr.Textbox(lines=18, label='AI回答', placeholder='等待AI回答')
+        gr.Textbox(lines=20, label='AI回答', placeholder='等待AI回答')
     ],
     title="AI助手",
     description="请输入您的问题，AI助手会给出回答。支持连续对话，可以记录对话历史。重新开始对话勾选清空聊天记录，输出清空成功表示重新开启对话。"
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()