Update app.py

app.py

@@ -177,20 +177,22 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
 
             # Extract attributes from the extracted text
             attributes = extract_attributes(extracted_text)
-            
-            # Convert extracted keys to match Salesforce API field names
-            mapped_attributes = {key.replace(" ", "_") + "__c": value for key, value in attributes.items()}
+             # Convert extracted keys to match Salesforce API field names
+            mapped_attributes = {}
+            for key, value in attributes.items():
+                field_name_sf = key.replace(" ", "_") + "__c"  # Convert to Salesforce format
+                if field_name_sf in valid_fields:
+                    mapped_attributes[field_name_sf] = value  # Only keep valid fields
 
-            # Filter attributes to only include valid fields
-            filtered_attributes = {k: v for k, v in attributes.items() if k in valid_fields}
-            filtered_attributes[field_name] = quantity  # Ensure Quantity is added
+            mapped_attributes[field_name] = quantity  # Ensure Quantity is added
 
-            if not filtered_attributes:
+            if not mapped_attributes:
                 return "No valid attributes found to export."
 
-              # Creating a new record with only valid attributes
-            sf_object.create(filtered_attributes)
+            # Creating a new record with only valid attributes
+            sf_object.create(mapped_attributes)
             return f"✅ Record created in {object_name} with extracted valid attributes and Quantity: {quantity}."
+
         elif mode == "Exit":
             if entry_type == "Sales":
                 object_name = "Inventory_Management__c"