Update app.py

app.py

@@ -150,80 +150,70 @@ 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
-        )
-
-        # Define the correct object and field mappings
-        object_mapping = {
-            ("Entry", "Sales"): ("VENKATA_RAMANA_MOTORS__c", "Quantity__c"),
-            ("Entry", "Non-Sales"): ("UNBILLING_DATA__c", "TotalQuantity__c"),
-            ("Exit", "Sales"): ("Inventory_Management__c", "Quantity_Sold__c"),
-            ("Exit", "Non-Sales"): ("Un_Billable__c", "Sold_Out__c"),
-        }
-
-        object_name, field_name = object_mapping.get((mode, entry_type), (None, None))
-        if not object_name or not field_name:
-            return "❌ Invalid mode or entry type."
-
-        product_field_name = "Product_Name__c"
-        model_field_name = "Modal_Name__c"
-
-        # Extract product name and model name using OCR and fuzzy matching
-        product_name = match_product_name(extracted_text)
-        attributes = extract_attributes(extracted_text)
-        model_name = attributes.get("Model Name", "").strip()
-
-        if not product_name:
-            return "❌ Could not match a valid Product Name from the image."
-
-        attributes["Product name"] = product_name
-
-        # Standardize case and remove extra spaces for better accuracy
-        product_name = product_name.strip().lower()
-        model_name = model_name.lower()
-
-        # Get valid fields for the object
-        valid_fields = {field["name"] for field in sf.__getattr__(object_name).describe()["fields"]}
-
-        # 🚀 Exit Mode: Find Exact Matching Product or Model & Update Quantity
-        if mode == "Exit":
-            query = f"""
-                SELECT Id, {field_name} FROM {object_name}
-                WHERE LOWER({product_field_name}) = '{product_name}'
-                OR LOWER({model_field_name}) = '{model_name}'
-                LIMIT 1
-            """
-            response = sf.query(query)
-
-            if response["records"]:  # ✅ If record exists, update quantity
-                record_id = response["records"][0]["Id"]
-                existing_quantity = response["records"][0].get(field_name, 0)
-                updated_quantity = existing_quantity + quantity  # Increase quantity
-
-                sf.__getattr__(object_name).update(record_id, {field_name: updated_quantity})
-
-                return f"✅ Updated '{product_name}' in {object_name}. New {field_name}: {updated_quantity}."
-            else:  # ❌ If no match, create a new record
-                filtered_attributes = {ATTRIBUTE_MAPPING[key]: value for key, value in attributes.items() if ATTRIBUTE_MAPPING[key] in valid_fields}
-                filtered_attributes[field_name] = quantity
-                sf.__getattr__(object_name).create(filtered_attributes)
-
-                return f"🆕 Created new record for '{product_name}' in {object_name} with {field_name}: {quantity}."
-
-        # 🚀 Entry Mode: Always Create a New Record
-        filtered_attributes = {ATTRIBUTE_MAPPING[key]: value for key, value in attributes.items() if ATTRIBUTE_MAPPING[key] in valid_fields}
-        filtered_attributes[field_name] = quantity
-        sf.__getattr__(object_name).create(filtered_attributes)
+def export_entry_sales(sf, product_data):
+    """
+    Exports data to VENKATA_RAMANA_MOTORS__c and UNBILLING_DATA__c objects when entry and sales are clicked.
+    """
+    venkata_data = {
+        'Product_Name__c': product_data['Product_Name__c'],
+        'Modal_Name__c': product_data['Modal_Name__c'],
+        'Quantity__c': product_data['Quantity']
+    }
+    sf.VENKATA_RAMANA_MOTORS__c.create(venkata_data)
+    
+    unbilling_data = {
+        'Product_Name__c': product_data['Product_Name__c'],
+        'Modal_Name__c': product_data['Modal_Name__c'],
+        'TotalQuantity__c': product_data['Quantity']
+    }
+    sf.UNBILLING_DATA__c.create(unbilling_data)
+    
+    print("Entry and Sales data exported successfully.")
+
+def update_exit_sales(sf, product_data):
+    """
+    Updates Inventory_Management__c and Un_Billable__c objects when exit mode sales is clicked.
+    """
+    # Find matching record in Inventory_Management__c
+    inventory_records = sf.query(f"""
+        SELECT Id, Quantity_Sold__c FROM Inventory_Management__c 
+        WHERE Product_Name__c = '{product_data['Product_Name__c']}' 
+        OR Modal_Name__c = '{product_data['Modal_Name__c']}'
+    """)
+    
+    if inventory_records['records']:
+        inventory_id = inventory_records['records'][0]['Id']
+        new_quantity_sold = inventory_records['records'][0]['Quantity_Sold__c'] + product_data['Quantity']
+        sf.Inventory_Management__c.update(inventory_id, {'Quantity_Sold__c': new_quantity_sold})
+        print("Inventory Management updated successfully.")
+    
+    # Find matching record in Un_Billable__c
+    unbillable_records = sf.query(f"""
+        SELECT Id, Sold_Out__c FROM Un_Billable__c 
+        WHERE Product_Name__c = '{product_data['Product_Name__c']}' 
+        OR Model_Name__c = '{product_data['Modal_Name__c']}'
+    """)
+    
+    if unbillable_records['records']:
+        unbillable_id = unbillable_records['records'][0]['Id']
+        new_sold_out_quantity = unbillable_records['records'][0]['Sold_Out__c'] + product_data['Quantity']
+        sf.Un_Billable__c.update(unbillable_id, {'Sold_Out__c': new_sold_out_quantity})
+        print("Un_Billable updated successfully.")
+
+# Example usage
+sf = Salesforce(username='your_username', password='your_password', security_token='your_security_token')
+
+product_data = {
+    'Product_Name__c': 'Example Product',
+    'Modal_Name__c': 'Example Model',
+    'Quantity': 10
+}
 
-        return f"✅ Successfully exported data to Salesforce object {object_name}."
+# When clicking entry and sales
+export_entry_sales(sf, product_data)
 
-    except Exception as e:
-        return f"❌ Error interacting with Salesforce: {str(e)}"
+# When clicking exit mode sales
+update_exit_sales(sf, product_data)
 
 # Function to pull structured data from Salesforce and display as a table
 def pull_data_from_salesforce():