update buttons

app.py

@@ -27,8 +27,16 @@ def fix_tokenizer(tokenizer, new_lang='ami_Latn'):
 
 fix_tokenizer(small_tokenizer)
 
+# Language mapping
+lang_map = {
+    "汉语 Chinese": "zho_Hant",
+    "Amis": "ami_Latn"
+}
+
 # Translation function
-def translate(text, src_lang, tgt_lang):
+def translate(text, src_lang_label, tgt_lang_label):
+    src_lang = lang_map[src_lang_label]
+    tgt_lang = lang_map[tgt_lang_label]
     tokenizer, model = small_tokenizer, small_model
     if src_lang == "zho_Hant":
         text = preproc_chinese(text)
@@ -61,15 +69,22 @@ def preproc_chinese(text):
     clean = replace_nonprint(clean)
     return unicodedata.normalize("NFKC", clean)
 
-
+# Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# AMIS - Chinese Translation Tool")
-    src_lang = gr.Radio(choices=["汉语 Chinese", "Amis"], value="zho_Hant", label="Source Language")
-    tgt_lang = gr.Radio(choices=["Amis", "汉语 Chinese"], value="ami_Latn", label="Target Language")
+    src_lang = gr.Radio(choices=["汉语 Chinese", "Amis"], value="汉语 Chinese", label="Source Language")
+    tgt_lang = gr.Radio(choices=["Amis", "汉语 Chinese"], value="Amis", label="Target Language")
     input_text = gr.Textbox(label="Input Text", placeholder="Enter text here...")
     output_text = gr.Textbox(label="Translated Text", interactive=False)
     translate_btn = gr.Button("Translate")
-    translate_btn.click(translate, inputs=[input_text, src_lang, tgt_lang], outputs=output_text)
+    
+    # Link translation function
+    translate_btn.click(
+        translate, 
+        inputs=[input_text, src_lang, tgt_lang], 
+        outputs=output_text
+    )
+    
     gr.Markdown("感謝您在此專案上的辛勤工作 這只是這些模型能力的一小部分展示。電子郵件: [email protected]")
 
 if __name__ == "__main__":