feat. 自动命名对话

ChuanhuChatbot.py

@@ -550,10 +550,14 @@ with gr.Blocks(theme=small_and_beautiful_theme) as demo:
                                                   chatgpt_predict_args).then(**end_outputing_args)
     user_input.submit(**get_usage_args)
 
+    user_input.submit(auto_name_chat_history, [current_model, user_question, chatbot, user_name], [historySelectList], show_progress=False)
+
     submitBtn.click(**transfer_input_args).then(**chatgpt_predict_args,
                                                 api_name="predict").then(**end_outputing_args)
     submitBtn.click(**get_usage_args)
 
+    submitBtn.click(auto_name_chat_history, [current_model, user_question, chatbot, user_name], [historySelectList], show_progress=False)
+
     index_files.change(handle_file_upload, [current_model, index_files, chatbot, language_select_dropdown], [
                        index_files, chatbot, status_display])
     summarize_btn.click(handle_summarize_index, [

modules/models/base_model.py

@@ -671,6 +671,15 @@ class BaseLLMModel:
         self.history_file_path = filename
         save_file(filename, self.system_prompt, self.history, chatbot, user_name)
         return init_history_list(user_name)
+    
+    def auto_name_chat_history(self, user_question, chatbot, user_name):
+        if chatbot == []:
+            filename = user_question[:12] + ".json"
+            return self.rename_chat_history(filename, chatbot, user_name)
+        else:
+            return gr.update()
+
+
 
     def auto_save(self, chatbot):
         save_file(self.history_file_path, self.system_prompt,

modules/utils.py

@@ -71,6 +71,9 @@ def set_system_prompt(current_model, *args):
 def rename_chat_history(current_model, *args):
     return current_model.rename_chat_history(*args)
 
+def auto_name_chat_history(current_model, *args):
+    return current_model.auto_name_chat_history(*args)
+
 def export_markdown(current_model, *args):
     return current_model.export_markdown(*args)