add Chinese

app.py

@@ -22,6 +22,9 @@ from prompts import TRANS_Prompt, TRANS_Prompt_template
 knowledge_cls = knowledge_class()
 
 
+language_dict = {"en": "English", "cn": "Chinese"}
+
+
 def update_knowledge():
     knowledge_cls.get_new_knowledge()
     choices_list = list(knowledge_cls.reference_dict.keys())
@@ -46,10 +49,13 @@ def answer_question(
     if not answer_text:
         yield history
         return
-    if language_flag:
+    if language_flag not in ["jp"]:
         info_title = info_title + f"日本語返信:\n {answer_text}\n"
-        user_prompt = TRANS_Prompt_template.format(japanese_text=answer_text)
-        answer_text = generate_item(user_prompt, TRANS_Prompt)
+        sys_prompt = TRANS_Prompt.format(lang=language_dict[language_flag])
+        user_prompt = TRANS_Prompt_template.format(
+            japanese_text=answer_text, lang=language_dict[language_flag]
+        )
+        answer_text = generate_item(user_prompt, sys_prompt)
 
     # text
     history.append({"role": "assistant", "content": ""})
@@ -150,8 +156,8 @@ with gr.Blocks(title="勉強会", css=custom_css, head=custom_head, js=js_func)
             label="返信スタイル",
         )
         language_toggle = gr.Radio(
-            [("English", True), ("日本語", False)],
-            value=False,
+            [("日本語", "jp"), ("中文", "cn"), ("English", "en")],
+            value="jp",
             label="言語スタイル",
         )
         with gr.Accordion("話者紹介", open=False):

prompts.py

@@ -3,8 +3,8 @@ You are a specialist in linguistic analysis, skilled at identifying, emulating,
 """
 
 TRANS_Prompt = """
-You are a skilled Japanese-to-English translator specializing in conversational text. 
-Translate the following Japanese dialogue into natural, fluent English, preserving the original tone, personality, and level of formality of each speaker.
+You are a skilled Japanese-to-{lang} translator specializing in conversational text. 
+Translate the following Japanese dialogue into natural, fluent {lang}, preserving the original tone, personality, and level of formality of each speaker.
 """
 
 SUMMARY_Prompt = """
@@ -19,8 +19,8 @@ Present your analysis in Japanese.
 """
 
 TRANS_Prompt_template = """
-Ensure the English reads like real dialogue, suitable for a script, subtitle, or chat. Do not include explanations or the original Japanese.
-Only output the English translation. Do not include the original Japanese text, any explanations, notes, or formatting.
+Ensure the {lang} reads like real dialogue, suitable for a script, subtitle, or chat. Do not include explanations or the original Japanese.
+Only output the {lang} translation. Do not include the original Japanese text, any explanations, notes, or formatting.
 Japanese text:
 {japanese_text}
 """