bugfix: 现在保存文件名可以被自动加载

ChuanhuChatbot.py

@@ -483,13 +483,14 @@ with gr.Blocks(theme=small_and_beautiful_theme) as demo:
             model_name=MODELS[DEFAULT_MODEL], access_key=my_api_key)[0]
         current_model.set_user_identifier(user_name)
         if not hide_history_when_not_logged_in or user_name:
-            system_prompt, chatbot = current_model.auto_load()
+            filename, system_prompt, chatbot = current_model.auto_load()
         else:
             system_prompt = gr.update()
+            filename = gr.update()
             chatbot = gr.Chatbot.update(label=MODELS[DEFAULT_MODEL])
-        return user_info, user_name, current_model, toggle_like_btn_visibility(DEFAULT_MODEL), system_prompt, chatbot, init_history_list(user_name)
+        return user_info, user_name, current_model, toggle_like_btn_visibility(DEFAULT_MODEL), filename, system_prompt, chatbot, init_history_list(user_name)
     demo.load(create_greeting, inputs=None, outputs=[
-              user_info, user_name, current_model, like_dislike_area, systemPromptTxt, chatbot, historySelectList], api_name="load")
+              user_info, user_name, current_model, like_dislike_area, saveFileName, systemPromptTxt, chatbot, historySelectList], api_name="load")
     chatgpt_predict_args = dict(
         fn=predict,
         inputs=[

modules/models/base_model.py

@@ -746,7 +746,8 @@ class BaseLLMModel:
     def auto_load(self):
         self.history_file_path = get_history_filepath(self.user_identifier)
         filename, system_prompt, chatbot = self.load_chat_history()
-        return system_prompt, chatbot
+        filename = filename[:-5]
+        return filename, system_prompt, chatbot
 
     def like(self):
         """like the last response, implement if needed