Pre FastAPI

data.json

@@ -1,77 +1,6 @@
 [
     {
-        "merchant": "TESCO SHREWSBURY CATTLE MARKET",
-        "date": "17/10/22",
-        "total_amount": 55.3,
-        "items": [
-            {
-                "description": "TAPE MEASURE",
-                "amount": 2.1
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.0
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.0
-            },
-            {
-                "description": "CONDOMS",
-                "amount": 7.0
-            },
-            {
-                "description": "PROCECCO",
-                "amount": 8.0
-            },
-            {
-                "description": "PROCECCO",
-                "amount": 8.0
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "TAPE MEASURE",
-                "amount": 2.1
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.0
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.0
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.0
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "KRONENBOURG",
-                "amount": 1.1
-            },
-            {
-                "description": "KRONENBOURG",
-                "amount": 1.0
-            }
-        ],
-        "fraud_check": false
-    },
-    {
+        "fraud_check": false,
         "merchant": "TESCO",
         "date": "17/10/22",
         "total_amount": 55.3,
@@ -140,194 +69,25 @@
                 "description": "KRONENBOURG",
                 "amount": 4.6
             }
-        ],
-        "fraud_check": false
-    },
-    {
-        "merchant": "TESCO SHREWSBURY CATTLE MARKET",
-        "date": "17/10/22",
-        "total_amount": 55.3,
-        "items": [
-            {
-                "description": "TAPE MEASURE",
-                "amount": 2.1
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.8
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.8
-            },
-            {
-                "description": "CONDOMS",
-                "amount": 10.0
-            },
-            {
-                "description": "PROCECCO",
-                "amount": 7.0
-            },
-            {
-                "description": "PROCECCO",
-                "amount": 7.0
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "KRONENBOURG",
-                "amount": 4.6
-            },
-            {
-                "description": "TAPE MEASURE",
-                "amount": 2.1
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.8
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.8
-            },
-            {
-                "description": "CONDOMS",
-                "amount": 10.0
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "KRONENBOURG",
-                "amount": 4.6
-            }
-        ],
-        "fraud_check": false
-    },
-    {
-        "fraud_check": false,
-        "merchant": "TESCO SHREWSBURY CATTLE MARKET",
-        "date": "17/10/22",
-        "total_amount": 55.3,
-        "items": [
-            {
-                "description": "TAPE MEASURE",
-                "amount": 2.1
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.8
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.8
-            },
-            {
-                "description": "CONDOMS",
-                "amount": 10.0
-            },
-            {
-                "description": "PROCECCO",
-                "amount": 7.0
-            },
-            {
-                "description": "PROCECCO",
-                "amount": 7.0
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "KRONENBOURG",
-                "amount": 4.6
-            }
         ]
     },
     {
-        "fraud_check": false,
-        "merchant": "Tesco",
-        "date": "17/10/22",
-        "total_amount": 55.3,
+        "fraud_check": [],
+        "merchant": "Walmart",
+        "date": "01/26/21",
+        "total_amount": 29.73,
         "items": [
             {
-                "description": "TAPE MEASURE",
-                "amount": 2.1
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.8
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.8
-            },
-            {
-                "description": "CONDOMS",
-                "amount": 10.0
-            },
-            {
-                "description": "PROCECCO",
-                "amount": 7.0
-            },
-            {
-                "description": "PROCECCO",
-                "amount": 7.0
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "KRONENBOURG",
-                "amount": 4.6
+                "description": "BOYS CREW S# 028944",
+                "amount": 9.48
             },
             {
-                "description": "TAPE MEASURE",
-                "amount": 2.1
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.8
-            },
-            {
-                "description": "PROTEIN COOKIE",
-                "amount": 1.8
+                "description": "BOYS SOCKS",
+                "amount": 4.97
             },
             {
-                "description": "CONDOMS",
-                "amount": 10.0
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "SAUVIGNON BLNC",
-                "amount": 7.0
-            },
-            {
-                "description": "KRONENBOURG",
-                "amount": 4.6
+                "description": "BOXER BRIEF",
+                "amount": 9.48
             }
         ]
     }

fraud.py

@@ -1,5 +1,6 @@
 import json
 import os
+from models import FraudData
 
 DATA_FILE = 'data.json'
 
@@ -34,7 +35,7 @@ def save_receipts(receipts):
         json.dump(receipts, f, indent=4)
 
 def receipts_are_equal(receipt1, receipt2):
-    """Check if two receipts are the same, comparing only basic fields."""
+    """Check if two receipts are the same, comparing only date and amount."""
     print(f"\n=== COMPARING RECEIPTS ===")
     print(f"Receipt1 merchant: {receipt1.get('merchant')}")
     print(f"Receipt2 merchant: {receipt2.get('merchant')}")
@@ -44,22 +45,15 @@ def receipts_are_equal(receipt1, receipt2):
         print("One or both receipts are empty")
         return False
     
-    # Check basic fields with special handling for merchant
-    merchant1 = receipt1.get('merchant', '').lower()
-    merchant2 = receipt2.get('merchant', '').lower()
+    # Only compare date and amount
     date1 = receipt1.get('date')
     date2 = receipt2.get('date')
     amount1 = receipt1.get('total_amount')
     amount2 = receipt2.get('total_amount')
     
-    print(f"Comparing merchant (lowercase): '{merchant1}' vs '{merchant2}'")
     print(f"Comparing date: '{date1}' vs '{date2}'")
     print(f"Comparing total_amount: '{amount1}' vs '{amount2}'")
     
-    if merchant1 != merchant2:
-        print("Merchant field doesn't match")
-        return False
-    
     if date1 != date2:
         print("Date field doesn't match")
         return False
@@ -68,7 +62,7 @@ def receipts_are_equal(receipt1, receipt2):
         print("Total amount field doesn't match")
         return False
     
-    print("All basic fields match - receipts are equal")
+    print("Date and amount match - receipts are equal")
     return True
 
 def is_duplicate(new_receipt, receipts):
@@ -78,9 +72,6 @@ def is_duplicate(new_receipt, receipts):
     
     for i, old_receipt in enumerate(receipts):
         print(f"\nChecking against receipt {i}:")
-        print(f"  Stored: {old_receipt.get('merchant')} - {old_receipt.get('date')} - {old_receipt.get('total_amount')}")
-        print(f"  New: {new_receipt.get('merchant')} - {new_receipt.get('date')} - {new_receipt.get('total_amount')}")
-        
         if receipts_are_equal(old_receipt, new_receipt):
             print(f"DUPLICATE FOUND at index {i}!")
             return True
@@ -102,11 +93,13 @@ def process_receipt(new_receipt):
     
     if is_duplicate(new_receipt, receipts):
         print("SETTING FRAUD CHECK TO TRUE")
-        new_receipt['fraud_check'] = True
+        # Create FraudData object and convert to dict
+        fraud_data = FraudData(fraud_detected=True, fraud_type="duplicate")
+        new_receipt['fraud_check'] = [fraud_data.dict()]
         # Do not save, just return
     else:
         print("SETTING FRAUD CHECK TO FALSE - SAVING RECEIPT")
-        new_receipt['fraud_check'] = False
+        new_receipt['fraud_check'] = []  # Empty list means no fraud detected
         receipts.append(new_receipt)
         save_receipts(receipts)
     

models.py

@@ -5,8 +5,12 @@ class ReceiptItem(BaseModel):
     description: str
     amount: float
 
+class FraudData(BaseModel):
+    fraud_detected: bool
+    fraud_type: Optional[str] = None  # Type of fraud if detected, e.g., "duplicate", "suspicious"
+
 class ReceiptData(BaseModel):
-    fraud_check: Optional[bool] = False  # Optional field for fraud detection
+    fraud_check: Optional[List[FraudData]] = False  # Optional field for fraud detection
     merchant: str
     date: str  
     total_amount: float

outputs/filled_child_fee_form_6dfe1052f5be4bbc8a78711a456a3cc9.pdf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:48066dd238b5a6eda1e8cd8647afc31074c61b7b95946e6b4329426f4a818331
+size 115543

outputs/filled_child_fee_form_e3fba6d8482a476dade270c022cc64e2.pdf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa5b02270d310e55abfdf36281e54e88928ad3b71d19f034b784b1616a2c9eb5
+size 115542

pipeline.py

@@ -20,8 +20,11 @@ reciept_system_prompt = (
     "class ReceiptItem(BaseModel):\n"
     "    description: str\n"
     "    amount: float\n\n"
+    "class FraudData(BaseModel):\n"
+    "    fraud_detected: bool \n"
+    "    fraud_type: Optional[str] = None  # Type of fraud if detected, e.g., \"duplicate\", \"suspicious\" \n\n"
     "class ReceiptData(BaseModel):\n"
-    "    fraud_check: Optional[bool] = False  # Optional field for fraud detection, always set to false"
+    "    fraud_check: Optional[List[FraudData]] = []  # Optional field for fraud detection, always set to empty list\n"
     "    merchant: str #Only extract the brand name, not the branch name - Only the brand\n"
     "    date: str\n"
     "    total_amount: float\n #Try your hardest to find the accurate total amount\n"
@@ -29,6 +32,7 @@ reciept_system_prompt = (
     "- Extract only the above given information.\n"
     "- If a value is missing, set it to null, \"\", or an empty list as appropriate.\n"
     "- For the items field, provide a list of objects with description and amount.\n"
+    "- For fraud_check, always set to an empty list [].\n"
     "- Only return a valid JSON object matching the model above.\n"
     "- Do not add any explanation or extra text—only the JSON."
 )
@@ -166,4 +170,3 @@ def extract_child_fee_info(img_input, emp_name, emp_code, department):
     except Exception as e:
         print("ERROR:", e)
         return None  # or f"Error: {str(e)}"
-