Update app.py

app.py

@@ -218,7 +218,7 @@ def weighted_fuzzy_score(s1, s2):
     return fuzz.token_set_ratio(str(s1).lower(), str(s2).lower())
 
 def find_po_number_in_json(po_number, invoice_json):
-    """Recursively search for PO number as substring anywhere in the invoice JSON (case-insensitive, stripped of spaces)."""
+    """Search for PO number (as string/int) anywhere in the invoice JSON, ignoring .0 floats, spaces, and case."""
     def _flatten(obj):
         fields = []
         if isinstance(obj, dict):
@@ -227,13 +227,29 @@ def find_po_number_in_json(po_number, invoice_json):
         elif isinstance(obj, list):
             for item in obj:
                 fields.extend(_flatten(item))
-        elif isinstance(obj, str):
-            fields.append(obj)
+        elif obj is not None:
+            fields.append(str(obj))
         return fields
 
-    all_strs = [str(s).replace(" ", "").lower() for s in _flatten(invoice_json)]
-    po_str = str(po_number).replace(" ", "").lower()
-    return any(po_str in s for s in all_strs if s and po_str)
+    # Clean up PO number
+    po_str = str(po_number).strip().replace(" ", "").replace(".0", "")
+    try:
+        po_int = str(int(float(po_number)))  # also get int version
+    except:
+        po_int = po_str
+
+    all_strs = [str(s).strip().replace(" ", "").replace(".0", "") for s in _flatten(invoice_json)]
+
+    # Check for exact match, or substring match, with all variants
+    for s in all_strs:
+        if not s:
+            continue
+        if po_str and (po_str in s or s in po_str):
+            return True
+        if po_int and (po_int in s or s in po_int):
+            return True
+    return False
+
 
 def find_best_po_match(inv, po_df):
     inv_hdr = inv["invoice_header"]
@@ -548,7 +564,7 @@ if po_df is not None:
 
 if extracted_info is not None and po_df is not None:
     st.markdown("---")
-    st.subheader("EZOFIS AP AGENT Decision (OpenAI Only)")
+    st.subheader("EZOFIS AP AGENT Decision")
     if st.button("Make a decision (EZOFIS AP AGENT)"):
         tools = [
             Tool(