Spaces:
Sleeping
Sleeping
File size: 1,853 Bytes
f725299 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
import os
from scripts.signature_blocks import SignatureBlockModel
from typing import List, Any, Tuple
def flatten_list(xss: List[List[Any]]) -> List[Any]:
return [x for xs in xss for x in xs]
def agreement_status(labels: List[str]) -> str:
if labels:
if len(set(labels)) > 1:
return "Partially Executed"
elif list(set(labels))[0] == "SIGNED_BLOCK":
return "Fully Executed"
elif list(set(labels))[0] == "UNSIGNED_BLOCK":
return "Unsigned"
else:
return "Unknown"
def execution_status(
images: List[Any], show: bool = False
) -> (int, str, List[Any], List[Any]):
if isinstance(images, list):
labels = []
boxes = []
crops = []
for page in images:
model = SignatureBlockModel(page)
if model.predictions[0]["boxes"].shape[0] > 0:
page_labels = model._get_labels_names()
labels.append(page_labels)
boxes.extend(model.get_box_crops())
crops.extend(model.get_boxes())
if show:
boxes = model.show_boxes()
# page.close()
num_sig_pages = len(labels)
execution_status = agreement_status(flatten_list(labels))
return num_sig_pages, execution_status, boxes, crops
else:
return None, None, None, None
if __name__ == "__main__":
from gabriel.parsers.pdf_parser import ParsePDF
filepath = "/Users/jordandavis/GitHub/gabriel/gabriel/datasets/MASTER_REVIEWED/SIGNATURE_PAGE/1a90afa457f328fc7f560d9b49af7b8f.pdf"
image = list(ParsePDF(filepath).yield_image())[0]
num_sig_pages, status, boxes, crops = execution_status(image)
print(f"Num Sig Pages: {num_sig_pages}")
print(f"Status: {status}")
print(f"Boxes: {boxes}")
print(f"Crops: {crops}")
|