Update main.py

main.py

@@ -312,14 +312,18 @@ def _soft_match_score(a: str, b: str) -> float:
     # combine (weights tuned to prefer phrase overlaps slightly)
     return min(1.0, 0.65 * jacc + 0.45 * big)
 
+
 def _detect_next_intent(user_query: str) -> bool:
     q = _norm_text(user_query)
     keys = [
-        "after", "after this", "what next", "whats next", "next step",
-        "then what", "following step", "continue", "subsequent", "proceed"
+        'after','after this','what next','whats next','next step',
+        'then what','following step','continue','subsequent','proceed',
+        # new entries
+        'what to do','what should i do','how to proceed','how do i continue','proceed further','next?'
     ]
     return any(k in q for k in keys)
 
+
 def _resolve_next_steps(user_query: str, numbered_text: str, max_next: int = 8, min_score: float = 0.25):
     """
     Robust next-step resolver:
@@ -932,38 +936,43 @@ async def chat_with_ai(input_data: ChatInput):
         context_preformatted = False
 
         if best_doc and detected_intent == "steps":
-            full_steps = get_best_steps_section_text(best_doc)
-            if not full_steps:
-                sec = (top_meta or {}).get("section")
-                if sec:
-                    full_steps = get_section_text(best_doc, sec)
-
+            
+            sec = (top_meta or {}).get('section', '')
+            actions = kb_results.get('actions', [])
+            sec_low = (sec or '').lower()
+            wants_delete = ('delete' in actions) or ('remove' in actions) or ('delete' in sec_low) or ('remov' in sec_low)
+            if wants_delete and sec:
+               full_steps = get_section_text(best_doc, sec)
+            else:
+               full_steps = get_best_steps_section_text(best_doc)
+               if not full_steps and sec:
+                  full_steps = get_section_text(best_doc, sec)
             if full_steps:
-                numbered_full = _ensure_numbering(full_steps)
-                next_only = _resolve_next_steps(
+               numbered_full = _ensure_numbering(full_steps)
+               next_only = _resolve_next_steps(
                     input_data.user_message,
                     numbered_full,
                     max_next=6,
-                    min_score=0.35
-                )
-                if next_only is not None:
-                    if len(next_only) == 0:
-                        context = "You are at the final step of this SOP. No further steps."
-                        next_step_applied = True
-                        next_step_info = {"count": 0}
-                        context_preformatted = True
-                    else:
-                        context = _format_steps_as_numbered(next_only)
-                        next_step_applied = True
-                        next_step_info = {"count": len(next_only)}
-                        context_preformatted = True
-                else:
-                    context = full_steps
-                    context_preformatted = False
-
-                # clear filter info for debug clarity
-                filt_info = {'mode': None, 'matched_count': None, 'all_sentences': None}
-                context_found = True
+                    min_score=0.35)
+               if next_only is not None:
+                 if len(next_only) == 0:
+                   context = "You are at the final step of this SOP. No further steps."
+                   next_step_applied = True
+                   next_step_info = {"count": 0}
+                   context_preformatted = True
+                 else:
+                     
+                   context = _format_steps_as_numbered(next_only)
+                   next_step_applied = True
+                   next_step_info = {"count": len(next_only)}
+                   context_preformatted = True
+               else:
+                   context = full_steps
+                   context_preformatted = False
+                   
+               # clear filter info for debug clarity
+               filt_info = {'mode': None, 'matched_count': None, 'all_sentences': None}
+               context_found = True
 
         elif best_doc and detected_intent == "errors":
             full_errors = get_best_errors_section_text(best_doc)
@@ -1081,7 +1090,7 @@ Return ONLY the rewritten guidance."""
         return {
             "bot_response": bot_text,
             "status": status,
-            "context_found": True,
+            "context_found": context_found,
             "ask_resolved": (status == "OK" and detected_intent != "errors"),
             "suggest_incident": (status == "PARTIAL"),
             "followup": (assist_followup if assist_followup else ("Is this helpful, or should I raise a ticket?" if status == "PARTIAL" else None)),