Update app.py

app.py

@@ -264,23 +264,48 @@ if st.session_state.analysis_stage >= 2 and st.session_state.ensemble_result is
     vote_ratio = vote_malicious / (vote_benign + vote_malicious)
     st.progress(vote_ratio, text=f"Malicious vote ratio: {vote_ratio*100:.0f}%")
 
-    # Individual model results
+    # Display individual model results
     st.markdown("### Individual Model Results")
+    
     model_cols = st.columns(4)
+    
     with model_cols[0]:
         st.markdown("**Random Forest**")
-        st.error("⚠️ Malicious") if results['rf'] == 1 else st.success("✅ Safe")
+        if results['rf'] == 1:
+            st.error("⚠️ Malicious")
+        else:
+            st.success("✅ Safe")
+    
     with model_cols[1]:
         st.markdown("**SVM**")
-        st.error("⚠️ Malicious") if results['svm'] == 1 else st.success("✅ Safe")
+        if results['svm'] == 1:
+            st.error("⚠️ Malicious")
+        else:
+            st.success("✅ Safe")
+    
     with model_cols[2]:
         st.markdown("**CNN**")
         cnn_prob = results['cnn']['probability'] * 100
-        st.error(f"⚠️ Malicious ({cnn_prob:.1f}%)") if results['cnn']['prediction'] == 1 else st.success(f"✅ Safe ({100-cnn_prob:.1f}%)")
+        if results['cnn']['prediction'] == 1:
+            st.error(f"⚠️ Malicious ({cnn_prob:.1f}%)")
+        else:
+            st.success(f"✅ Safe ({100-cnn_prob:.1f}%)")
+    
     with model_cols[3]:
         st.markdown("**LSTM**")
         lstm_prob = results['lstm']['probability'] * 100
-        st.error(f"⚠️ Malicious ({lstm_prob:.1f}%)") if results['lstm']['prediction'] == 1 else st.success(f"✅ Safe ({100-lstm_prob:.1f}%)")
+        if results['lstm']['prediction'] == 1:
+            st.error(f"⚠️ Malicious ({lstm_prob:.1f}%)")
+        else:
+            st.success(f"✅ Safe ({100-lstm_prob:.1f}%)")
+    
+    # Final ensemble verdict
+    st.markdown("### Ensemble Verdict")
+    if results['ensemble'] == 1:
+        st.error("🚨 SQL Injection Detected by Majority Vote!")
+    else:
+        st.success("✅ Query deemed safe by majority vote")
+    
 
     # Final ensemble verdict
     st.markdown("### Ensemble Verdict")