Update app.py

app.py

@@ -12,7 +12,7 @@ translator = pipeline("translation_id_to_en", model="Helsinki-NLP/Opus-MT-ID-EN"
 terjemahkan = pipeline("translation", model="Helsinki-NLP/opus-mt-en-id")
 
 # Menggunakan pipa untuk klasifikasi diagnosis penyakit
-disease_classifier = pipeline("text-classification", model="BenK10/disease-diagnosis")
+disease_classifier = pipeline("text-classification", model="Zabihin/Symptom_to_Diagnosis")
 
 text_generation = pipeline("text-generation", model="SumayyaAli/tiny-llama-1.1b-chat-medical")
 
@@ -26,28 +26,32 @@ def preprocess_text(teks):
     return preprocessed_text
 
 def predict_disease(gejala):
-    # Pra-pemrosesan teks gejala
-    gejala_diproses = preprocess_text(gejala)
-
-    # Melakukan terjemahan
-    terjemahan = translator(gejala_diproses, max_length=100)
-    terjemahan_inggris = terjemahan[0]["translation_text"]
-
-    # Klasifikasi diagnosis penyakit dengan fitur tambahan
-    klasifikasi = disease_classifier(terjemahan_inggris)
-
-    # Mengembalikan diagnosis dan skor kepercayaan
-    diagnosis = klasifikasi[0]["label"]
-    kepercayaan = klasifikasi[0]["score"]
-    definisi = f"What is {diagnosis}?"
-    definisi = text_generation(definisi, max_length=100, num_return_sequences=1)
-    definisi = definisi[0]['generated_text']
-    # Melakukan terjemahan
-    terjemahan_definisi = terjemahkan(definisi, max_length=100)
-    definisi = terjemahan_definisi[0]["translation_text"]
-    
-    return diagnosis, kepercayaan, definisi
+    try:
+        # Pra-pemrosesan teks gejala
+        gejala_diproses = preprocess_text(gejala)
+
+        # Melakukan terjemahan
+        terjemahan = translator(gejala_diproses, max_length=100)
+        terjemahan_inggris = terjemahan[0]["translation_text"]
+
+        # Klasifikasi diagnosis penyakit dengan fitur tambahan
+        klasifikasi = disease_classifier(terjemahan_inggris)
+
+        # Mengembalikan diagnosis dan skor kepercayaan
+        diagnosis = klasifikasi[0]["label"]
+        kepercayaan = klasifikasi[0]["score"]
+        definisi = f"What is {diagnosis}?"
+        definisi = text_generation(definisi, max_length=100, num_return_sequences=1)
+        definisi = definisi[0]['generated_text']
+        # Melakukan terjemahan
+        terjemahan_definisi = terjemahkan(definisi, max_length=100)
+        definisi = terjemahan_definisi[0]["translation_text"]
+        
+        return diagnosis, kepercayaan, definisi
     
+    except Exception as e:
+        return "Tidak dapat mendiagnosis", 0.0, "Terjadi kesalahan saat mendiagnosis penyakit."
+
 # Halaman Streamlit
 st.title("Diagnosis Penyakit")
 
@@ -69,8 +73,11 @@ with st.form("diagnosis_form"):
     if submitted:
         if gejala and usia and jenis_kelamin:
             diagnosis, kepercayaan, definisi = predict_disease(gejala)
-            st.success(f"Diagnosis: {diagnosis}")
-            st.write(f"Dengan tingkat akurasi: {kepercayaan:.2f}")
-            st.write(f"{definisi}")
+            if diagnosis != "Tidak dapat mendiagnosis":
+                st.success(f"Diagnosis: {diagnosis}")
+                st.write(f"Dengan tingkat akurasi: {kepercayaan:.2f}")
+                st.write(f"{definisi}")
+            else:
+                st.error(definisi)
         else:
             st.error("Harap isi semua kolom sebelum menekan tombol Diagnosis.")