Spaces:
Sleeping
Sleeping
Tuchuanhuhuhu
commited on
Commit
•
ba1c857
1
Parent(s):
eab29c1
改动:将历史记录文件保存在文件夹里
Browse files- ChuanhuChatbot.py +13 -8
ChuanhuChatbot.py
CHANGED
@@ -10,6 +10,7 @@ import requests
|
|
10 |
my_api_key = "" # 在这里输入你的 API 密钥
|
11 |
initial_prompt = "You are a helpful assistant."
|
12 |
API_URL = "https://api.openai.com/v1/chat/completions"
|
|
|
13 |
|
14 |
|
15 |
|
@@ -18,7 +19,7 @@ if os.environ.get('dockerrun') == 'yes':
|
|
18 |
dockerflag = True
|
19 |
else:
|
20 |
dockerflag = False
|
21 |
-
|
22 |
if dockerflag:
|
23 |
my_api_key = os.environ.get('my_api_key')
|
24 |
if my_api_key == "empty":
|
@@ -159,25 +160,29 @@ def delete_last_conversation(chatbot, history):
|
|
159 |
history.pop()
|
160 |
return chatbot, history
|
161 |
|
162 |
-
def save_chat_history(
|
163 |
-
if
|
164 |
return
|
165 |
-
if not
|
166 |
-
|
|
|
167 |
json_s = {"system": system, "history": history, "chatbot": chatbot}
|
168 |
-
with open(
|
169 |
json.dump(json_s, f)
|
170 |
|
171 |
|
172 |
def load_chat_history(filename):
|
173 |
-
with open(filename, "r") as f:
|
174 |
json_s = json.load(f)
|
175 |
return filename, json_s["system"], json_s["history"], json_s["chatbot"]
|
176 |
|
177 |
|
178 |
def get_history_names(plain=False):
|
179 |
# find all json files in the current directory and return their names
|
180 |
-
|
|
|
|
|
|
|
181 |
if plain:
|
182 |
return files
|
183 |
else:
|
|
|
10 |
my_api_key = "" # 在这里输入你的 API 密钥
|
11 |
initial_prompt = "You are a helpful assistant."
|
12 |
API_URL = "https://api.openai.com/v1/chat/completions"
|
13 |
+
HISTORY_DIR = "history"
|
14 |
|
15 |
|
16 |
|
|
|
19 |
dockerflag = True
|
20 |
else:
|
21 |
dockerflag = False
|
22 |
+
|
23 |
if dockerflag:
|
24 |
my_api_key = os.environ.get('my_api_key')
|
25 |
if my_api_key == "empty":
|
|
|
160 |
history.pop()
|
161 |
return chatbot, history
|
162 |
|
163 |
+
def save_chat_history(filename, system, history, chatbot):
|
164 |
+
if filename == "":
|
165 |
return
|
166 |
+
if not filename.endswith(".json"):
|
167 |
+
filename += ".json"
|
168 |
+
os.makedirs(HISTORY_DIR, exist_ok=True)
|
169 |
json_s = {"system": system, "history": history, "chatbot": chatbot}
|
170 |
+
with open(os.path.join(HISTORY_DIR, filename), "w") as f:
|
171 |
json.dump(json_s, f)
|
172 |
|
173 |
|
174 |
def load_chat_history(filename):
|
175 |
+
with open(os.path.join(HISTORY_DIR, filename), "r") as f:
|
176 |
json_s = json.load(f)
|
177 |
return filename, json_s["system"], json_s["history"], json_s["chatbot"]
|
178 |
|
179 |
|
180 |
def get_history_names(plain=False):
|
181 |
# find all json files in the current directory and return their names
|
182 |
+
try:
|
183 |
+
files = [f for f in os.listdir(HISTORY_DIR) if f.endswith(".json")]
|
184 |
+
except FileNotFoundError:
|
185 |
+
files = []
|
186 |
if plain:
|
187 |
return files
|
188 |
else:
|