Update app.py

app.py

@@ -120,32 +120,43 @@ if uploaded_py is not None:
         st.markdown("### 📄 Code Preview")
         st.code(code_text, language="python")
 
-        # Extract approximate metrics, predict
+            # Extract approximate metrics from code and get ML prediction
         metrics_vector = extract_simple_metrics_from_code(code_text)
         metrics_array = np.array(metrics_vector).reshape(1, -1)
 
-        # Scale and predict
         scaled = scaler.transform(metrics_array)
-        pred = model.predict(scaled)[0]
+        ml_pred = model.predict(scaled)[0]
 
         if hasattr(model, "predict_proba"):
-            proba = model.predict_proba(scaled)[0][1]
+            ml_proba = model.predict_proba(scaled)[0][1]
         else:
-            proba = None
+            ml_proba = None
+
+        # Rule-based suspicious lines
+        suspicious = find_suspicious_lines(code_text)
+
+        # 🔴 HYBRID DECISION:
+        # If ML says defect OR we found suspicious lines → treat as defect
+        is_defect = (ml_pred == 1) or (len(suspicious) > 0)
 
         st.markdown("### 🔍 File-level Defect Prediction")
-        if pred == 1:
+        if is_defect:
             st.error("⚠️ Defect Likely")
         else:
             st.success("✅ No Defect Predicted")
 
-        if proba is not None:
-            st.write(f"Estimated probability of defect: **{proba:.2f}**")
+        if ml_proba is not None:
+            st.write(f"Estimated probability from ML model: **{ml_proba:.2f}**")
 
         # Suspicious lines section
         st.markdown("### 🧷 Suspicious Lines (Rule-Based Heuristics)")
-        suspicious = find_suspicious_lines(code_text)
-
+        if not suspicious:
+            st.info("No suspicious patterns found by the simple rules.")
+        else:
+            for line_no, line_text, reason in suspicious:
+                st.write(f"**Line {line_no}** — {reason}")
+                st.code(line_text, language="python")
+   
         if not suspicious:
             st.info("No suspicious patterns found by the simple rules.")
         else: