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 ensemble verdict
-    st.markdown("### Ensemble Verdict")
-    vote_benign = results['vote_count'][0]  # Votes for safe
-    vote_malicious = results['vote_count'][1]  # Votes for malicious
+    # Final verdict combining both approaches
+    st.subheader("Final Analysis")
     
-    # 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.")
+    is_malicious_regex, _ = st.session_state.regex_result
+    is_malicious_ensemble = results['ensemble'] == 1
+    
+    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.")
     
     # Reset button
     if st.button("Analyze Another Query"):