去掉更新文档参数按钮

ChuanhuChatbot.py

@@ -98,7 +98,6 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                     two_column = gr.Checkbox(label="双栏pdf", value=advance_docs["pdf"].get("two_column", False))
                     # TODO: 公式ocr
                     # formula_ocr = gr.Checkbox(label="识别公式", value=advance_docs["pdf"].get("formula_ocr", False))
-                    updateDocConfigBtn = gr.Button("更新解析文件参数")
 
                 with gr.Tab(label="Prompt"):
                     systemPromptTxt = gr.Textbox(
@@ -314,7 +313,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     )
     reduceTokenBtn.click(**get_usage_args)
 
-    updateDocConfigBtn.click(update_doc_config, [two_column], None)
+    two_column.change(update_doc_config, [two_column], None)
 
     # ChatGPT
     keyTxt.change(submit_key, keyTxt, [user_api_key, status_display]).then(**get_usage_args)

modules/config.py

@@ -127,7 +127,7 @@ def retrieve_proxy(proxy=None):
         os.environ["HTTP_PROXY"] = http_proxy
         os.environ["HTTPS_PROXY"] = https_proxy
         yield http_proxy, https_proxy # return new proxy
-        
+
         # return old proxy
         os.environ["HTTP_PROXY"], os.environ["HTTPS_PROXY"] = old_var
 
@@ -137,7 +137,6 @@ advance_docs = defaultdict(lambda: defaultdict(dict))
 advance_docs.update(config.get("advance_docs", {}))
 def update_doc_config(two_column_pdf):
     global advance_docs
-    if two_column_pdf:
-        advance_docs["pdf"]["two_column"] = True
-    
+    advance_docs["pdf"]["two_column"] = two_column_pdf
+
     logging.info(f"更新后的文件参数为：{advance_docs}")