Update app.py

app.py

@@ -34,19 +34,19 @@ if st.button("Diagnosa"):
     if gejala_id and usia and jenis_kelamin:
         # Terjemahkan gejala dari Indonesia ke Inggris
         terjemahan = translator(gejala_id, max_length=100)
-        gejala_en = terjemahan <sup> </sup>["translation_text"]
+        gejala_en = terjemahan[0]["translation_text"]
 
-        informasi_pasien = f"Saya berusia {usia} tahun, {jenis_kelamin}. Gejala yang saya alami adalah {gejala_en}"
+        informasi_pasien = f"I am {usia} years old, {jenis_kelamin}. And my current symptoms are {gejala_en}"
 
         # Pesan untuk model
         pesan = [
-            {"role": "system", "content": "Anda adalah seorang dokter yang perlu mendiagnosis penyakit pasien. Berikan satu diagnosis yang Anda yakini paling akurat."},
-            {"role": "user", "content": f"Berdasarkan informasi ini, {informasi_pasien}, apa penyakit yang mungkin saya derita?"}
+            {"role": "system", "content": "You are a doctor who needs to diagnose a patient's illness. Provide one diagnosis that you believe is most confident."},
+            {"role": "user", "content": f"Based on your assessment, {gabungan}, what illness could it be?"}
         ]
 
         # Fungsi untuk mendapatkan konten dari role 'assistant'
         def get_assistant_content(response):
-            generated_text = response <sup> </sup>['generated_text']
+            generated_text = response [0]['generated_text']
             return generated_text
 
         # Dapatkan respon dari pipe1 (model TinyLlama)
@@ -58,7 +58,7 @@ if st.button("Diagnosa"):
         asisten_konten2 = get_assistant_content(response2)
 
         # Gabungkan hasil dari pipe1 dan pipe2 untuk pertanyaan akhir
-        pertanyaan_akhir = [{'role': 'user', 'content': f"{asisten_konten1}. {asisten_konten2}. Berdasarkan dua kalimat ini, apa kesimpulan akhir Anda tentang gejala saya? Berikan jawaban singkat dengan satu diagnosis."}]
+        pertanyaan_akhir = [{'role': 'user', 'content': f"{asisten_konten1}. {asisten_konten2}. Based on these two sentences, what is your final conclusion of my current symptom? Please provide a brief answer with one diagnosis."}]
 
         # Dapatkan hasil akhir diagnosis
         hasil_diagnosis = pipe1(pertanyaan_akhir)
@@ -66,7 +66,7 @@ if st.button("Diagnosa"):
 
         # Terjemahkan hasil akhir ke bahasa Indonesia
         terjemahan_hasil = terjamah(asisten_konten3, max_length=100)
-        diagnosa_terjemahan = terjemahan_hasil <sup> </sup>["translation_text"]
+        diagnosa_terjemahan = terjemahan_hasil[0]["translation_text"]
 
         # Tampilkan hasil ke Streamlit
         st.subheader("Hasil Diagnosis:")