add zh-tw translation function

app.py

@@ -276,8 +276,8 @@ def process_text(text, src_language, target_language, max_iterations_value, thre
     return orig_output, plan2align_output, best_of_n_output, mpc_output
 
 # ---------- Gradio ----------
-target_languages = ["Chinese", "English", "Russian", "German", "Japanese", "Korean"]
-src_languages = ["Chinese", "English", "Russian", "German", "Japanese", "Korean"]
+target_languages = ["Traditional Chinese", "Simplified Chinese", "English", "Russian", "German", "Japanese", "Korean"]
+src_languages = ["Traditional Chinese", "Simplified Chinese", "English", "Russian", "German", "Japanese", "Korean"]
 
 with gr.Blocks(title="Test-Time Machine Translation with Plan2Align") as demo:
     state = gr.State({})
@@ -294,7 +294,7 @@ with gr.Blocks(title="Test-Time Machine Translation with Plan2Align") as demo:
             )
             src_language_input = gr.Dropdown(
                 choices=src_languages,
-                value="Chinese",
+                value="Traditional Chinese",
                 label="Source Language"
             )
             task_language_input = gr.Dropdown(
@@ -350,11 +350,11 @@ with gr.Blocks(title="Test-Time Machine Translation with Plan2Align") as demo:
     
     gr.Examples(
         examples=[
-            ["台灣夜市文化豐富多彩，從士林夜市到饒河街夜市，提供各種美食、遊戲和購物體驗，吸引了無數遊客。", "Chinese", "English", 2, 0.7, 1, ["Original", "Plan2Align"]],
-            ["台北101曾經是世界最高的建築物，它不僅是台灣的地標，也象徵著經濟成就和創新精神。", "Chinese", "Russian", 2, 0.7, 1, ["Original", "Plan2Align"]],
-            ["阿里山日出和森林鐵路是台灣最著名的自然景觀之一，每年吸引數十萬遊客前來欣賞雲海和壯麗的日出。", "Chinese", "German", 2, 0.7, 1, ["Original", "Plan2Align"]],
-            ["珍珠奶茶，這款源自台灣的獨特飲品，不僅在台灣本地深受喜愛，更以其獨特的風味和口感，在全球掀起了一股熱潮，成為了一種跨越文化、風靡全球的時尚飲品。", "Chinese", "Japanese", 3, 0.7, 3, ["Original", "Plan2Align"]],
-            ["原住民文化如同一片深邃的星空，閃爍著無數璀璨的傳統與藝術光芒。他們的歌舞，是與祖靈對話的旋律，是與自然共鳴的節奏，每一個舞步、每一聲吟唱，都承載著古老的傳說與智慧。編織，是他們巧手下的詩篇，一絲一線，交織出生命的紋理，也編織出對土地的熱愛與敬畏。木雕，則是他們與自然對話的雕塑，每一刀、每一鑿，都刻畫著對萬物的觀察與敬意，也雕琢出對祖先的追憶與傳承。", "Chinese", "Korean", 5, 0.7, 5, ["Original", "Plan2Align"]]
+            ["台灣夜市文化豐富多彩，從士林夜市到饒河街夜市，提供各種美食、遊戲和購物體驗，吸引了無數遊客。", "Traditional Chinese", "English", 2, 0.7, 1, ["Original", "Plan2Align"]],
+            ["台北101曾經是世界最高的建築物，它不僅是台灣的地標，也象徵著經濟成就和創新精神。", "Traditional Chinese", "Russian", 2, 0.7, 1, ["Original", "Plan2Align"]],
+            ["阿里山日出和森林鐵路是台灣最著名的自然景觀之一，每年吸引數十萬遊客前來欣賞雲海和壯麗的日出。", "Traditional Chinese", "German", 2, 0.7, 1, ["Original", "Plan2Align"]],
+            ["珍珠奶茶，這款源自台灣的獨特飲品，不僅在台灣本地深受喜愛，更以其獨特的風味和口感，在全球掀起了一股熱潮，成為了一種跨越文化、風靡全球的時尚飲品。", "Traditional Chinese", "Japanese", 3, 0.7, 3, ["Original", "Plan2Align"]],
+            ["原住民文化如同一片深邃的星空，閃爍著無數璀璨的傳統與藝術光芒。他們的歌舞，是與祖靈對話的旋律，是與自然共鳴的節奏，每一個舞步、每一聲吟唱，都承載著古老的傳說與智慧。編織，是他們巧手下的詩篇，一絲一線，交織出生命的紋理，也編織出對土地的熱愛與敬畏。木雕，則是他們與自然對話的雕塑，每一刀、每一鑿，都刻畫著對萬物的觀察與敬意，也雕琢出對祖先的追憶與傳承。", "Traditional Chinese", "Korean", 5, 0.7, 5, ["Original", "Plan2Align"]]
         ],
         inputs=[
             source_text,

vecalign/plan2align.py

@@ -25,7 +25,8 @@ lang_map = {
     "Japanese": ("ja", "ja_core_news_sm"),
     "Korean": ("ko", "ko_core_news_sm"),
     "Spanish": ("es", "es_core_news_sm"),
-    "Chinese": ("zh", "zh_core_web_sm")
+    "Simplified Chinese": ("zh", "zh_core_web_sm"),
+    "Traditional Chinese": ("zh", "zh_core_web_sm")
 }
 
 ################################# folder / file processing #################################