Update helpers/indexer.py

helpers/indexer.py

@@ -1,10 +1,9 @@
-# helpers/indexer.py
 from typing import List, Dict, Optional
 
 def build_indexed_response(
     texts: List[Dict[str, str]],
-    title_index: int,
-    preamble_end: int,
+    title_index: Optional[int] = None,
+    preamble_end: Optional[int] = None,
     body_end: Optional[int] = None,
     *,
     preamble_start: Optional[int] = None,
@@ -20,19 +19,31 @@ def build_indexed_response(
         return []
 
     last_idx = len(texts) - 1
-    title_index = min(title_index, last_idx)
-    preamble_end = min(preamble_end, last_idx)
-    body_end = min(body_end if body_end is not None else last_idx, last_idx)
-
-    preamble_start = min(preamble_start or title_index + 1, last_idx)
-    body_start = min(body_start or preamble_end + 1, last_idx)
-
     result = []
 
-    result.append({"text": texts[title_index]["value"].strip(), "type": "title"})
-    for i in range(preamble_start, preamble_end + 1):
-        result.append({"text": texts[i]["value"].strip(), "type": "preamble"})
-    for i in range(body_start, body_end + 1):
-        result.append({"text": texts[i]["value"].strip(), "type": "body"})
+    # ===== العنوان =====
+    if title_index is not None and 0 <= title_index <= last_idx:
+        result.append({
+            "text": texts[title_index]["value"].strip(),
+            "type": "title"
+        })
+
+    # ===== المقدمة =====
+    if preamble_end is not None and preamble_end >= 0:
+        start = preamble_start or (title_index + 1 if title_index is not None else 0)
+        for i in range(start, min(preamble_end, last_idx) + 1):
+            result.append({
+                "text": texts[i]["value"].strip(),
+                "type": "preamble"
+            })
+
+    # ===== النص الأساسي (body) =====
+    start_body = body_start or (preamble_end + 1 if preamble_end is not None else 0)
+    end_body = body_end or last_idx
+    for i in range(start_body, min(end_body, last_idx) + 1):
+        result.append({
+            "text": texts[i]["value"].strip(),
+            "type": "body"
+        })
 
     return result