add data viewer

app/main.py

@@ -760,62 +760,24 @@ async def view_all_document_chunks():
 def build_chunk_tree(chunks: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
     """
     Xây dựng cấu trúc cây từ danh sách chunks phẳng.
-    Đảm bảo thứ tự phân cấp đúng: PHAN -> PHU_LUC -> CHUONG -> MUC -> DIEU -> KHOAN -> DIEM
+    Cách đơn giản: tìm root nodes (cha=None) trước, sau đó tìm children.
     """
     if not chunks:
         return []
     
-    # Sắp xếp chunks theo thứ tự xuất hiện (dựa vào content để đoán thứ tự)
-    # Chunks có level cao hơn thường xuất hiện trước
-    level_priority = {
-        "PHAN": 1,
-        "PHU_LUC": 1,
-        "CHUONG": 2,
-        "MUC": 3,
-        "DIEU": 4,
-        "KHOAN": 5,
-        "DIEM": 6,
-        "CONTENT": 7
-    }
-    
-    def get_chunk_level(chunk):
-        """Đoán level của chunk dựa vào content"""
-        content = chunk.get('content', '').upper()
-        if any(keyword in content for keyword in ['PHẦN', 'PHẦN THỨ']):
-            return 1
-        elif 'PHỤ LỤC' in content:
-            return 1
-        elif 'CHƯƠNG' in content:
-            return 2
-        elif 'MỤC' in content:
-            return 3
-        elif 'ĐIỀU' in content:
-            return 4
-        elif any(keyword in content for keyword in ['KHOẢN', 'ĐIỂM']):
-            return 5
-        else:
-            return 7
-    
-    # Sắp xếp chunks theo level (thấp hơn = cao hơn trong phân cấp)
-    sorted_chunks = sorted(chunks, key=lambda x: get_chunk_level(x))
-    
     # Tạo dictionary để truy cập nhanh
-    chunks_dict = {chunk['id']: chunk for chunk in sorted_chunks}
+    chunks_dict = {chunk['id']: chunk for chunk in chunks}
     
-    # Tạo cấu trúc cây
-    root_chunks = []
-    processed_nodes = {}  # Để theo dõi các node đã được xử lý
-    
-    for chunk in sorted_chunks:
-        chunk_id = chunk['id']
-        parent_id = chunk.get('cha')
+    def build_node(chunk_id: str) -> Dict[str, Any]:
+        """Tạo node và tìm tất cả children của nó."""
+        chunk = chunks_dict[chunk_id]
         
-        # Tạo node mới với cấu trúc cây
-        tree_node = {
+        # Tạo node
+        node = {
             "id": chunk_id,
             "content": chunk.get('content', ''),
             "vanbanid": chunk.get('vanbanid'),
-            "cha": parent_id,
+            "cha": chunk.get('cha'),
             "document_title": chunk.get('document_title', ''),
             "article_number": chunk.get('article_number'),
             "article_title": chunk.get('article_title', ''),
@@ -826,22 +788,22 @@ def build_chunk_tree(chunks: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
             "children": []
         }
         
-        # Lưu node vào processed_nodes
-        processed_nodes[chunk_id] = tree_node
+        # Tìm tất cả children của node này
+        for other_chunk in chunks:
+            if other_chunk.get('cha') == chunk_id:
+                child_node = build_node(other_chunk['id'])
+                node["children"].append(child_node)
         
-        if parent_id is None:
-            # Đây là root node
-            root_chunks.append(tree_node)
-        else:
-            # Tìm parent và thêm vào children
-            parent_node = processed_nodes.get(parent_id)
-            if parent_node:
-                # Parent đã được xử lý trước đó
-                parent_node["children"].append(tree_node)
-            else:
-                # Parent chưa được xử lý, coi như root
-                root_chunks.append(tree_node)
+        return node
+    
+    # Tìm tất cả root nodes (cha=None)
+    root_chunks = []
+    for chunk in chunks:
+        if chunk.get('cha') is None:
+            root_node = build_node(chunk['id'])
+            root_chunks.append(root_node)
     
+    logger.info(f"[TREE] Built tree with {len(root_chunks)} root nodes from {len(chunks)} total chunks")
     return root_chunks
 
 @app.get("/api/document-chunks/status")