Update app.py

app.py

@@ -135,7 +135,6 @@ def interact_with_salesforce(mode, entry_type, quantity, attributes):
             security_token=SALESFORCE_SECURITY_TOKEN
         )
 
-        # Mapping mode and entry_type to Salesforce object and field
         object_name = None
         field_name = None
         field_names = []
@@ -162,23 +161,19 @@ def interact_with_salesforce(mode, entry_type, quantity, attributes):
         if not object_name or (not field_name and not field_names):
             return "Invalid mode or entry type."
 
-        # Get valid fields for the specified Salesforce object
         sf_object = sf.__getattr__(object_name)
         schema = sf_object.describe()
         valid_fields = {field["name"] for field in schema["fields"]}
 
-        # Filter attributes for valid Salesforce fields
         filtered_attributes = filter_valid_attributes(attributes, valid_fields)
 
-        # Handling "Exit" Mode (Updating Records)
         if mode == "Exit":
-            # Query should match exact product name, model name, stage, and hp if available
             query_conditions = [f"{product_field_name} = '{attributes['Product name']}'"]
-            if "Model Name" in attributes:
+            if "Model Name" in attributes and attributes["Model Name"]:
                 query_conditions.append(f"{model_field_name} = '{attributes['Model Name']}'")
-            if "Stage" in attributes:
+            if "Stage" in attributes and attributes["Stage"]:
                 query_conditions.append(f"{stage_field_name} = '{attributes['Stage']}'")
-            if "H.P." in attributes:
+            if "H.P." in attributes and attributes["H.P."]:
                 query_conditions.append(f"{hp_field_name} = '{attributes['H.P.']}'")
 
             query = f"SELECT Id, {', '.join(field_names)} FROM {object_name} WHERE {' AND '.join(query_conditions)} LIMIT 1"
@@ -192,7 +187,6 @@ def interact_with_salesforce(mode, entry_type, quantity, attributes):
             else:
                 return f"❌ No matching record found for product '{attributes['Product name']}' in {object_name}."
 
-        # Handling "Entry" Mode (Creating Records)
         else:
             filtered_attributes[field_name] = quantity
             sf_object.create(filtered_attributes)