20

app.py

@@ -111,13 +111,14 @@ def load_css():
         margin: 1rem 0;
     }
     
-    /* Ingredient list styling */
-    .ingredient-item {
+    /* Ingredient list styling - UPDATED */
+    .ingredients-container {
         background: #f8f9fa;
-        padding: 0.5rem 1rem;
-        margin: 0.25rem 0;
-        border-radius: 20px;
-        border-left: 3px solid #667eea;
+        padding: 1rem;
+        border-radius: 10px;
+        border-left: 4px solid #667eea;
+        margin: 1rem 0;
+        line-height: 1.6;
     }
     
     /* Footer */
@@ -143,11 +144,13 @@ def load_model():
     with open("saved_models/XGBoost_model.pkl", "rb") as f:
         return pickle.load(f)
 
-# === Prediksi ===
+# === Prediksi dengan probabilitas ===
 def predict_allergen(model, vectorizer, input_text):
     X_input = vectorizer.transform([input_text])
     prediction = model.predict(X_input)
-    return prediction[0]
+    # Mendapatkan probabilitas prediksi
+    probabilities = model.predict_proba(X_input)
+    return prediction[0], probabilities[0]
 
 # === Scraping bahan dari Cookpad ===
 def get_ingredients_from_cookpad(url):
@@ -174,30 +177,36 @@ def get_ingredients_from_cookpad(url):
     except Exception as e:
         return None, f"Terjadi kesalahan: {str(e)}"
 
-# === Display results with custom styling ===
-def display_results(results):
+# === Display results with custom styling and percentage ===
+def display_results(results, probabilities):
     st.markdown("### 🎯 Hasil Analisis Alergen")
     
     positive_results = []
     negative_results = []
     
-    for allergen, status in results.items():
+    labels = ['Susu', 'Kacang', 'Telur', 'Makanan Laut', 'Gandum']
+    
+    for i, (allergen, status) in enumerate(results.items()):
+        # Mendapatkan probabilitas untuk kelas positif (index 1)
+        positive_prob = probabilities[i][1] * 100 if len(probabilities[i]) > 1 else 0
+        
         if status == 1:
-            positive_results.append(allergen)
+            positive_results.append((allergen, positive_prob))
         else:
-            negative_results.append(allergen)
+            negative_results.append((allergen, positive_prob))
     
     # Display positive results (allergens detected)
     if positive_results:
         st.markdown("#### ⚠️ **Alergen Terdeteksi:**")
-        for allergen in positive_results:
-            st.markdown(f'<div class="result-positive">🚨 <strong>{allergen}</strong></div>', unsafe_allow_html=True)
+        for allergen, prob in positive_results:
+            st.markdown(f'<div class="result-positive">🚨 <strong>{allergen}</strong> - Tingkat Kepercayaan: {prob:.1f}%</div>', unsafe_allow_html=True)
     
     # Display negative results (safe allergens)
     if negative_results:
         st.markdown("#### ✅ **Aman dari Alergen:**")
-        for allergen in negative_results:
-            st.markdown(f'<div class="result-negative">✓ {allergen}</div>', unsafe_allow_html=True)
+        for allergen, prob in negative_results:
+            confidence = 100 - prob  # Confidence untuk kelas negatif
+            st.markdown(f'<div class="result-negative">✓ {allergen} - Tingkat Kepercayaan: {confidence:.1f}%</div>', unsafe_allow_html=True)
     
     # Show summary
     if not positive_results:
@@ -290,11 +299,11 @@ def main():
                         st.warning("⚠️ Mohon masukkan bahan makanan terlebih dahulu.")
                     else:
                         with st.spinner("🔄 Sedang menganalisis..."):
-                            pred = predict_allergen(model, vectorizer, input_text)
+                            pred, probabilities = predict_allergen(model, vectorizer, input_text)
                             results = dict(zip(labels, pred))
                         
                         st.success("✅ Analisis selesai!")
-                        display_results(results)
+                        display_results(results, probabilities)
         
         elif input_mode == "🔗 URL Cookpad":
             st.markdown("### 🔗 Analisis dari URL Cookpad")
@@ -345,18 +354,18 @@ def main():
                             else:
                                 st.success("✅ Bahan berhasil diambil!")
                                 
-                                # Display ingredients in a nice format
+                                # Display ingredients in a single container - UPDATED
                                 st.markdown("**🧾 Daftar Bahan:**")
-                                for ing in ingredients:
-                                    st.markdown(f'<div class="ingredient-item">• {ing}</div>', unsafe_allow_html=True)
+                                ingredients_text = ", ".join(ingredients)
+                                st.markdown(f'<div class="ingredients-container">{ingredients_text}</div>', unsafe_allow_html=True)
                                 
                                 # Predict allergens
                                 joined_ingredients = " ".join(ingredients)
-                                pred = predict_allergen(model, vectorizer, joined_ingredients)
+                                pred, probabilities = predict_allergen(model, vectorizer, joined_ingredients)
                                 results = dict(zip(labels, pred))
                                 
                                 st.markdown("---")
-                                display_results(results)
+                                display_results(results, probabilities)
                     
                     # Clear progress indicators
                     progress_bar.empty()