Update inference.py

inference.py

@@ -362,18 +362,67 @@ def extract_part_json(image, part_name, model, processor, device):
     }
 
 
+# def merge_page_results(page_results):
+#     final_json = {}
+
+#     for item in page_results:
+#         if item["status"] != "success" or not item["parsed"]:
+#             continue
+
+#         parsed = item["parsed"]
+#         for key, value in parsed.items():
+#             final_json[key] = value
+
+#     return final_json
+
+
+# Adding these to handle json in structured format add from line 381 to 425
+
 def merge_page_results(page_results):
-    final_json = {}
+    final_json = {
+        "PART-1": {},
+        "PART-2": {},
+        "PART-3": {},
+        "PART-4": {},
+        "PART-5": {},
+        "PART-6": {}
+    }
 
     for item in page_results:
         if item["status"] != "success" or not item["parsed"]:
             continue
 
+        part = item["part"]
         parsed = item["parsed"]
-        for key, value in parsed.items():
-            final_json[key] = value
 
-    return final_json
+        final_json[part] = _merge_values(final_json[part], parsed)
+
+    return {key: value for key, value in final_json.items() if value}
+
+
+
+def _merge_values(old_value, new_value):
+    if old_value is None:
+        return new_value
+
+    if isinstance(old_value, list) and isinstance(new_value, list):
+        return old_value + new_value
+
+    if isinstance(old_value, dict) and isinstance(new_value, dict):
+        merged = dict(old_value)
+
+        for key, value in new_value.items():
+            if key in merged:
+                merged[key] = _merge_values(merged[key], value)
+            else:
+                merged[key] = value
+
+        return merged
+
+    if old_value in ("", None, [], {}):
+        return new_value
+
+    return old_value
 
 
 def process_document(file_path):