From fa154160245af6e9a575e39a638533b086bb6c7e Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Sun, 20 Aug 2023 16:40:00 +0800
Subject: [PATCH] fix #595

Former-commit-id: 5c052836a0a19afd556418f1f6da8db8fc3cd37d
---
 src/llmtuner/webui/interface.py |  2 +-
 src/llmtuner/webui/manager.py   | 13 +++++++------
 2 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/src/llmtuner/webui/interface.py b/src/llmtuner/webui/interface.py
index 1a5c0c19..3b351f63 100644
--- a/src/llmtuner/webui/interface.py
+++ b/src/llmtuner/webui/interface.py
@@ -59,7 +59,7 @@ def create_web_demo() -> gr.Blocks:
     chat_model = WebChatModel(lazy_init=False)
 
     with gr.Blocks(title="Web Demo", css=CSS) as demo:
-        lang = gr.Dropdown(choices=["en", "zh"])
+        lang = gr.Dropdown(choices=["en", "zh"], value="en")
 
         _, _, _, chat_elems = create_chat_box(chat_model, visible=True)
 
diff --git a/src/llmtuner/webui/manager.py b/src/llmtuner/webui/manager.py
index 63a4fa5b..0593657f 100644
--- a/src/llmtuner/webui/manager.py
+++ b/src/llmtuner/webui/manager.py
@@ -19,14 +19,15 @@ class Manager:
         }
 
         user_config = load_config()
-        refresh_dict = {}
+        if not lang:
+            if user_config.get("lang", None):
+                lang = user_config["lang"]
+            else:
+                lang = "en"
 
-        if "lang" in user_config:
-            refresh_dict["lang"] = {"value": user_config["lang"]}
-        else:
-            refresh_dict["lang"] = {"value": "en"}
+        refresh_dict["lang"] = {"value": lang}
 
-        if user_config.get("last_model"):
+        if user_config.get("last_model", None):
             refresh_dict["model_name"] = {"value": user_config["last_model"]}
             refresh_dict["model_path"] = {"value": get_model_path(user_config["last_model"])}