Update json_handling.py

json_handling.py

@@ -34,28 +34,42 @@ import json
         
 def process_whole_doc(file_path):
     images = load_input(file_path)
-    
-    complete_json = ""
-    
 
-    # PART_NAME = "PART-1 - BILL OF ENTRY SUMMARY"  # you can change per doc type
-    for i, image in enumerate(images, start=1):
-        partial_json = process_document(image)
-        # print(partial_json)
-        complete_json = complete_json+partial_json
-        # # 🔹 Extract only "result"
-        # result_data = partial_json.get("result", {})
-        # page_key = f"Page {i}"
-        # Ensure structure exists
-        # # if PART_NAME not in complete_json:
-        #     complete_json[] = [{}]
+    complete_json = "["
+    first = True
 
-        # if not complete_json[PART_NAME]:
-        #     complete_json[PART_NAME].append({})
+    for image in images:
+        partial_json = process_document(image)  # returns string
 
-        # ✅ Merge directly
-        # complete_json[page_key]=result_data
+        if not first:
+            complete_json += ","
+        complete_json += partial_json
+        first = False
+
+    complete_json += "]"
 
     return complete_json
+    # complete_json = ""
+    
+    
+    # # PART_NAME = "PART-1 - BILL OF ENTRY SUMMARY"  # you can change per doc type
+    # for i, image in enumerate(images, start=1):
+    #     partial_json = process_document(image)
+    #     # print(partial_json)
+    #     complete_json = complete_json+partial_json
+    #     # # 🔹 Extract only "result"
+    #     # result_data = partial_json.get("result", {})
+    #     # page_key = f"Page {i}"
+    #     # Ensure structure exists
+    #     # # if PART_NAME not in complete_json:
+    #     #     complete_json[] = [{}]
+
+    #     # if not complete_json[PART_NAME]:
+    #     #     complete_json[PART_NAME].append({})
+
+    #     # ✅ Merge directly
+    #     # complete_json[page_key]=result_data
+
+    # return complete_json