Update app.py

app.py

@@ -210,33 +210,33 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
 
     except Exception as e:
         return f"❌ Error interacting with Salesforce: {str(e)}"
+elif mode == "Exit":
+if entry_type == "Sales":
+    object_name = "Inventory_Management__c"
+    field_name = "Quantity_Sold__c"
+elif entry_type == "Non-Sales":
+    object_name = "Un_Billable__c"
+    field_name = "Sold_Out__c"
+    
+    # Extract product name
+    product_name = match_product_name(extracted_text)
+    if not product_name:
+        return "Product name could not be matched from the extracted text."
+        
+        query = f"SELECT Id, {field_name} FROM {object_name} WHERE Product_Name__c = '{product_name}' LIMIT 1"
+        response = sf.query(query)
+        
+        if response["records"]:
+            record_id = response["records"][0]["Id"]
+            updated_quantity = quantity
+            sf.__getattr__(object_name).update(record_id, {field_name: updated_quantity})
+            return f"✅ Updated record for product '{product_name}' in {object_name}. New {field_name}: {updated_quantity}."
+        else:
+            return f"❌ No matching record found for product '{product_name}' in {object_name}."
+
+except Exception as e:
+return f"❌ Error interacting with Salesforce: {str(e)}"
 
-        elif mode == "Exit":
-            if entry_type == "Sales":
-                object_name = "Inventory_Management__c"
-                field_name = "Quantity_Sold__c"
-            elif entry_type == "Non-Sales":
-                object_name = "Un_Billable__c"
-                field_name = "Sold_Out__c"
-
-            # Extract product name
-            product_name = match_product_name(extracted_text)
-            if not product_name:
-                return "Product name could not be matched from the extracted text."
-
-            query = f"SELECT Id, {field_name} FROM {object_name} WHERE Product_Name__c = '{product_name}' LIMIT 1"
-            response = sf.query(query)
-
-            if response["records"]:
-                record_id = response["records"][0]["Id"]
-                updated_quantity = quantity
-                sf.__getattr__(object_name).update(record_id, {field_name: updated_quantity})
-                return f"✅ Updated record for product '{product_name}' in {object_name}. New {field_name}: {updated_quantity}."
-            else:
-                return f"❌ No matching record found for product '{product_name}' in {object_name}."
-
-    except Exception as e:
-        return f"❌ Error interacting with Salesforce: {str(e)}"
 # Function to pull structured data from Salesforce and display as a table
 def pull_data_from_salesforce():
     try: