Update app.py

app.py

@@ -2,10 +2,8 @@ import openai
 import gradio as gr
 import random
 
-
 openai.api_key = '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'
 
-# 上下文列表，包含交互历史
 context = [
     {"role": "system", "content": "你是一个聪明的AI助手。"},
     {"role": "user", "content": "你会说中文吗？"},
@@ -13,41 +11,41 @@ context = [
 ]
 
 def get_assistant_response(user_question, context):
-    # 将问题添加到上下文列表
     context.append({"role": "user", "content": user_question})
-    # 调用API
     response = openai.ChatCompletion.create(
-        model = 'gpt-3.5-turbo',
-        messages = context,
-        temperature = 0  
-      )
-
-    # 获取回答
+        model='gpt-3.5-turbo',
+        messages=context,
+        temperature=0
+    )
     assistant_response = response.choices[0].message['content']
-
-    # 将回答添加到上下文列表
     context.append({"role": "assistant", "content": assistant_response})
-
     return assistant_response
 
-def greet(question):
+def greet(user_question, clear_history):
     global context
-    if question=='clear':
+    if clear_history:
         context = [
-              {"role": "system", "content": "你是一个聪明的AI助手。"},
-              {"role": "user", "content": "你会说中文吗？"},
-              {"role": "assistant", "content": "是的，我可以说中文。"}
-          ]
-        return 'clear success'
+            {"role": "system", "content": "你是一个聪明的AI助手。"},
+            {"role": "user", "content": "你会说中文吗？"},
+            {"role": "assistant", "content": "是的，我可以说中文。"}
+        ]
+        return '清空成功', ''
     else:
-        return get_assistant_response(question, context)
-        # return type(question)
-
-demo = gr.Interface(fn=greet, inputs=gr.Textbox(lines=20, placeholder='Please enter your question'), 
-                    outputs=gr.Textbox(lines=20, placeholder='Waiting for AI to answer'),
-                    title="AI助手",
-                    description="请输入您的问题，AI助手会给出回答。支持连续对话，可以记录对话历史。重新开始对话输入clear，输出clear success表示重新开启对话。")
+        return '', get_assistant_response(user_question, context)
+
+demo = gr.Interface(
+    fn=greet,
+    inputs=[
+        gr.Textbox(lines=21, label='请输入问题', placeholder='请输入您的问题'),
+        gr.Checkbox(label='清空聊天记录', default=False)
+    ],
+    outputs=[
+        gr.Textbox(lines=1, label='聊天记录已清空', placeholder='等待清空聊天记录'),
+        gr.Textbox(lines=20, label='AI回答', placeholder='等待AI回答')
+    ],
+    title="AI助手",
+    description="请输入您的问题，AI助手会给出回答。支持连续对话，可以记录对话历史。重新开始对话勾选清空聊天记录，输出清空成功表示重新开启对话。"
+)
 
 if __name__ == "__main__":
-    # demo.launch(share='True')
-    demo.launch()
+    demo.launch()