add process_whole_doc fun in app.py

app.py

@@ -22,6 +22,8 @@ from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import HTMLResponse
 import shutil
 from inference import process_document
+from json_handling import process_whole_doc
+
 
 app = FastAPI()
 
@@ -47,6 +49,7 @@ async def extract(file: UploadFile = File(...)):
     with open(file_path, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
 
-    result = process_document(file_path)
+    # result = process_document(file_path)
+    result = process_whole_doc(file_path)
 
     return {"result": result}