Update app.py

app.py

@@ -438,18 +438,18 @@ if st.session_state.analysis_stage >= 2 and st.session_state.ensemble_result is
             - No significant red flags were detected in the ensemble
             """)
     
-    # Final verdict combining both approaches
-    st.subheader("Final Analysis")
-    
-    is_malicious_regex, _ = st.session_state.regex_result
-    is_malicious_ensemble = results['ensemble'] == 1
+    # Final ensemble verdict
+    st.markdown("### Ensemble Verdict")
+    vote_benign = results['vote_count'][0]  # Votes for safe
+    vote_malicious = results['vote_count'][1]  # Votes for malicious
     
-    if is_malicious_regex or is_malicious_ensemble:
-        st.error("⚠️ This query appears to contain SQL injection patterns. Review carefully before executing.")
-    else:
-        st.success("✅ This query appears safe based on both rule-based and ensemble detection.")
-            
-    st.info("ℹ️ Remember: Always use parameterized queries and proper input validation in production systems.")
+    # Decision-making based on votes
+    if vote_benign > 3:  # Safe votes must be greater than 3
+        st.success("✅ Query deemed safe by majority vote")
+    elif vote_malicious > 3:  # Malicious votes greater than 3
+        st.error("🚨 SQL Injection Detected by Majority Vote!")
+    else:  # Tie or ambiguous case (e.g., 2 safe, 2 malicious)
+    st.warning("⚠️ Ambiguous result: The votes are evenly split. Please cross-check the query manually.")
     
     # Reset button
     if st.button("Analyze Another Query"):