Update app.py

app.py

@@ -93,8 +93,8 @@ def detect_anomalies(df):
     df["is_anomaly"] = model.fit_predict(X_scaled)
     return df
 
-def calculate_fraud_score(amount, is_anomaly, items_listed):
-    """Calculate fraud score based on amount, anomaly, and items."""
+def calculate_fraud_score(amount, is_anomaly, text_length):
+    """Calculate fraud score based on amount, anomaly, and text length."""
     score = 0.0
     reasoning = []
 
@@ -109,9 +109,9 @@ def calculate_fraud_score(amount, is_anomaly, items_listed):
         score += 30
         reasoning.append("Invoice flagged as an anomaly.")
 
-    if len(items_listed.split()) > 100:
+    if text_length > 500:
         score += 10
-        reasoning.append("Excessive number of items listed.")
+        reasoning.append("Excessive text length in invoice.")
 
     return min(score, 100), "; ".join(reasoning)
 
@@ -123,33 +123,31 @@ def process_invoice(pdf_file):
 
     vendor_name, amount = extract_entities(text)
     invoice_date = datetime.now().date()
-    items_listed = text[:500]
+    text_length = len(text)
 
     data = {
         "invoice_id": str(uuid.uuid4()),
         "vendor_name": vendor_name,
         "amount": amount,
         "invoice_date": invoice_date,
-        "items_listed": items_listed
+        "text_length": text_length
     }
     df = pd.DataFrame([data])
 
     df = detect_anomalies(df)
 
     fraud_score, fraud_reasoning = calculate_fraud_score(
-        df["amount"].iloc[0], df["is_anomaly"].iloc[0], items_listed
+        df["amount"].iloc[0], df["is_anomaly"].iloc[0], text_length
     )
 
     output = {
         "Invoice_Record__c": {
-            "Vendor_Name__c": vendor_name,
+            "Name": vendor_name,  # Map vendor_name to Name field
             "Invoice_Amount__c": amount,
             "Invoice_Date__c": str(invoice_date),
-            "Items_Listed__c": items_listed,
             "Fraud_Score__c": fraud_score,
             "Fraud_Reasoning__c": fraud_reasoning,
             "Flagged__c": fraud_score > 50,
-            "Reviewed_By__c": None,
             "Status__c": "Flagged" if fraud_score > 50 else "Cleared"
         },
         "Entities": {
@@ -168,14 +166,12 @@ def process_invoice(pdf_file):
     if sf is not None:
         try:
             sf.Invoice_Record__c.create({
-                "Vendor_Name__c": vendor_name,
+                "Name": vendor_name,  # Map vendor_name to Name field
                 "Invoice_Amount__c": amount,
                 "Invoice_Date__c": str(invoice_date),
-                "Items_Listed__c": items_listed,
                 "Fraud_Score__c": fraud_score,
                 "Fraud_Reasoning__c": fraud_reasoning,
                 "Flagged__c": fraud_score > 50,
-                "Reviewed_By__c": None,
                 "Status__c": "Flagged" if fraud_score > 50 else "Cleared"
             })
             print("Salesforce record created successfully.")
@@ -203,4 +199,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()