Update app.py

app.py

@@ -158,23 +158,21 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
             security_token=SALESFORCE_SECURITY_TOKEN
         )
 
-        # Determine Salesforce Object and Quantity Field for Entry Mode
+        # Determine Salesforce Object and Quantity Field
         object_name = None
-        product_field_name = "Product_Name__c"  # Needs verification
-        model_field_name = "Modal_Name__c"  # Needs verification
+        product_field_name = "Productname__c"  # Ensure correct field name for Entry Mode
+        model_field_name = "Model__c"  # Ensure correct field name for Entry Mode
         quantity_field = None
 
         if mode == "Entry":
             if entry_type == "Sales":
                 object_name = "VENKATA_RAMANA_MOTORS__c"
-                quantity_field = "Quantity__c"  # Needs verification
-                product_field_name = "Productname__c"  # Adjusted based on ENTRY Mode fields
-                model_field_name = "Model__c"  # Adjusted based on ENTRY Mode fields
+                quantity_field = "Quantity__c"  # Check the correct field name in Salesforce
             elif entry_type == "Non-Sales":
                 object_name = "UNBILLING_DATA__c"
                 quantity_field = "TotalQuantity__c"
 
-        elif mode == "Exit":  # Exit Mode is working fine, so we don't touch it
+        elif mode == "Exit":  # Keep Exit Mode logic unchanged
             if entry_type == "Sales":
                 object_name = "Inventory_Management__c"
                 quantity_field = "Quantity_Sold__c"
@@ -193,11 +191,6 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
         if not product_name and not model_name:
             return "❌ No product name or model name detected. Please provide a valid product or model."
 
-        # Adjust the field names for Entry Mode to match correct fields in Salesforce
-        if mode == "Entry":
-            product_field_name = "Productname__c"  # Adjusted based on ENTRY Mode
-            model_field_name = "Model__c"  # Adjusted based on ENTRY Mode
-
         # Build Query to Find Matching Product
         query_conditions = []
         if product_name:
@@ -214,13 +207,13 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
             record_id = response["records"][0]["Id"]
             existing_quantity = response["records"][0].get(quantity_field, 0)
 
-            # Update quantity directly
-            updated_quantity = quantity  # Just set the new quantity without adding/subtracting
+            # Update quantity directly without modifying other fields
+            updated_quantity = quantity  # Directly set the new quantity
 
             sf.__getattr__(object_name).update(record_id, {quantity_field: updated_quantity})
             return f"✅ Successfully updated quantity for '{product_name or model_name}' in {object_name}. New {quantity_field}: {updated_quantity}."
         else:
-            return f"❌ No matching product found in {object_name}. Check field names or product/model data."
+            return f"❌ No matching product found in {object_name}. Please check the product/model name."
 
     except Exception as e:
         return f"❌ Error interacting with Salesforce: {str(e)}"