Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import os
 from paddleocr import PaddleOCR
 from PIL import Image
@@ -165,6 +164,7 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
         field_names = []
         product_field_name = "Product_Name__c"
         model_field_name = "Model_Name__c"  # Correct field for model name
+
         if mode == "Entry":
             if entry_type == "Sales":
                 object_name = "VENKATA_RAMANA_MOTORS__c"
@@ -172,18 +172,18 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
             elif entry_type == "Non-Sales":
                 object_name = "UNBILLING_DATA__c"
                 field_name = "TotalQuantity__c"
-            elif mode == "Exit":
-                if entry_type == "Sales":
-                    object_name = "Inventory_Management__c"
-                    field_names = ["Quantity_Sold__c", "leftstock__c"]
-                elif entry_type == "Non-Sales":
-                    object_name = "Un_Billable__c"
-                    field_name = "Sold_Out__c"
-                    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
+        elif mode == "Exit":
+            if entry_type == "Sales":
+                object_name = "Inventory_Management__c"
+                field_names = ["Quantity_Sold__c", "LeftStock__c"]
+            elif entry_type == "Non-Sales":
+                object_name = "Un_Billable__c"
+                field_name = "Sold_Out__c"
+
+        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"]}
@@ -206,14 +206,14 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
                 query_conditions.append(f"{model_field_name} = '{model_name}'")
             query_conditions.append(f"{product_field_name} = '{product_name}'")
 
-            query = f"SELECT Id, {field_name} FROM {object_name} WHERE {' OR '.join(query_conditions)} LIMIT 1"
+            query = f"SELECT Id, {', '.join(field_names)} FROM {object_name} WHERE {' OR '.join(query_conditions)} LIMIT 1"
             response = sf.query(query)
 
             if response["records"]:
                 record_id = response["records"][0]["Id"]
-                updated_quantity = quantity  # Overwrite the quantity
-                sf_object.update(record_id, {field_name: updated_quantity})
-                return f"✅ Updated record for product '{product_name}' ({model_name}) in {object_name}. New {field_name}: {updated_quantity}."
+                updated_fields = {field: quantity for field in field_names}
+                sf_object.update(record_id, 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 '{product_name}' ({model_name}) in {object_name}."
 
@@ -226,6 +226,7 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
 
     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:
@@ -322,5 +323,4 @@ def app():
     ], ["📥 OCR Processing", "📊 Salesforce Data Export"])
 
 if __name__ == "__main__":
-    app().launch(share=True)
-
+    app().launch(share=True)