Update bin_public/app/Chatbot.py

bin_public/app/Chatbot.py

@@ -117,24 +117,31 @@ with gr.Blocks(css=customCSS) as demo:
                 with gr.Tab(label="保存/加载"):
                     with gr.Accordion(label="保存/加载对话历史记录", open=True):
                         with gr.Column():
+                            with gr.Row():
+                                with gr.Column(scale=6):
+                                    historyFileSelectDropdown = gr.Dropdown(
+                                        label="从列表中加载对话",
+                                        choices=get_history_names(plain=True),
+                                        multiselect=False,
+                                        value=get_history_names(plain=True)[0],
+                                        visible=False
+                                    )
+
                             with gr.Row():
                                 with gr.Column(scale=6):
                                     saveFileName = gr.Textbox(
-                                        show_label=True, placeholder=f"在这里输入保存的文件名...", label="设置保存文件名",
-                                        value="对话历史记录").style(container=True)
+                                        show_label=True,
+                                        placeholder=f"设置文件名: 默认为.json，可选为.md",
+                                        label="设置保存文件名",
+                                        value="对话历史记录",
+                                    ).style(container=True)
                                 with gr.Column(scale=1):
                                     saveHistoryBtn = gr.Button("💾 保存对话")
-                                    chat_history_file = gr.File(file_types=['json'], interactive=True)
-                            #with gr.Row():
-                                '''with gr.Column(scale=6):
-                                    historyFileSelectDropdown = gr.Dropdown(label="从列表中加载对话",
-                                                                            choices=get_history_names(plain=True),
-                                                                            multiselect=False,
-                                                                            value=get_history_names(plain=True)[0])'''
-                                #with gr.Column(scale=1):
-                                    #historyRefreshBtn = gr.Button("🔄 刷新")
-                                    # 加一个invisible button，用来清除文件，不加会无法清楚生成的文件
-                                    #fileRefreshBtn = gr.Button("🔄 文件刷新", visible=False)
+                                    exportMarkdownBtn = gr.Button("📝 导出为Markdown")
+                                    gr.Markdown("默认保存于history文件夹")
+                            with gr.Row():
+                                with gr.Column():
+                                    downloadFile = gr.File(interactive=True)
 
     gr.HTML("""
     <div style="text-align: center; margin-top: 20px; margin-bottom: 20px;">
@@ -167,25 +174,29 @@ with gr.Blocks(css=customCSS) as demo:
                                              temperature, use_streaming_checkbox, model_select_dropdown],
                          [chatbot, history, status_display, token_count], show_progress=True)
     # History
-    saveHistoryBtn.click(save_chat_history, [
-        saveFileName, systemPromptTxt, history, chatbot], outputs=chat_history_file, show_progress=True)
-
-    #saveHistoryBtn.click(get_history_names, None, None)
-
+    saveHistoryBtn.click(
+        save_chat_history,
+        [saveFileName, systemPromptTxt, history, chatbot],
+        downloadFile,
+        show_progress=True,
+    )
+    saveHistoryBtn.click(get_history_names, None, [historyFileSelectDropdown])
+    exportMarkdownBtn.click(
+        export_markdown,
+        [saveFileName, systemPromptTxt, history, chatbot],
+        downloadFile,
+        show_progress=True,
+    )
     #historyRefreshBtn.click(get_history_names, None, [historyFileSelectDropdown])
-
-    #fileRefreshBtn.click(reset_file, inputs=chat_history_file, outputs=None)
-
-    '''historyFileSelectDropdown.change(
+    historyFileSelectDropdown.change(
         load_chat_history,
         [historyFileSelectDropdown, systemPromptTxt, history, chatbot],
         [saveFileName, systemPromptTxt, history, chatbot],
         show_progress=True,
-    )'''
-
-    chat_history_file.change(
+    )
+    downloadFile.change(
         load_chat_history,
-        [chat_history_file, systemPromptTxt, history, chatbot],
+        [downloadFile, systemPromptTxt, history, chatbot],
         [saveFileName, systemPromptTxt, history, chatbot],
     )