Update app.py

app.py

@@ -56,6 +56,7 @@ with st.sidebar:
 # Label dictionary for multilingual UI
 LABELS = {
     "user_type": {"English": "I am a...", "中文": "我的身份是...", "Malay": "Saya seorang..."},
+    "user_type_options": {"English": ["Patient", "Medical Professional"], "中文": ["病患", "醫護人員"], "Malay": ["Pesakit", "Profesional Perubatan"]},
     "input_header": {"English": "Enter Health Info", "中文": "輸入健康資料", "Malay": "Masukkan Maklumat Kesihatan"},
     "age": {"English": "Patient's Age", "中文": "年齡", "Malay": "Umur"},
     "height": {"English": "Patient's Height (cm)", "中文": "身高 (cm)", "Malay": "Tinggi (cm)"},
@@ -91,18 +92,18 @@ with st.sidebar:
 if selected == MENU[lang][0]:
     with st.sidebar:
         st.header(LABELS["input_header"][lang])
-        user_type = st.selectbox(LABELS["user_type"][lang], ["Patient", "Medical Professional"])
+        user_type = st.selectbox(LABELS["user_type"][lang], LABELS["user_type_options"][lang])
         age = st.number_input(LABELS["age"][lang], min_value=18, max_value=100, value=50)
         height = st.number_input(LABELS["height"][lang], min_value=100, max_value=250, value=170)
         weight = st.number_input(LABELS["weight"][lang], min_value=30, max_value=200, value=70)
         systolic = st.number_input(LABELS["systolic"][lang], min_value=80, max_value=250, value=120)
         diastolic = st.number_input(LABELS["diastolic"][lang], min_value=40, max_value=150, value=80)
-        cholesterol = st.selectbox(LABELS["cholesterol"][lang], [0, 1, 2], format_func=lambda x: {0: "Normal", 1: "High", 2: "Very High"}[x])
-        gluc = st.selectbox(LABELS["gluc"][lang], [0, 1, 2], format_func=lambda x: {0: "Normal", 1: "High", 2: "Very High"}[x])
-        gender = st.selectbox(LABELS["gender"][lang], [0, 1], format_func=lambda x: {0: "Female", 1: "Male"}[x])
-        smoke = st.selectbox(LABELS["smoke"][lang], [0, 1], format_func=lambda x: "Yes" if x else "No")
-        alco = st.selectbox(LABELS["alco"][lang], [0, 1], format_func=lambda x: "Yes" if x else "No")
-        active = st.selectbox(LABELS["active"][lang], [0, 1], format_func=lambda x: "Yes" if x else "No")
+        cholesterol = st.selectbox(LABELS["cholesterol"][lang], [0, 1, 2], format_func=lambda x: chol_options[lang][x])
+        gluc = st.selectbox(LABELS["gluc"][lang], [0, 1, 2], format_func=lambda x: gluc_options[lang][x])
+        gender = st.selectbox(LABELS["gender"][lang], [0, 1], format_func=lambda x: gender_options[lang][x])
+        smoke = st.selectbox(LABELS["smoke"][lang], [0, 1], format_func=lambda x: yn_options[lang][x])
+        alco = st.selectbox(LABELS["alco"][lang], [0, 1], format_func=lambda x: yn_options[lang][x])
+        active = st.selectbox(LABELS["active"][lang], [0, 1], format_func=lambda x: yn_options[lang][x])
         input_data = pd.DataFrame([{ "age": age * 365, "height": height, "weight": weight, "ap_hi": systolic, "ap_lo": diastolic, "cholesterol": cholesterol, "gluc": gluc, "gender": gender, "smoke": smoke, "alco": alco, "active": active }])
         predict_clicked = st.button(LABELS["predict"][lang])
         
@@ -151,7 +152,7 @@ if selected == MENU[lang][0]:
             original_text = response.text
 
             if lang == "中文":
-                translated = GoogleTranslator(source="auto", target="zh-tw").translate(original_text)
+                translated = GoogleTranslator(source="auto", target="zh-TW").translate(original_text)
                 st.markdown("### 🔹 AI 健康建議")
                 st.write(translated)
             elif lang == "Malay":