Create app.py

app.py

@@ -0,0 +1,59 @@
+import streamlit as st
+from transformers import pipeline
+import nltk
+from nltk.tokenize import word_tokenize
+from nltk.corpus import stopwords
+
+nltk.download('punkt')
+nltk.download("stopwords")
+
+# Buat objek terjemahan
+translator = pipeline("translation_id_to_en", model="Helsinki-NLP/Opus-MT-ID-EN")
+
+# Menggunakan pipa untuk klasifikasi diagnosis penyakit
+disease_classifier = pipeline("text-classification", model="BenK10/disease-diagnosis")
+
+def preprocess_text(teks):
+    tokens = word_tokenize(teks)
+    tokens = [token for token in tokens if token.isalnum()]
+    tokens = [token for token in tokens if token.lower() not in stopwords.words("indonesian")]
+    preprocessed_text = " ".join(tokens)
+    return preprocessed_text
+
+def predict_disease(gejala, usia, jenis_kelamin):
+    # Pra-pemrosesan teks gejala
+    gejala_diproses = preprocess_text(gejala)
+
+    # Melakukan terjemahan
+    terjemahan = translator(gejala_diproses, max_length=50)
+    terjemahan_inggris = terjemahan[0]["translation_text"]
+
+    # Membuat fitur tambahan untuk klasifikasi
+    fitur = {"usia": usia, "jenis_kelamin": jenis_kelamin}
+
+    # Klasifikasi diagnosis penyakit dengan fitur tambahan
+    klasifikasi = disease_classifier(terjemahan_inggris, features=fitur)
+
+    # Mengembalikan diagnosis dan skor kepercayaan
+    diagnosis = klasifikasi[0]["label"]
+    kepercayaan = klasifikasi[0]["score"]
+    return diagnosis, kepercayaan
+
+# Halaman Streamlit
+st.title("Diagnosis Penyakit")
+
+# Masukan gejala
+gejala = st.text_area("Masukkan gejala Anda:", "")
+
+# Masukan usia
+usia = st.number_input("Masukkan usia Anda:", min_value=0, max_value=120)
+
+# Masukan jenis kelamin
+jenis_kelamin_pilihan = ["Laki-laki", "Perempuan"]
+jenis_kelamin = st.selectbox("Masukkan jenis kelamin Anda:", jenis_kelamin_pilihan)
+
+# Memanggil fungsi prediksi
+if st.button("Diagnosis"):
+    diagnosis, kepercayaan = predict_disease(gejala, usia, jenis_kelamin)
+    st.success(f"Diagnosis: {diagnosis}")
+    st.write(f"Kepercayaan: {kepercayaan:.2f}")