my-aws-helpers 3.0.0.dev2__tar.gz → 3.0.0.dev4__tar.gz

{my_aws_helpers-3.0.0.dev2 → my_aws_helpers-3.0.0.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: my_aws_helpers
-Version: 3.0.0.dev2
+Version: 3.0.0.dev4
 Summary: AWS Helpers
 Home-page: https://github.com/JarrodMccarthy/aws_helpers.git
 Author: Jarrod McCarthy

{my_aws_helpers-3.0.0.dev2 → my_aws_helpers-3.0.0.dev4}/my_aws_helpers/bedrock.py

@@ -183,15 +183,23 @@ class Bedrock:
                 except Exception as e:
                     self.logger.error(f"Could not get pix map for page {i}")
                     continue
+            header_ocr_result = None
             if len(image_bytes_list) > 1:
                 headers_prompt = self._get_prompt(prompt_type=PromptType.transaction_headers.value)
-                ocr_result = self._ocr(prompt=headers_prompt, image_bytes=image_bytes_list[0])
-                if ocr_result is None:
-                    self.logger.info(f"No ocr result returned when getting headers {PromptType.transaction_headers.value}")
-                headers = ocr_result.content
+                for i in range(2):
+                    # try to get headers from the first or second page
+                    header_ocr_result = self._ocr(prompt=headers_prompt, image_bytes=image_bytes_list[i])
+                    if header_ocr_result is None:
+                        self.logger.info(f"No ocr result returned when getting headers {PromptType.transaction_headers.value}")
+                    headers = header_ocr_result.content.get("headers")
+                    if (len(headers) < 1) or (headers is None):
+                        continue
+                    else:
+                        break
 
             transactions_prompt = self._get_prompt(prompt_type=prompt_type)
-            transactions_prompt.replace("#### TABLE HEADERS ####", json.dumps(headers))
+            if header_ocr_result:
+                transactions_prompt = transactions_prompt.replace("#### TABLE HEADERS ####", json.dumps(header_ocr_result.content))
 
             self.logger.info("Got Prompt")
             results = list()

{my_aws_helpers-3.0.0.dev2 → my_aws_helpers-3.0.0.dev4}/my_aws_helpers/prompts/transactions_headers_prompt.txt

@@ -1,19 +1,13 @@
 You are an intelligent document and image interpreter.
 
-Your task is to analyze the provided image and extract transaction table headers.
+Your task is to analyze the provided image and extract the headers of a table that contains transactions.
 
 ## Output Requirements
 
 - Return **only valid JSON**.
 - If the image contains one or more **tables**, use only the transactions table represent the headers as a JSON object.
-- Use **the table headers as keys**.
-- if the table headers are not present use:
-#### TABLE HEADERS ####
-- if i have just told you to use "TABLE HEADERS", ignore that instruction
-- if there are multiple sections, only include the row data in transactions
-- Every row should have the consistent headers
+- **If there is no table containing transactions, return an empty array for the headers**
 - **Do not fabricate** values not visible in the image
-- if a descripion contains 2 lines of text, only include the most important text and the text should reside in 1 key in json response, do not split it
 - Do not return anything except the json content
 
 ## Example

{my_aws_helpers-3.0.0.dev2 → my_aws_helpers-3.0.0.dev4}/my_aws_helpers/prompts/transactions_prompt.txt

@@ -9,6 +9,7 @@ Your task is to analyze the provided image and extract all meaningful data as st
 - Use **the table headers as keys**.
 - if the table headers are not present use:
 #### TABLE HEADERS ####
+- if i have just told you to use "TABLE HEADERS", ignore that instruction
 - if there are multiple sections, only include the row data in transactions
 - Every row should have the consistent headers
 - If content is unclear, use: `"[Unclear]"`.

{my_aws_helpers-3.0.0.dev2 → my_aws_helpers-3.0.0.dev4}/my_aws_helpers.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: my-aws-helpers
-Version: 3.0.0.dev2
+Version: 3.0.0.dev4
 Summary: AWS Helpers
 Home-page: https://github.com/JarrodMccarthy/aws_helpers.git
 Author: Jarrod McCarthy

{my_aws_helpers-3.0.0.dev2 → my_aws_helpers-3.0.0.dev4}/setup.py

@@ -3,7 +3,7 @@ from setuptools import find_namespace_packages, setup
 
 base_path = os.path.abspath(os.path.dirname(__file__))
 
-version = "3.0.0.dev2"
+version = "3.0.0.dev4"
 
 setup(
     name="my_aws_helpers",