Update helpers/indexer.py

helpers/indexer.py

@@ -1,12 +1,11 @@
-from typing import List, Dict, Any, Optional
-
+from typing import List, Dict, Optional
 
 def detect_extra_type(text: str) -> Optional[str]:
     """يتعرف على إن كان النص عبارة عن صورة أو رابط."""
     lower = text.lower()
 
     # رابط
-    if lower.startswith("http://") or lower.startswith("https://"):
+    if lower.startswith(("http://", "https://")):
         # رابط صورة؟
         if lower.endswith((".jpg", ".jpeg", ".png", ".gif", ".webp")):
             return "image"
@@ -47,7 +46,7 @@ def build_indexed_response(
         })
 
     # ================================================================
-    # 1) إذا لم يوجد title_index ولا preamble_end → كل شيء نص عادي
+    # (1) إذا لم يوجد title_index ولا preamble_end → كل شيء نص عادي
     # ================================================================
     if title_index is None and preamble_end is None:
         for t in texts:
@@ -55,22 +54,20 @@ def build_indexed_response(
         return result
 
     # ================================================================
-    # 2) معالجة العنوان
+    # (2) العنوان
     # ================================================================
     if title_index is not None:
         title_index = min(title_index, last_idx)
         append_item(texts[title_index]["text"], "title")
 
     # ================================================================
-    # 3) معالجة المقدمة (preamble)
-    #    - إذا تم تقديم preamble_end فقط → نستنتج start تلقائياً
+    # (3) المقدمة
     # ================================================================
     if preamble_end is not None:
         preamble_end = min(preamble_end, last_idx)
 
-        # استنتاج البداية تلقائياً
+        # استنتاج البداية تلقائياً إذا لم يحددها المستخدم
         if preamble_start is None:
-            # أفضل تخمين منطقي: السطر بعد العنوان
             preamble_start = (title_index + 1) if title_index is not None else 0
 
         preamble_start = min(preamble_start, last_idx)
@@ -79,13 +76,10 @@ def build_indexed_response(
             append_item(texts[i]["text"], "preamble")
 
     # ================================================================
-    # 4) النص الأساسي (body)
-    #    - إذا لم يتم إرسال body_start → نستنتجه = preamble_end + 1
-    #    - إذا لم يتم إرسال body_end → يصبح آخر سطر
+    # (4) النص الأساسي (body)
     # ================================================================
-    if preamble_end is not None:
-        if body_start is None:
-            body_start = preamble_end + 1
+    if preamble_end is not None and body_start is None:
+        body_start = preamble_end + 1
 
     if body_start is not None:
         body_start = min(body_start, last_idx)