Update app.py

app.py

@@ -1,3 +1,6 @@
+HUGGING FACE ALL FUNCTIONALITIES WORKING 
+
+
 import os
 from paddleocr import PaddleOCR
 from PIL import Image
@@ -13,7 +16,7 @@ import kaleido  # Ensure kaleido is imported
 
 # Attribute mappings: readable names to Salesforce API names
 ATTRIBUTE_MAPPING = {
-    "Productname": "Productname__c",
+    "Product name": "Productname__c",
     "Colour": "Colour__c",
     "Motortype": "Motortype__c",
     "Frequency": "Frequency__c",
@@ -161,6 +164,8 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
         # Mapping mode and entry_type to Salesforce object and field
         object_name = None
         field_name = None
+        product_field_name = "Product_Name__c"
+        model_field_name = "Modal_Name__c"  # Correct field for model name
 
         if mode == "Entry":
             if entry_type == "Sales":
@@ -169,72 +174,60 @@ 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_name = "Quantity_Sold__c"
+            elif entry_type == "Non-Sales":
+                object_name = "Un_Billable__c"
+                field_name = "Sold_Out__c"
 
-            # Get valid fields from Salesforce object schema
-            sf_object = sf.__getattr__(object_name)
-            schema = sf_object.describe()
-            valid_fields = {field["name"] for field in schema["fields"]}
-
-            # Extract attributes from the extracted text
-            attributes = extract_attributes(extracted_text)
+        if not object_name or not field_name:
+            return "Invalid mode or entry type."
 
-            # Ensure Product Name is extracted and mapped correctly
-            product_name = match_product_name(extracted_text)  # Match product name using fuzzy logic
-            if not product_name:
-                return "❌ Product Name could not be extracted from the image. Please check the image."
+        # 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"]}
 
-            # Convert extracted keys to match Salesforce API field names
-            mapped_attributes = {}
-            for key, value in attributes.items():
-                sf_field_name = ATTRIBUTE_MAPPING.get(key, key.replace(" ", "_") + "__c")  # Convert to Salesforce format
-                if sf_field_name in valid_fields:
-                    mapped_attributes[sf_field_name] = value  # Only keep valid fields
+        # Extract product name and attributes
+        product_name = match_product_name(extracted_text)
+        attributes = extract_attributes(extracted_text)
+        model_name = attributes.get("Model Name", "").strip()
 
-            # Ensure Product Name and Quantity are added to mapped attributes
-            if entry_type == "Sales":
-                mapped_attributes["Product_Name__c"] = product_name  # Product Name field for Sales
-            elif entry_type == "Non-Sales":
-                mapped_attributes["Productname__c"] = product_name  # Product Name field for Non-Sales
-            mapped_attributes[field_name] = quantity  # Quantity field
+        if not product_name:
+            return "Product name could not be matched from the extracted text."
 
-            # Ensure at least one valid field exists
-            if not mapped_attributes:
-                return "❌ No valid attributes found to export."
+        attributes["Product name"] = product_name
 
-            # Creating a new record with all valid attributes
-            sf_object.create(mapped_attributes)
-            return f"✅ Record created in {object_name} with extracted valid attributes, Product Name: {product_name}, and Quantity: {quantity}."
+        # Handling "Exit" Mode (Updating Records)
+        if mode == "Exit":
+            # Query should only match exact product name or exact model name
+            query_conditions = []
+            if model_name:
+                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"
+            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}."
+            else:
+                return f"❌ No matching record found for product '{product_name}' ({model_name}) in {object_name}."
+
+        # Handling "Entry" Mode (Creating Records)
+        else:
+            filtered_attributes = filter_valid_attributes(attributes, valid_fields)
+            filtered_attributes[field_name] = quantity
+            sf_object.create(filtered_attributes)
+            return f"✅ Data successfully exported to Salesforce object {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: