Update app.py

app.py

@@ -3,6 +3,9 @@ from pytorch_tabular.tabular_datamodule import TabularDatamodule
 from streamlit_option_menu import option_menu
 import google.generativeai as genai
 import os
+from googletrans import Translator
+
+translator = Translator()
 
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
@@ -46,6 +49,26 @@ dm._inferred_config = SimpleNamespace(output_cardinality=[2])
 with open("FTTransformerModel/threshold.json", "r") as f:
     threshold = json.load(f)["threshold"]
 
+# Sidebar: language selection
+with st.sidebar:
+    lang = st.selectbox("🌐 Language / 語言 / Bahasa", ["English", "中文", "Malay"])
+
+# Label dictionary for multilingual UI
+LABELS = {
+    "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"},
+    "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?"},
+    "alco": {"English": "Do patient drink alcohol?", "中文": "是否喝酒？", "Malay": "Adakah minum alkohol?"},
+    "active": {"English": "Are patient physically active?", "中文": "是否練習運動？", "Malay": "Adakah aktif secara fizikal?"},
+    "predict": {"English": "Predict CVD Risk", "中文": "預測心血管風險", "Malay": "Ramalkan Risiko CVD"}
+}
+
 
 # User input
 with st.sidebar:
@@ -129,14 +152,22 @@ if selected == "Prediction Tool":
 
             model_gemini = genai.GenerativeModel(
                 model_name="models/gemini-2.0-flash",
-                generation_config=genai.types.GenerationConfig(
-                    temperature=0.2
-                )
+                generation_config=genai.types.GenerationConfig(temperature=0.2)
             )
             response = model_gemini.generate_content(prompt)
-
-            st.markdown("### 💡 AI Health Advice")
-            st.write(response.text)
+            original_text = response.text
+
+            if lang == "中文":
+                translated = translator.translate(original_text, dest="zh-tw").text
+                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.write(translated)
+            else:
+                st.markdown("### 🔹 AI Health Advice")
+                st.write(original_text)