Update main.py

main.py

@@ -185,6 +185,46 @@ def _normalize_lines(text: str) -> List[str]:
         return [raw.strip()] if raw.strip() else []
 
 
+# --- NEW: action-aware filter for numbered steps (creation/update/delete) ---
+def _filter_numbered_steps_by_actions(numbered_text: str,
+                                      wanted: set[str],
+                                      exclude: set[str]) -> str:
+    """
+    Keep only lines that match the 'wanted' actions and drop lines that match 'exclude'.
+    Works on already-numbered/bulleted text (one step per line).
+    """
+    # Keep synonyms aligned with kb_creation.ACTION_SYNONYMS
+    ACTION_SYNONYMS = {
+        "create": ("create", "creation", "add", "new", "generate"),
+        "update": ("update", "modify", "change", "edit"),
+        "delete": ("delete", "remove"),
+        "navigate": ("navigate", "go to", "open"),
+    }
+
+    def _has_any(line: str, keys: set[str]) -> bool:
+        low = (line or "").lower()
+        for k in keys:
+            for syn in ACTION_SYNONYMS.get(k, (k,)):
+                if syn in low:
+                    return True
+        return False
+
+    out_lines = []
+    for ln in (numbered_text or "").splitlines():
+        # If an exclude action appears in the line, drop it
+        if _has_any(ln, exclude):
+            continue
+        # If caller asked for specific actions, keep only those
+        if wanted:
+            if _has_any(ln, wanted):
+                out_lines.append(ln)
+        else:
+            # no explicit wanted actions → keep as-is
+            out_lines.append(ln)
+
+    # If over-filtering made it empty, fall back to original text
+    return "\n".join(out_lines).strip() or (numbered_text or "").strip()
+
 def _ensure_numbering(text: str) -> str:
     """
     Normalize raw SOP steps into a clean numbered list using circled digits.
@@ -320,6 +360,8 @@ def _detect_next_intent(user_query: str) -> bool:
     ]
     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:
@@ -938,6 +980,25 @@ async def chat_with_ai(input_data: ChatInput):
 
             if full_steps:
                 numbered_full = _ensure_numbering(full_steps)
+                wanted_actions = set((kb_results.get("actions") or []))
+        # If user specifically asked for creation, prefer only creation and exclude update/delete
+                if "create" in wanted_actions and not ({"update", "delete"} & wanted_actions):
+                    numbered_full = _filter_numbered_steps_by_actions(
+                       numbered_full,
+                       wanted={"create"},
+                       exclude={"update", "delete"}
+            )
+                elif "update" in wanted_actions and "delete" not in wanted_actions:
+                   numbered_full = _filter_numbered_steps_by_actions(
+                      numbered_full, 
+                      wanted={"update"}, 
+                      exclude={"create", "delete"}
+            )
+                elif "delete" in wanted_actions and "update" not in wanted_actions:
+                   numbered_full = _filter_numbered_steps_by_actions(
+                      numbered_full, wanted={"delete"}, exclude={"create", "update"}
+                   )
+                    
                 next_only = _resolve_next_steps(
                     input_data.user_message,
                     numbered_full,