Update app.py

app.py

@@ -183,12 +183,6 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
             if not product_name:
                 return "❌ Product Name could not be extracted from the image. Please check the image."
 
-            # Assign Product Name to the correct Salesforce field
-            if entry_type == "Sales":
-                attributes["Product_Name__c"] = product_name  # For VENKATA_RAMANA_MOTORS__c
-            elif entry_type == "Non-Sales":
-                attributes["Productname__c"] = product_name  # For UNBILLING_DATA__c
-
             # Convert extracted keys to match Salesforce API field names
             mapped_attributes = {}
             for key, value in attributes.items():
@@ -197,7 +191,10 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
                     mapped_attributes[sf_field_name] = value  # Only keep valid fields
 
             # Ensure Product Name and Quantity are added to mapped attributes
-            mapped_attributes["Product_Name__c"] = product_name  # Product Name field
+            if entry_type == "Sales":
+                mapped_attributes["Product_Name__c"] = product_name  # Product Name field for Sales
+            elif entry_type == "Non-Sales":
+                mapped_attributes["Productname__c"] = product_name  # Product Name field for Non-Sales
             mapped_attributes[field_name] = quantity  # Quantity field
 
             # Ensure at least one valid field exists
@@ -209,14 +206,14 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
             return f"✅ Record created in {object_name} with extracted valid attributes, Product Name: {product_name}, and Quantity: {quantity}."
 
     except Exception as e:
-        return f"❌ Error interacting with Salesforce: {str(e)}"
-elif mode == "Exit":
-if entry_type == "Sales":
-    object_name = "Inventory_Management__c"
-    field_name = "Quantity_Sold__c"
-elif entry_type == "Non-Sales":
-    object_name = "Un_Billable__c"
-    field_name = "Sold_Out__c"
+        return f"❌ Error interacting with Salesforce: {str(e)}"elif mode == "Exit":
+        
+        if entry_type == "Sales":
+            object_name = "Inventory_Management__c"
+            field_name = "Quantity_Sold__c"
+        elif entry_type == "Non-Sales":
+            object_name = "Un_Billable__c"
+            field_name = "Sold_Out__c"
     
     # Extract product name
     product_name = match_product_name(extracted_text)