Update app.py

app.py

@@ -5,6 +5,7 @@ import google.generativeai as genai
 import os
 from deep_translator import GoogleTranslator
 
+translator = GoogleTranslator()
 
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
@@ -54,69 +55,62 @@ with st.sidebar:
 
 # Label dictionary for multilingual UI
 LABELS = {
+    "user_type": {"English": "I am a...", "中文": "我的身份是...", "Malay": "Saya seorang..."},
+    "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)"},
     "weight": {"English": "Patient's Weight (kg)", "中文": "體重 (kg)", "Malay": "Berat (kg)"},
     "systolic": {"English": "Patient's Systolic BP", "中文": "收縮壓", "Malay": "Tekanan Sistolik"},
-    "diastolic": {"English": "Patient's Diastolic BP", "中文": "舒张壓", "Malay": "Tekanan Diastolik"},
+    "diastolic": {"English": "Patient's Diastolic BP", "中文": "舒張壓", "Malay": "Tekanan Diastolik"},
     "cholesterol": {"English": "Patient's Cholesterol", "中文": "膽固醇", "Malay": "Kolesterol"},
     "gluc": {"English": "Patient's Glucose", "中文": "血糖", "Malay": "Glukosa"},
     "gender": {"English": "Patient's Gender", "中文": "性別", "Malay": "Jantina"},
-    "smoke": {"English": "Do patient smoke?", "中文": "是否吸煙？", "Malay": "Adakah merokok?"},
+    "smoke": {"English": "Do patient smoke?", "中文": "是否吸菸？", "Malay": "Adakah merokok?"},
     "alco": {"English": "Do patient drink alcohol?", "中文": "是否喝酒？", "Malay": "Adakah minum alkohol?"},
-    "active": {"English": "Are patient physically active?", "中文": "是否練習運動？", "Malay": "Adakah aktif secara fizikal?"},
+    "active": {"English": "Are patient physically active?", "中文": "是否有運動習慣？", "Malay": "Adakah aktif secara fizikal?"},
     "predict": {"English": "Predict CVD Risk", "中文": "預測心血管風險", "Malay": "Ramalkan Risiko CVD"}
 }
 
+MENU = {
+    "English": ["Prediction Tool", "Let's know more about CVD!"],
+    "中文": ["預測工具", "了解更多心血管資訊"],
+    "Malay": ["Alat Ramalan", "Ketahui lebih lanjut tentang CVD"]
+}
+
 
 # User input
 with st.sidebar:
     selected = option_menu(
         menu_title="Navigation",
-        options=["Prediction Tool", "Let's know more about CVD!"],
+        options=MENU[lang],
         icons=["activity", "book"],
         menu_icon="cast",
         default_index=0,
     )
 
-if selected == "Prediction Tool":
-
+if selected == MENU[lang][0]:
     with st.sidebar:
-        st.header("Enter Health Info")
-        user_type = st.selectbox("I am a...", ["Patient", "Medical Professional"])
-        age = st.number_input("Patient's Age", min_value=18, max_value=100, value=50)
-        height = st.number_input("Patient's Height (cm)", min_value=100, max_value=250, value=170)
-        weight = st.number_input("Patient's Weight (kg)", min_value=30, max_value=200, value=70)
-        systolic = st.number_input("Patient's Systolic BP", min_value=80, max_value=250, value=120)
-        diastolic = st.number_input("Patient's Diastolic BP", min_value=40, max_value=150, value=80)
-        cholesterol = st.selectbox("Patient's Cholesterol", [0, 1, 2], format_func=lambda x: {0: "Normal", 1: "High", 2: "Very High"}[x])
-        gluc = st.selectbox("Patient's Glucose", [0, 1, 2], format_func=lambda x: {0: "Normal", 1: "High", 2: "Very High"}[x])
-        gender = st.selectbox("Patient's Gender", [0, 1], format_func=lambda x: {0: "Female", 1: "Male"}[x])
-        smoke = st.selectbox("Do patient smoke?", [0, 1], format_func=lambda x: "Yes" if x else "No")
-        alco = st.selectbox("Do patient drink alcohol?", [0, 1], format_func=lambda x: "Yes" if x else "No")
-        active = st.selectbox("Are patient physically active?", [0, 1], format_func=lambda x: "Yes" if x else "No")
-
-        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("Predict CVD Risk")
-
+        st.header(LABELS["input_header"][lang])
+        user_type = st.selectbox(LABELS["user_type"][lang], ["Patient", "Medical Professional"])
+        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")
+        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])
+        
     if predict_clicked:
         input_data["bmi"] = input_data["weight"] / ((input_data["height"]/100)**2)
         input_data["pulse_pressure"] = input_data["ap_hi"] - input_data["ap_lo"]
         input_data["hypertension"] = ((input_data["ap_hi"] > 140) | (input_data["ap_lo"] > 90)).astype(int)
         
-       
 
         preds = model.predict(input_data)
         proba = preds["cardio_1_probability"].iloc[0]
@@ -158,11 +152,11 @@ if selected == "Prediction Tool":
 
             if lang == "中文":
                 translated = translator.translate(original_text, dest="zh-tw").text
-                st.markdown("### 🔹 AI 健康建議（繁體中文）")
+                st.markdown("### 🔹 AI 健康建議")
                 st.write(translated)
             elif lang == "Malay":
                 translated = translator.translate(original_text, dest="ms").text
-                st.markdown("### 🔹 Nasihat Kesihatan AI (Bahasa Melayu)")
+                st.markdown("### 🔹 Nasihat Kesihatan AI")
                 st.write(translated)
             else:
                 st.markdown("### 🔹 AI Health Advice")