Update app.py

app.py

@@ -167,26 +167,44 @@ def interact_with_salesforce(mode, entry_type, quantity, attributes):
 
         filtered_attributes = filter_valid_attributes(attributes, valid_fields)
 
-        if mode == "Exit":
-            query_conditions = [
-        f"{product_field_name} = '{attributes['Product name']}'",
-        f"{model_field_name} = '{attributes['Model Name']}'",
-        f"{stage_field_name} = '{attributes['Stage']}'",
-        f"{hp_field_name} = '{attributes['H.P.']}'"
-    ]
-
-    query = f"SELECT Id, {', '.join(field_names)} FROM {object_name} WHERE {' AND '.join(query_conditions)} LIMIT 1"
-    print(f"Executing query: {query}")  # Debugging line
-    response = sf.query(query)
-
-    if response["records"]:  # If a matching record is found
-        record_id = response["records"][0]["Id"]
-        updated_fields = {field: quantity for field in field_names}
-        print(f"Updating record ID {record_id} with fields {updated_fields}")  # Debugging line
-        sf_object.update(record_id, updated_fields)
-        return f"✅ Updated record for product '{attributes['Product name']}' in {object_name}. Updated fields: {updated_fields}."
+        try:
+    if mode == "Exit":
+        query_conditions = [
+            f"{product_field_name} = '{attributes['Product name']}'",
+            f"{model_field_name} = '{attributes['Model Name']}'",
+            f"{stage_field_name} = '{attributes['Stage']}'",
+            f"{hp_field_name} = '{attributes['H.P.']}'"
+        ]
+
+        query = f"SELECT Id, {', '.join(field_names)} FROM {object_name} WHERE {' AND '.join(query_conditions)} LIMIT 1"
+        print(f"Executing query: {query}")  # Debugging line
+
+        response = sf.query(query)  # Execute SOQL query
+
+        if response["records"]:  # If a matching record is found
+            record_id = response["records"][0]["Id"]
+            updated_fields = {field: quantity for field in field_names}  # Prepare update data
+            print(f"Updating record ID {record_id} with fields {updated_fields}")  # Debugging line
+
+            sf_object.update(record_id, updated_fields)  # Update record
+            result_message = f"✅ Updated record for product '{attributes['Product name']}' in {object_name}. Updated fields: {updated_fields}."
+        
+        else:
+            # No exact match found, create a new record
+            filtered_attributes = {
+                product_field_name: attributes["Product name"],
+                model_field_name: attributes["Model Name"],
+                stage_field_name: attributes["Stage"],
+                hp_field_name: attributes["H.P."],
+                field_names[0]: quantity  # Assuming there's only one field to update
+            }
+            print(f"Creating new record in {object_name} with attributes {filtered_attributes}")  # Debugging line
+            
+            sf_object.create(filtered_attributes)  # Create new record
+            result_message = f"✅ No exact match found. A new record has been created in {object_name}."
+
     else:
-        # No exact match found, so create a new record
+        # If mode is not "Exit", create a new record directly
         filtered_attributes = {
             product_field_name: attributes["Product name"],
             model_field_name: attributes["Model Name"],
@@ -195,8 +213,14 @@ def interact_with_salesforce(mode, entry_type, quantity, attributes):
             field_names[0]: quantity  # Assuming there's only one field to update
         }
         print(f"Creating new record in {object_name} with attributes {filtered_attributes}")  # Debugging line
-        sf_object.create(filtered_attributes)
-        return f"✅ No exact match found. A new record has been created in {object_name}."
+        
+        sf_object.create(filtered_attributes)  # Create new record
+        result_message = f"✅ Data successfully exported to Salesforce object {object_name}."
+
+except Exception as e:
+    result_message = f"❌ Error interacting with Salesforce: {str(e)}"
+
+return result_message
 
 # Function to process image, extract attributes, and allow editing
 def process_image(image, mode, entry_type, quantity):