Update app.py

app.py

@@ -158,20 +158,23 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
             security_token=SALESFORCE_SECURITY_TOKEN
         )
 
-        # Determine Salesforce Object and Quantity Field
+        # Determine Salesforce Object and Quantity Field for Entry Mode
         object_name = None
-        product_field_name = "Product_Name__c"
-        model_field_name = "Modal_Name__c"
+        product_field_name = "Product_Name__c"  # Needs verification
+        model_field_name = "Modal_Name__c"  # Needs verification
         quantity_field = None
 
         if mode == "Entry":
             if entry_type == "Sales":
                 object_name = "VENKATA_RAMANA_MOTORS__c"
-                quantity_field = "Quantity__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
             elif entry_type == "Non-Sales":
                 object_name = "UNBILLING_DATA__c"
                 quantity_field = "TotalQuantity__c"
-        elif mode == "Exit":
+
+        elif mode == "Exit":  # Exit Mode is working fine, so we don't touch it
             if entry_type == "Sales":
                 object_name = "Inventory_Management__c"
                 quantity_field = "Quantity_Sold__c"
@@ -190,6 +193,11 @@ 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:
@@ -212,7 +220,7 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
             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}."
+            return f"❌ No matching product found in {object_name}. Check field names or product/model data."
 
     except Exception as e:
         return f"❌ Error interacting with Salesforce: {str(e)}"
@@ -235,6 +243,80 @@ def pull_data_from_salesforce():
        
         df = pd.DataFrame(records)
         df = df.drop(columns=['attributes'], errors='ignore')
+def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
+    try:
+        sf = Salesforce(
+            username=SALESFORCE_USERNAME,
+            password=SALESFORCE_PASSWORD,
+            security_token=SALESFORCE_SECURITY_TOKEN
+        )
+
+        # Determine Salesforce Object and Quantity Field for Entry Mode
+        object_name = None
+        product_field_name = "Product_Name__c"  # Needs verification
+        model_field_name = "Modal_Name__c"  # Needs verification
+        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
+            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
+            if entry_type == "Sales":
+                object_name = "Inventory_Management__c"
+                quantity_field = "Quantity_Sold__c"
+            elif entry_type == "Non-Sales":
+                object_name = "Un_Billable__c"
+                quantity_field = "Sold_Out__c"
+
+        if not object_name or not quantity_field:
+            return "Invalid mode or entry type."
+
+        # Extract Product Name or Model Name
+        product_name = match_product_name(extracted_text)
+        attributes = extract_attributes(extracted_text)
+        model_name = attributes.get("Model Name", "").strip()
+
+        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:
+            query_conditions.append(f"{product_field_name} = '{product_name}'")
+        if model_name:
+            query_conditions.append(f"{model_field_name} = '{model_name}'")
+
+        query_condition_string = " OR ".join(query_conditions)
+
+        query = f"SELECT Id, {quantity_field} FROM {object_name} WHERE {query_condition_string} LIMIT 1"
+        response = sf.query(query)
+
+        if response["records"]:
+            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
+
+            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."
+
+    except Exception as e:
+        return f"❌ Error interacting with Salesforce: {str(e)}"
        
         # Rename columns for better readability
         df.rename(columns={