feat: refactor app

src/streamlit_app.py

@@ -8,25 +8,21 @@ SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
 MODEL_PATH = os.path.join(SCRIPT_DIR, "best.pt")
 
 SYMPTOM_QUESTIONS = {
-    "G1": "Daun berwana putih kecoklatan memanjang seperti mengering pada bagian tepi daun?",
+    "G1": "Daun berwana putih kecoklatan memanjang seperti mengering pada bagian tepi daun",
     "G2": "Pada pagi hari, dapat ditemukan cairan bakteri terlihat seperti butiran air pada bagian terinfeksi",
-    "G3": "Gejala terlihat seperti anak panah terdapat di antara urat daun?",
-    "G4": "Daun Berwarna kuning kecoklatan?",
-    "G5": "Luka pada daun terlihat transparan bila dihadapkan cahaya matahari?",
-    "G6": "Daun berbentuk belah ketupat?",
-    "G7": "Pada daun bagian tengah berwarna abu-abu atau berwarna putih dengan bagian tepi berwarna kecoklatan?",
-    "G8": "Infeksi pada malai/leher berwarna abu-abu?",
-    "G9": "Tanaman kerdil dan Gejala pada daun terjadi perubahan warna dari hijau menjadi jingga atau kemerahan?",
-    "G10": "Daun menguning, menggulung, mengering dan menjadi layu?",
-    "G11": "Bibit menjadi layu (kresek) tapi sulit dicabut?",
-    "G12": "Warna luka bercak menjadi jingga kekuningan dari ujung daun ke pangkal daun?",
-    "G13": "Ada bulatan kecil berwarna kuning pada pelepah daun?",
-    "G14": "Bercak cokelat pada daun, pelepah, dan bulir gabah?",
-    "G15": "Bercak daun sering menutupi permukaan daun, menyebabkan daun menjadi layu?",
-    "G16": "Bercak cokelat pada batang/leher malai?",
-    "G17": "Terdapat bercak berbentuk oval berwarna abu-abu kehijauan pada pelepah (tulang daun)?",
-    "G18": "Batang tanaman padi menjadi rapuh dan mudah rebah/jatuh?",
-    "G19": "Bibit yang sakit menjadi layu dan akhirnya mati?"
+    "G3": "Gejala terlihat seperti anak panah terdapat di antara urat daun",
+    "G4": "Daun Berwarna kuning kecoklatan",
+    "G5": "Luka pada daun terlihat transparan bila dihadapkan cahaya matahari",
+    "G6": "Daun berbentuk belah ketupat",
+    "G7": "Pada daun bagian tengah berwarna abu-abu atau berwarna putih dengan bagian tepi berwarna kecoklatan",
+    "G8": "Infeksi pada malai/leher berwarna abu-abu",
+    "G9": "Tanaman kerdil dan Gejala pada daun terjadi perubahan warna dari hijau menjadi jingga atau kemerahan",
+    "G10": "Bercak cokelat pada daun, pelepah, dan bulir gabah",
+    "G11": "Bercak daun sering menutupi permukaan daun, menyebabkan daun menjadi layu",
+    "G12": "Bercak cokelat pada batang/leher malai",
+    "G13": "Terdapat bercak berbentuk oval berwarna abu-abu kehijauan pada pelepah (tulang daun)",
+    "G14": "Batang tanaman padi menjadi rapuh dan mudah rebah/jatuh",
+    "G15": "Bibit yang sakit menjadi layu dan akhirnya mati"
 }
 
 YOLO_PROMPTS = {
@@ -34,7 +30,7 @@ YOLO_PROMPTS = {
     "bacterial_leaf_streak": ["G4", "G5"],
     "blast": ["G6", "G7", "G8"], 
     "tungro": ["G9"],
-    "brown_spot": ["G14", "G15", "G16"], 
+    "brown_spot": ["G10", "G11", "G12"], 
     "downy_mildew": [],
     "dead_heart": [],
     "hispa": [],
@@ -42,13 +38,42 @@ YOLO_PROMPTS = {
 }
 
 PRODUCTION_RULES = [
-    {"disease": "Bacterial Leaf Blight (Hawar Daun Bakteri)", "symptoms": ["G1", "G2", "G3"], "source": "Literatur 1"},
-    {"disease": "Bakteri Daun Bergaris (Bacterial Leaf Streak)", "symptoms": ["G4", "G5"], "source": "Literatur 1"},
-    {"disease": "Blas (Blast)", "symptoms": ["G6", "G7", "G8"], "source": "Literatur 1"},
-    {"disease": "Tungro", "symptoms": ["G9"], "source": "Literatur 1"},
-    {"disease": "Hawar Daun Bakteri (Bacterial Leaf Blight)", "symptoms": ["G10", "G11", "G12", "G13"], "source": "Literatur 2"},
-    {"disease": "Bercak Daun Coklat (Brown Spot)", "symptoms": ["G14", "G15", "G16"], "source": "Literatur 3"},
-    {"disease": "Hawar Pelepah (Sheath Blight)", "symptoms": ["G17", "G18", "G19"], "source": "Literatur 3"},
+    {
+        "yolo_class": "bacterial_leaf_blight", 
+        "disease": "Bacterial Leaf Blight (Hawar Daun Bakteri)", 
+        "symptoms": ["G1", "G2", "G3"], 
+        "source": "Literatur 1"
+    },
+    {
+        "yolo_class": "bacterial_leaf_streak",
+        "disease": "Bakteri Daun Bergaris (Bacterial Leaf Streak)", 
+        "symptoms": ["G4", "G5"], 
+        "source": "Literatur 1"
+    },
+    {
+        "yolo_class": "blast",
+        "disease": "Blas (Blast)", 
+        "symptoms": ["G6", "G7", "G8"], 
+        "source": "Literatur 1"
+    },
+    {
+        "yolo_class": "tungro",
+        "disease": "Tungro", 
+        "symptoms": ["G9"], 
+        "source": "Literatur 1"
+    },
+    {
+        "yolo_class": "brown_spot",
+        "disease": "Bercak Daun Coklat (Brown Spot)", 
+        "symptoms": ["G10", "G11", "G12"], 
+        "source": "Literatur 3"
+    },
+    {
+        "yolo_class": None,
+        "disease": "Hawar Pelepah (Sheath Blight)", 
+        "symptoms": ["G13", "G14", "G15"], 
+        "source": "Literatur 3"
+    },
 ]
 
 @st.cache_resource
@@ -61,65 +86,84 @@ def load_model():
 
 model = load_model()
 
-st.title("🌾 Rice Doctor: Sistem Pakar Hibrida")
+st.title("Rice Doctor: Sistem Pakar Penyakit Padi")
 st.markdown("""
-Sistem ini menggunakan metode **Hybrid AI**:
-1. **Model AI (YOLO)** untuk memberikan hipotesis awal.
-2. **Backward Chaining** untuk memverifikasi hipotesis tersebut.
-3. **Forward Chaining** untuk mencari diagnosis alternatif jika hipotesis salah.
+Dalam sistem **Rice Doctor**, pengguna dapat:
+1. **Unggah Gambar Padi**. Untuk mendeteksi penyakit padi.
+2. **Verifikasi Gejala Penyakit Padi**. Untuk memverifikasi hasil deteksi penyakit padi.
+3. **Pengecekan Gejala Penyakit Padi Manual**. Jika hasil deteksi penyakit padi sebelummnya tidak terverifikasi.
 """)
 
 if 'step' not in st.session_state:
     st.session_state.step = 1
 if 'prediction' not in st.session_state:
     st.session_state.prediction = None
+if 'forward_matches' not in st.session_state:
+    st.session_state.forward_matches = []
 if 'hypothesis_codes' not in st.session_state:
     st.session_state.hypothesis_codes = []
 if 'user_facts' not in st.session_state:
     st.session_state.user_facts = set()
+if 'system_method' not in st.session_state:
+    st.session_state.system_method = None
 
 if st.session_state.step == 1:
-    uploaded_file = st.file_uploader("Langkah 1: Unggah Gambar Daun", type=["jpg", "png", "jpeg"])
+    st.divider()
+    st.subheader("Langkah 1: Unggah Gambar Padi")
+    uploaded_file = st.file_uploader(type=["jpg", "png", "jpeg"])
 
     if uploaded_file is not None:
         image = Image.open(uploaded_file)
-        st.image(image, caption="Gambar yang Diunggah", width=300)
+        col1, col2, col3 = st.columns([1, 2, 1])
+        with col2: 
+            st.image(image, caption="Gambar yang Diunggah", width=300)
         
-        if st.button("Analisis Gambar (AI Inference)", type="primary"):
+        if st.button("Deteksi Penyakit Padi", type="primary"):
             if model:
-                with st.spinner("Model AI sedang menganalisis hipotesis..."):
+                with st.spinner("Model AI sedang menganalisis..."):
                     results = model.predict(image, imgsz=640)
+                    
                     raw_name = results[0].names[results[0].probs.top1]
                     top_class_name = raw_name.lower().strip()
+                    probs = results[0].probs
+                    top_conf = probs.top1conf.item()
                     
+                    st.session_state.image = image
                     st.session_state.prediction = top_class_name
+                    st.session_state.prediction_conf = top_conf
                     st.session_state.hypothesis_codes = YOLO_PROMPTS.get(top_class_name, [])
                     st.session_state.step = 2 
                     st.rerun()
 
 elif st.session_state.step == 2:
     st.divider()
-    st.subheader("Langkah 2: Backward Chaining (Verifikasi Hipotesis)")
+    st.subheader("Langkah 2: Verifikasi Gejala Penyakit Padi")
     
+    image = st.session_state.image
     pred = st.session_state.prediction
+    conf = st.session_state.prediction_conf
+    conf_threshold = 0.9
     codes = st.session_state.hypothesis_codes
     
-    st.info(f"🔍 **Hipotesis AI:** `{pred}`")
-    st.write("Sistem sedang memverifikasi hipotesis ini menggunakan **Backward Chaining**.")
+    st.info(f"🔍 **Hasil Deteksi Penyakit Padi:** {pred}")
     
-    if not codes:
-        st.warning("Model AI memprediksi kelas yang tidak memiliki aturan verifikasi spesifik atau 'Normal'.")
-        if st.button("Lanjut ke Pengecekan Manual (Forward Chaining)"):
+    if not codes or (conf < conf_threshold):
+        st.warning("Model AI tidak dapat memprediksi penyakit padi dengan keyakinan tinggi.")
+        if st.button("Lanjut ke Pengecekan Manual"):
             st.session_state.step = 3
             st.rerun()
     else:
-        st.write("Jawablah pertanyaan berikut untuk memverifikasi hipotesis:")
+        col1, col2, col3 = st.columns([1, 2, 1])
+        with col2:
+            st.image(image, caption="Gambar yang Diunggah", width=300)
+
+        st.write("Jawablah pertanyaan berikut untuk memverifikasi penyakit:")
         
         bc_answers = {}
         for code in codes:
             question = SYMPTOM_QUESTIONS.get(code, "Unknown")
             ans = st.radio(
-                f"**({code})** {question}", 
+                f"**({code})** {question}?", 
                 options=["Tidak", "Ya"], 
                 index=0, 
                 key=f"bc_{code}"
@@ -129,10 +173,11 @@ elif st.session_state.step == 2:
             else:
                 bc_answers[code] = False
         
-        if st.button("Verifikasi Hipotesis", type="primary"):
+        if st.button("Verifikasi Penyakit Padi", type="primary"):
             all_yes = all(bc_answers.values())
             
             if all_yes:
+                st.session_state.system_method = "backward"
                 st.session_state.step = 4
                 st.rerun()
             else:
@@ -142,9 +187,7 @@ elif st.session_state.step == 2:
 
 elif st.session_state.step == 3:
     st.divider()
-    st.subheader("Langkah 3: Forward Chaining (Pencarian Fakta)")
-    st.error("❌ **Hipotesis AI Ditolak**")
-    st.warning("Karena hipotesis awal ditolak (ada gejala yang tidak cocok), sistem beralih ke **Forward Chaining** untuk mencari diagnosis alternatif.")
+    st.subheader("Langkah 3: Pengecekan Gejala Penyakit Padi Manual")
     
     all_symptoms_list = [f"{k}: {v}" for k,v in SYMPTOM_QUESTIONS.items()]
     
@@ -156,7 +199,7 @@ elif st.session_state.step == 3:
         default=pre_selected
     )
     
-    if st.button("Proses Diagnosis (Forward Chaining)", type="primary"):
+    if st.button("Analisis Penyakit Padi", type="primary"):
         final_facts = set()
         for item in other_checks:
             code = item.split(":")[0].strip()
@@ -172,27 +215,35 @@ elif st.session_state.step == 3:
                     matches.append(rule)
             
             if matches:
-                st.success("✅ **Hasil Forward Chaining Ditemukan!**")
-                for m in matches:
-                    st.markdown(f"""
-                    ---
-                    ### {m['disease']}
-                    - **Sumber Aturan:** {m['source']}
-                    - **Gejala Terpenuhi:** {', '.join(m['symptoms'])}
-                    """)
+                st.session_state.system_method = "forward"
+                st.session_state.forward_matches = matches
+                st.session_state.step = 4
+                st.rerun()
+
             else:
-                st.info("ℹ️ **Tidak Ditemukan Diagnosis Pasti**")
+                st.info("ℹ️ **Sistem Tidak Menemukan Penyakit Padi Secara Pasti**")
                 st.write("Kombinasi gejala yang Anda pilih tidak cocok sepenuhnya dengan aturan penyakit manapun dalam basis pengetahuan.")
-    
-    if st.button("Mulai Ulang Analisis"):
-        st.session_state.step = 1
-        st.session_state.user_facts = set()
-        st.rerun()
 
 elif st.session_state.step == 4:
     st.divider()
-    st.success(f"✅ **Diagnosis Terkonfirmasi: {st.session_state.prediction}**")
-    st.markdown(f"Hipotesis Backward Chaining bernilai **BENAR**. Semua gejala untuk **{st.session_state.prediction}** telah terpenuhi.")
+    st.success("**✅ Penyakit Padi Ditemukan!**")
+
+    if st.session_state.system_method == "forward":
+        for m in st.session_state.forward_matches:
+            st.markdown(f"""
+            ### {m['disease']}
+            - **Sumber Aturan:** {m['source']}
+            """)
+
+    if st.session_state.system_method == "backward":
+        matched_rule = next((r for r in PRODUCTION_RULES if r.get("yolo_class") == st.session_state.prediction), None)
+        disease = matched_rule['disease'] if matched_rule else st.session_state.prediction
+        source = matched_rule['source'] if matched_rule else "Tidak diketahui"
+
+        st.markdown(f"""
+            ### {disease}
+            - **Sumber Aturan:** {source}
+        """)
     
     if st.button("Mulai Ulang Analisis"):
         st.session_state.step = 1