Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 from fastapi import FastAPI, HTTPException, Query
 from pydantic import BaseModel, HttpUrl, Field
 from typing import List, Dict, Optional, Any, Tuple
@@ -54,6 +53,10 @@ def validate_ranges(ranges: List[List[int]]) -> List[Tuple[int, int]]:
 @app.post("/extract", response_model=LegalDocumentResponse)
 async def extract_text_endpoint(request: URLRequest):
     try:
+        # إذا طلب الحفظ، نضمن أن التحليل مفعل
+        if request.save_to_supabase:
+            request.return_parsed = True
+
         # 1) استخراج جميع النصوص
         all_texts = await extract_text_from_url(str(request.url), request.timeout)
 
@@ -67,11 +70,22 @@ async def extract_text_endpoint(request: URLRequest):
         # 3) تحليل القانون إذا طلب
         if request.return_parsed:
             parsed_document = parse_law_from_texts(filtered_texts)
+
+            # تعديل الرسالة
+            parsed_document["message"] = "success"
+
+            # تعديل all_blocks ليحتوي count و items
+            parsed_document["all_blocks"] = {
+                "count": len(filtered_texts),
+                "items": filtered_texts
+            }
+
             if request.save_to_supabase:
                 save_law_to_supabase(parsed_document["law"])
                 parsed_document["saved_to_db"] = True
             else:
                 parsed_document["saved_to_db"] = False
+
             return LegalDocumentResponse(parsed_document=parsed_document)
 
         # 4) النصوص الخام
@@ -92,19 +106,32 @@ async def extract_text_get(
     timeout: int = Query(10, ge=1, le=60, description="مهلة الانتظار بالثواني")
 ):
     try:
+        # إذا طلب الحفظ، نضمن أن التحليل مفعل
+        if save_to_supabase:
+            return_parsed = True
+
         # 1) استخراج جميع النصوص
         all_texts = await extract_text_from_url(str(url), timeout)
 
-        # لا يوجد دعم للنطاقات في GET حالياً، يمكن إضافة لاحقاً إذا أردت
-
         # 2) تحليل القانون إذا طلب
         if return_parsed:
             parsed_document = parse_law_from_texts(all_texts)
+
+            # تعديل الرسالة
+            parsed_document["message"] = "success"
+
+            # تعديل all_blocks ليحتوي count و items
+            parsed_document["all_blocks"] = {
+                "count": len(all_texts),
+                "items": all_texts
+            }
+
             if save_to_supabase:
                 save_law_to_supabase(parsed_document["law"])
                 parsed_document["saved_to_db"] = True
             else:
                 parsed_document["saved_to_db"] = False
+
             return LegalDocumentResponse(parsed_document=parsed_document)
 
         # 3) النصوص الخام