Update app.py

app.py

@@ -1,15 +1,6 @@
 import os
 from dotenv import load_dotenv
-
-# Load environment variables from .env file
-load_dotenv()
-
-os.environ["CUDA_VISIBLE_DEVICES"] = ""  # Disable GPU usage
-os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"  # Disable oneDNN optimizations
-
 import logging
-logging.getLogger("transformers").setLevel(logging.ERROR)
-
 import pdfplumber
 import pandas as pd
 import numpy as np
@@ -21,9 +12,18 @@ import uuid
 from datetime import datetime
 import re
 import gradio as gr
-
 from simple_salesforce import Salesforce, SalesforceAuthenticationFailed
 
+# Load environment variables from .env file
+load_dotenv()
+
+# Configure environment for CPU usage
+os.environ["CUDA_VISIBLE_DEVICES"] = ""  # Disable GPU usage
+os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"  # Disable oneDNN optimizations
+
+# Set up logging to suppress transformers warnings
+logging.getLogger("transformers").setLevel(logging.ERROR)
+
 # Read Salesforce credentials from environment variables
 SF_USERNAME = os.getenv("SF_USERNAME")
 SF_PASSWORD = os.getenv("SF_PASSWORD")
@@ -33,7 +33,6 @@ print(f"Salesforce login info: username={SF_USERNAME}")
 
 # Salesforce connection with error handling
 try:
-    # Default domain is "login" (production); use domain="test" for sandbox/Developer Edition orgs
     sf = Salesforce(
         username=SF_USERNAME,
         password=SF_PASSWORD,
@@ -42,7 +41,7 @@ try:
     print("Salesforce login successful.")
 except SalesforceAuthenticationFailed as e:
     print(f"Salesforce authentication failed: {e}")
-    sf = None  # To avoid crashing later
+    sf = None
 
 # Initialize Hugging Face NER pipeline (force CPU)
 ner_pipeline = pipeline("ner", model="dslim/bert-base-NER", tokenizer="dslim/bert-base-NER", device=-1)
@@ -141,11 +140,11 @@ def process_invoice(pdf_file):
 
     output = {
         "Invoice_Record__c": {
-            "Name": vendor_name,  # Map vendor_name to Name field
+            "Name": vendor_name,
             "Invoice_Amount__c": amount,
             "Invoice_Date__c": str(invoice_date),
             "Fraud_Score__c": fraud_score,
-            "Fraud_Reasoning__c": fraud_reasoning,
+            "FRAUD_REASON__c": fraud_reasoning,  # Corrected field name
             "Flagged__c": fraud_score > 50,
             "Status__c": "Flagged" if fraud_score > 50 else "Cleared"
         },
@@ -158,18 +157,21 @@ def process_invoice(pdf_file):
 
     # Save to JSON file
     output_file = "fraud_detection_results.json"
-    with open(output_file, "w") as f:
-        json.dump([output], f, indent=2)
+    try:
+        with open(output_file, "w") as f:
+            json.dump([output], f, indent=2)
+    except Exception as e:
+        print(f"Error saving JSON file: {e}")
 
     # Create record in Salesforce if logged in successfully
     if sf is not None:
         try:
             sf.Invoice_Record__c.create({
-                "Name": vendor_name,  # Map vendor_name to Name field
+                "Name": vendor_name,
                 "Invoice_Amount__c": amount,
                 "Invoice_Date__c": str(invoice_date),
                 "Fraud_Score__c": fraud_score,
-                "Fraud_Reasoning__c": fraud_reasoning,
+                "FRAUD_REASON__c": fraud_reasoning,  # Corrected field name
                 "Flagged__c": fraud_score > 50,
                 "Status__c": "Flagged" if fraud_score > 50 else "Cleared"
             })
@@ -198,4 +200,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(share=False)  # share=False for local execution