Update main.py

main.py

@@ -176,6 +176,29 @@ def _get_steps_for_action(best_doc: str, actions: list) -> Optional[str]:
             return txt.strip()
     return None
 
+def _get_steps_by_action_tag(best_doc: str, action: Optional[str]) -> Optional[str]:
+    """
+    Generic: return the section text from `best_doc` whose meta.action_tag == action.
+    No hard-coded SOP words; relies purely on KB metadata.
+    """
+    if not best_doc or not action:
+        return None
+    target = (action or "").strip().lower()
+    for d in bm25_docs:
+        m = d.get("meta", {}) or {}
+        if m.get("intent_tag") != "steps":
+            continue
+        if (m.get("filename") or "") != best_doc:
+            continue
+        tag = (m.get("action_tag") or "").strip().lower()
+        if tag and tag == target:
+            sec = (m.get("section") or "").strip()
+            if sec:
+                txt = get_section_text(best_doc, sec)
+                if txt and txt.strip():
+                    return txt.strip()
+    return None
+
 def _get_steps_for_action_global(actions: list, prefer_doc: Optional[str] = None) -> Tuple[Optional[str], Optional[str]]:
     if not actions:
         return None, None
@@ -205,11 +228,26 @@ def _get_steps_for_action_global(actions: list, prefer_doc: Optional[str] = None
     _, best_doc_global, best_text = candidates[0]
     return best_doc_global, best_text
 
-def _pick_default_action_section(best_doc: str) -> Optional[str]:
+def _pick_default_action_section_with_preference(best_doc: str, prefer_action: Optional[str]) -> Optional[str]:
+    """
+    Generic: if there are sections tagged with the requested action (via action_tag),
+    prefer them; otherwise fall back to your original order.
+    """
+    # 1) Prefer exact action_tag match
+    if prefer_action:
+        for d in bm25_docs:
+            m = d.get("meta", {}) or {}
+            if m.get("filename") == best_doc and m.get("intent_tag") == "steps":
+                tag = (m.get("action_tag") or "").strip().lower()
+                sec = (m.get("section") or "").strip()
+                if tag == prefer_action.lower() and sec:
+                    return sec
+
+    # 2) Fall back to your original generic order (no SOP-specific words)
     order = ("creation", "updation", "update", "deletion", "delete", "cancel")
     sections = []
     for d in bm25_docs:
-        m = d.get("meta", {})
+        m = d.get("meta", {}) or {}
         if m.get("filename") == best_doc and m.get("intent_tag") == "steps":
             title = (m.get("section") or "").strip().lower()
             if title:
@@ -394,7 +432,7 @@ 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 = 6, min_score: float = 0.35):
+def _resolve_next_steps(user_query: str, numbered_text: str, max_next: int = 6, min_score: float = 0.30):
     """
     Smarter anchor: verb overlap boost + keyword anchors.
     """
@@ -1065,6 +1103,9 @@ async def chat_with_ai(input_data: ChatInput):
             context_preformatted = False
             full_steps = None
 
+# Detect asked action generically using ACTION_SYNONYMS_EXT (already defined)
+            asked_action = _detect_action_from_query(input_data.user_message)  # returns 'create'/'update'/'delete'/None
+
             action_steps = _get_steps_for_action(best_doc, kb_results.get("actions", []))
             if action_steps:
                 full_steps = action_steps
@@ -1074,8 +1115,16 @@ async def chat_with_ai(input_data: ChatInput):
                     if alt_steps:
                         best_doc = alt_doc
                         full_steps = alt_steps
-                asked_action = _detect_action_from_query(input_data.user_message)
-                
+               
+                if not full_steps and asked_action:
+                   via_tag_steps = _get_steps_by_action_tag(best_doc, asked_action)
+                   if via_tag_steps:
+                      full_steps = via_tag_steps
+                if not full_steps:
+                   default_sec = _pick_default_action_section_with_preference(best_doc, asked_action)
+                   if default_sec:
+                      full_steps = get_section_text(best_doc, default_sec)
+
                 if not full_steps and asked_action:
                     exact_action_steps = _get_steps_text_for_action(best_doc, asked_action)
                     if exact_action_steps:
@@ -1125,10 +1174,9 @@ async def chat_with_ai(input_data: ChatInput):
                 else:
                     context = full_steps
                     context_preformatted = False
-
-                filt_info = {'mode': None, 'matched_count': None, 'all_sentences': None}
                 context_found = True
-
+                filt_info = {'mode': None, 'matched_count': None, 'all_sentences': None}
+                
         elif best_doc and detected_intent == "errors":
             full_errors = get_best_errors_section_text(best_doc)
             if full_errors: