Merge pull request Byaidu#399 from namazuchin/feature/env-config-gui-…

…lang
hellofinch · Jan 4, 2025 · ca813fd · ca813fd
2 parents abdff65 + eb9bd9a
commit ca813fd
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/docs/README_GUI.md b/docs/README_GUI.md
@@ -8,6 +8,28 @@ This subfolder provides the GUI mode of `pdf2zh`.
 
 2. Drop the PDF file into the window and click `Translate`.
 
+### Environment Variables
+
+You can set the source and target languages using environment variables:
+
+- `PDF2ZH_LANG_FROM`: Sets the source language. Defaults to "English".
+- `PDF2ZH_LANG_TO`: Sets the target language. Defaults to "Simplified Chinese".
+
+### Supported Languages
+
+The following languages are supported:
+
+- English
+- Simplified Chinese
+- Traditional Chinese
+- French
+- German
+- Japanese
+- Korean
+- Russian
+- Spanish
+- Italian
+
 ## Preview
 
 <img src="./images/before.png" width="500"/>

diff --git a/pdf2zh/gui.py b/pdf2zh/gui.py
@@ -409,12 +409,12 @@ def progress_bar(t: tqdm.tqdm):
                 lang_from = gr.Dropdown(
                     label="Translate from",
                     choices=lang_map.keys(),
-                    value="English",
+                    value=os.getenv("PDF2ZH_LANG_FROM", "English"),
                 )
                 lang_to = gr.Dropdown(
                     label="Translate to",
                     choices=lang_map.keys(),
-                    value="Simplified Chinese",
+                    value=os.getenv("PDF2ZH_LANG_TO", "Simplified Chinese"),
                 )
             page_range = gr.Radio(
                 choices=page_map.keys(),