Update app.py

app.py

@@ -15,64 +15,15 @@ import kaleido  # Ensure kaleido is imported
 ATTRIBUTE_MAPPING = {
     "Product name": "Productname__c",
     "Colour": "Colour__c",
-    "Motortype": "Motortype__c",
     "Frequency": "Frequency__c",
-    "Grossweight": "Grossweight__c",
-    "Ratio": "Ratio__c",
-    "MotorFrame": "Motorframe__c",
     "Model": "Model__c",
     "Speed": "Speed__c",
     "Quantity": "Quantity__c",
     "Voltage": "Voltage__c",
-    "Material": "Material__c",
     "Type": "Type__c",
-    "Horsepower": "Horsepower__c",
-    "Consignee": "Consignee__c",
-    "LOT": "LOT__c",
     "Stage": "Stage__c",
     "Outlet": "Outlet__c",
-    "Serialnumber": "Serialnumber__c",
-    "HeadSize": "Headsize__c",
-    "Deliverysize": "Deliverysize__c",
     "Phase": "Phase__c",
-    "Size": "Size__c",
-    "MRP": "MRP__c",
-    "Usebefore": "Usebefore__c",
-    "Height": "Height__c",
-    "MaximumDischarge Flow": "Maximumdischargeflow__c",
-    "DischargeRange": "Dischargeflow__c",
-    "Assembledby": "Manufacturer__c",
-    "Manufacturedate": "Manufacturedate__c",
-    "Companyname": "Companyname__c",
-    "Customercarenumber": "Customercarenumber__c",
-    "SellerAddress": "Selleraddress__c",
-    "Selleremail": "Selleremail__c",
-    "GSTIN": "GSTIN__c",
-    "Totalamount": "Totalamount__c",
-    "Paymentstatus": "Paymentstatus__c",
-    "Paymentmethod": "Paymentstatus__c",
-    "Invoicedate": "Manufacturedate__c",
-    "Warranty": "Warranty__c",
-    "Brand": "Brand__c",
-    "Motorhorsepower": "Motorhorsepower__c",
-    "Power": "Power__c",
-    "Motorphase": "Motorphase__c",
-    "Enginetype": "Enginetype__c",
-    "Tankcapacity": "Tankcapacity__c",
-    "Head": "Head__c",
-    "Usage/Application": "Usage_Application__c",
-    "Volts": "volts__c",
-    "Hertz": "Hertz__c",
-    "Frame": "frame__c",
-    "Mounting": "Mounting__c",
-    "Tollfreenumber": "Tollfreenumber__c",
-    "Pipesize": "Pipesize__c",
-    "Manufacturer": "Manufacturer__c",
-    "Office": "Office__c",
-    "SRnumber": "SRnumber__c",
-    "TypeOfEndUse": "TypeOfEndUse__c",
-    "Model Name": "Model_Name_Number__c",
-    "coolingmethod": "coolingmethod__c",
     "H.P.": "H_p__c"
 }
 
@@ -141,6 +92,8 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
         field_names = []
         product_field_name = "Productname__c"
         model_field_name = "Model__c"
+        stage_field_name = "Stage__c"
+        hp_field_name = "H_p__c"
 
         if mode == "Entry":
             if entry_type == "Sales":
@@ -153,11 +106,9 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
             if entry_type == "Sales":
                 object_name = "Inventory_Management__c"
                 field_names = ["Quantity_Sold__c", "soldstock__c"]
-                model_field_name = "Model__c"
             elif entry_type == "Non-Sales":
                 object_name = "Un_Billable__c"
                 field_names = ["Sold_Out__c", "soldstock__c"]
-                model_field_name = "Model__c"
 
         if not object_name or (not field_name and not field_names):
             return "Invalid mode or entry type."
@@ -171,6 +122,8 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
         product_name = match_product_name(extracted_text)
         attributes = extract_attributes(extracted_text)
         model_name = attributes.get("Model Name", "").strip()
+        stage = attributes.get("Stage", "").strip()
+        hp = attributes.get("H.P.", "").strip()
 
         if not product_name:
             return "Product name could not be matched from the extracted text."
@@ -179,22 +132,35 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
 
         # Handling "Exit" Mode (Updating Records)
         if mode == "Exit":
-            # Query should only match exact product name or exact model name
-            query_conditions = []
+            # Query should match exact product name, model name, stage, and hp if available
+            query_conditions = [f"{product_field_name} = '{product_name}'"]
             if model_name:
-                query_conditions.append(f"{model_field_name} = '{Model__c}'")
-            query_conditions.append(f"{product_field_name} = '{Productname__c}'")
+                query_conditions.append(f"{model_field_name} = '{model_name}'")
+            if stage:
+                query_conditions.append(f"{stage_field_name} = '{stage}'")
+            if hp:
+                query_conditions.append(f"{hp_field_name} = '{hp}'")
 
-            query = f"SELECT Id, {', '.join(field_names)} FROM {object_name} WHERE {' OR '.join(query_conditions)} LIMIT 1"
+            query = f"SELECT Id, {', '.join(field_names)} FROM {object_name} WHERE {' AND '.join(query_conditions)} LIMIT 1"
             response = sf.query(query)
 
             if response["records"]:
                 record_id = response["records"][0]["Id"]
                 updated_fields = {field: quantity for field in field_names}
                 sf_object.update(record_id, updated_fields)
-                return f"✅ Updated record for product '{Productname__c}' ({Model__c}) in {object_name}. Updated fields: {updated_fields}."
+                return f"✅ Updated record for product '{product_name}' ({model_name}) in {object_name}. Updated fields: {updated_fields}."
             else:
-                return f"❌ No matching record found for product '{Productname__c}' ({Model__c}) in {object_name}."
+                # If no matching record found with all conditions, try with only product name
+                query_conditions = [f"{product_field_name} = '{product_name}'"]
+                query = f"SELECT Id, {', '.join(field_names)} FROM {object_name} WHERE {' AND '.join(query_conditions)} LIMIT 1"
+                response = sf.query(query)
+                if response["records"]:
+                    record_id = response["records"][0]["Id"]
+                    updated_fields = {field: quantity for field in field_names}
+                    sf_object.update(record_id, updated_fields)
+                    return f"✅ Updated record for product '{product_name}' in {object_name}. Updated fields: {updated_fields}."
+                else:
+                    return f"❌ No matching record found for product '{product_name}' in {object_name}."
 
         # Handling "Entry" Mode (Creating Records)
         else: