leonsimon23 commited on
Commit
7fad2da
·
verified ·
1 Parent(s): 1604e64

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +34 -0
app.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import openai
3
+
4
+ # 设置 OpenAI API 密钥
5
+ openai.api_key = "your-openai-api-key-here"
6
+
7
+ # 定义 AI 响应函数
8
+ def ai_response(message, chat_history):
9
+ # 组合历史聊天记录和用户输入的信息
10
+ conversation = [{"role": "system", "content": "You are a helpful assistant."}]
11
+ for msg in chat_history:
12
+ conversation.append({"role": msg['role'], "content": msg['message']})
13
+ conversation.append({"role": "user", "content": message})
14
+
15
+ # 请求 OpenAI 的 GPT-3.5 或 GPT-4 模型生成响应
16
+ response = openai.ChatCompletion.create(
17
+ model="gpt-4", # 可以根据需要选择不同的模型
18
+ messages=conversation
19
+ )
20
+
21
+ # 获取模型的回答
22
+ assistant_message = response['choices'][0]['message']['content']
23
+
24
+ # 返回新的聊天记录
25
+ chat_history.append({"role": "user", "message": message})
26
+ chat_history.append({"role": "assistant", "message": assistant_message})
27
+
28
+ return assistant_message, chat_history
29
+
30
+ # 创建 Gradio ChatInterface
31
+ chat_interface = gr.ChatInterface(fn=ai_response, live=True)
32
+
33
+ # 启动 Gradio 应用
34
+ chat_interface.launch()