Upload 37 files

ChuanhuChatbot.py

@@ -33,6 +33,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
         with gr.Column():
             gr.HTML(title)
             user_info = gr.Markdown(value="", elem_id="user_info")
+        gr.HTML('<center><a href="https://huggingface.co/spaces/JohnSmith9982/ChuanhuChatGPT?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a></center>')
         status_display = gr.Markdown(get_geoip(), elem_id="status_display")
 
         # https://github.com/gradio-app/gradio/pull/3296
@@ -98,7 +99,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 +314,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
 
@@ -139,5 +139,7 @@ def update_doc_config(two_column_pdf):
     global advance_docs
     if two_column_pdf:
         advance_docs["pdf"]["two_column"] = True
-    
+    else:
+        advance_docs["pdf"]["two_column"] = False
+
     logging.info(f"更新后的文件参数为：{advance_docs}")