Răsfoiți Sursa

GUI 改为 OpenAI-liked 默认服务;新增传参是否允许局域网访问

mrh 1 an în urmă
părinte
comite
964e0aa757
2 a modificat fișierele cu 17 adăugiri și 9 ștergeri
  1. 1 0
      .gitignore
  2. 16 9
      pdf2zh/gui.py

+ 1 - 0
.gitignore

@@ -172,3 +172,4 @@ cython_debug/
 uv.lock
 *.pdf
 *.docx
+.aider*

+ 16 - 9
pdf2zh/gui.py

@@ -43,15 +43,18 @@ from pdf2zh.translator import (
 
 # The following variables associate strings with translators
 service_map: dict[str, BaseTranslator] = {
+    "OpenAI-liked": OpenAIlikedTranslator,
+    "DeepSeek": DeepseekTranslator,
+    "Zhipu": ZhipuTranslator,
     "Google": GoogleTranslator,
     "Bing": BingTranslator,
+    "Groq": GroqTranslator,
+    "OpenAI": OpenAITranslator,
     "DeepL": DeepLTranslator,
     "DeepLX": DeepLXTranslator,
     "Ollama": OllamaTranslator,
     "Xinference": XinferenceTranslator,
     "AzureOpenAI": AzureOpenAITranslator,
-    "OpenAI": OpenAITranslator,
-    "Zhipu": ZhipuTranslator,
     "ModelScope": ModelScopeTranslator,
     "Silicon": SiliconTranslator,
     "Gemini": GeminiTranslator,
@@ -61,9 +64,6 @@ service_map: dict[str, BaseTranslator] = {
     "AnythingLLM": AnythingLLMTranslator,
     "Argos Translate": ArgosTranslator,
     "Gork": GorkTranslator,
-    "Groq": GroqTranslator,
-    "DeepSeek": DeepseekTranslator,
-    "OpenAI-liked": OpenAIlikedTranslator,
 }
 
 # The following variables associate strings with specific languages
@@ -399,7 +399,7 @@ with gr.Blocks(
             service = gr.Dropdown(
                 label="Service",
                 choices=service_map.keys(),
-                value="Google",
+                value="OpenAI-liked",
             )
             envs = []
             for i in range(3):
@@ -518,6 +518,13 @@ with gr.Blocks(
             gr.Markdown("## Preview")
             preview = PDF(label="Document Preview", visible=True, height=2000)
 
+    # Page load event
+    demo.load(
+        on_select_service,
+        inputs=[service],
+        outputs=envs,
+    )
+
     # Event handlers
     file_input.upload(
         lambda x: x,
@@ -606,7 +613,7 @@ def parse_user_passwd(file_path: str) -> tuple:
 
 
 def setup_gui(
-    share: bool = False, auth_file: list = ["", ""], server_port=7860
+    share: bool = False, auth_file: list = ["", ""], server_port=7860, public: bool = False
 ) -> None:
     """
     Setup the GUI with the given parameters.
@@ -625,7 +632,7 @@ def setup_gui(
         if len(user_list) == 0:
             try:
                 demo.launch(
-                    server_name="0.0.0.0",
+                    server_name="127.0.0.1" if not public else "0.0.0.0",
                     debug=True,
                     inbrowser=True,
                     share=share,
@@ -653,7 +660,7 @@ def setup_gui(
         else:
             try:
                 demo.launch(
-                    server_name="0.0.0.0",
+                    server_name="127.0.0.1" if not public else "0.0.0.0",
                     debug=True,
                     inbrowser=True,
                     share=share,