Update app.py

app.py

@@ -1,3 +1,6 @@
+HUGGING FACE ALL FUNCTIONALITIES WORKING 
+
+
 import os
 from paddleocr import PaddleOCR
 from PIL import Image
@@ -158,83 +161,73 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
             security_token=SALESFORCE_SECURITY_TOKEN
         )
 
-        # Function to extract attributes and their values
-        def extract_attributes(extracted_text):
-            attributes = {}
-            
-            for readable_attr, sf_attr in ATTRIBUTE_MAPPING.items():
-                pattern = rf"{re.escape(readable_attr)}[:\-]?\s*(.+)"
-                match = re.search(pattern, extracted_text, re.IGNORECASE)
-                if match:
-                    attributes[readable_attr] = match.group(1).strip()
-                    
-                    return attributes
-                    
-                    # Function to filter attributes for valid Salesforce fields
-                    def filter_valid_attributes(attributes, valid_fields):
-                        return {ATTRIBUTE_MAPPING[key]: value for key, value in attributes.items() if ATTRIBUTE_MAPPING[key] in valid_fields}
-                        # Function to interact with Salesforce based on mode and type
-                        def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
-                            try:
-                                sf = Salesforce(
-                                    username=SALESFORCE_USERNAME,
-                                    password=SALESFORCE_PASSWORD,
-                                    security_token=SALESFORCE_SECURITY_TOKEN
-                                )
-                                
-                                # Mapping mode and entry_type to Salesforce object and field
-                                object_name = None
-                                field_name = None
-                                product_field_name = "Product_Name__c"
-                                if mode == "Entry":
-                                    if entry_type == "Sales":
-                                        object_name = "VENKATA_RAMANA_MOTORS__c"
-                                        field_name = "Quantity__c"
-                                    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"
-                                            if not object_name or not field_name:
-                                                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"]}
-                                                
-                                                # Extract product name from the extracted text
-                                                product_name = match_product_name(extracted_text)
-                                                if not product_name:
-                                                    return "Product name could not be matched from the extracted text."
-                                                    # Handling "Exit" Mode (Updating Records)
-                                                    if mode == "Exit":
-                                                        query = f"SELECT Id, {field_name} FROM {object_name} WHERE {product_field_name} = '{product_name}' 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}' in {object_name}. New {field_name}: {updated_quantity}."
-                                                        else:
-                                                            return f"❌ No matching record found for product '{product_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)}"
+        # 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":
+                object_name = "VENKATA_RAMANA_MOTORS__c"
+                field_name = "Quantity__c"
+            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"
+
+        if not object_name or not field_name:
+            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"]}
+
+        # Extract product name and attributes
+        product_name = match_product_name(extracted_text)
+        attributes = extract_attributes(extracted_text)
+        model_name = attributes.get("Model Name", "").strip()
+
+        if not product_name:
+            return "Product name could not be matched from the extracted text."
 
+        attributes["Product name"] = product_name
+
+        # 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)}"
 # Function to pull structured data from Salesforce and display as a table
 def pull_data_from_salesforce():
     try:
@@ -332,3 +325,4 @@ def app():
 
 if __name__ == "__main__":
     app().launch(share=True)
+