Update main.py

main.py

@@ -200,6 +200,42 @@ def _get_steps_for_action(best_doc: str, actions: list) -> Optional[str]:
     return None
 
 
+def _get_steps_for_action_global(actions: list, prefer_doc: Optional[str] = None) -> Tuple[Optional[str], Optional[str]]:
+    """
+    Search ALL SOP docs for a 'steps' section whose action_tag matches one of `actions`.
+    Returns (doc_name, steps_text). Prefers sections from `prefer_doc` and the 'appointments' module.
+    """
+    if not actions:
+        return None, None
+    act_set = set(a.strip().lower() for a in actions if a)
+
+    candidates: List[Tuple[float, str, str]] = []  # (score, doc, text)
+    for d in bm25_docs:
+        m = d.get("meta", {}) or {}
+        if m.get("intent_tag") != "steps":
+            continue
+        tag = ((m.get("action_tag") or "").strip().lower())
+        if tag and tag in act_set:
+            doc = m.get("filename")
+            sec = (m.get("section") or "").strip()
+            txt = get_section_text(doc, sec)
+            if not txt or not txt.strip():
+                continue
+            score = 1.0
+            if prefer_doc and doc == prefer_doc:
+                score += 0.5
+            mtags = (m.get("module_tags") or "").lower()
+            if "appointments" in mtags:
+                score += 0.3
+            candidates.append((score, doc, txt.strip()))
+
+    if not candidates:
+        return None, None
+
+    candidates.sort(key=lambda x: x[0], reverse=True)
+    _, best_doc_global, best_text = candidates[0]
+    return best_doc_global, best_text
+
 # --- Default section picker when query doesn't reveal action ---
 def _pick_default_action_section(best_doc: str) -> Optional[str]:
     """
@@ -1219,19 +1255,30 @@ async def chat_with_ai(input_data: ChatInput):
             if action_steps:
                 full_steps = action_steps
             else:
-                # 2) If no user action, pick a sensible default section (creation > update > deletion)
-                default_sec = _pick_default_action_section(best_doc)
-                if default_sec:
-                    full_steps = get_section_text(best_doc, default_sec)
-                # 3) Last resort: aggregated steps for the doc
+                if kb_results.get("actions"):
+                   alt_doc, alt_steps = _get_steps_for_action_global(kb_results["actions"], prefer_doc=best_doc)
+                   if alt_steps:
+                      best_doc = alt_doc   # switch to the doc that actually has the section
+                      full_steps = alt_steps
+                       
+                asked_action = _detect_action_from_query(input_data.user_message)
+                if not kb_results.get("actions") and asked_action:
+                   alt_doc, alt_steps = _get_steps_for_action_global([asked_action], prefer_doc=best_doc)
+                   if alt_steps:
+                      best_doc = alt_doc   # switch to the doc that actually has the section
+                      full_steps = alt_steps
+
                 if not full_steps:
-                    full_steps = get_best_steps_section_text(best_doc)
-                # 4) Final fallback: top hit section
+                   default_sec = _pick_default_action_section(best_doc)
+                   if default_sec:
+                      full_steps = get_section_text(best_doc, default_sec)
                 if not full_steps:
-                    sec = (top_meta or {}).get("section")
-                    if sec:
-                        full_steps = get_section_text(best_doc, sec)
-
+                   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)
+    
             if full_steps:
                 # Always add Save lines if present anywhere in the doc (independent of query wording)
                 save_lines = _find_save_lines_in_doc(best_doc, max_lines=2)