Update app.py

app.py

@@ -149,155 +149,76 @@ def extract_attributes(extracted_text):
 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 handle sales entry
-def add_sales_entry(data):
+#📊  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,
+            security_token=SALESFORCE_SECURITY_TOKEN
         )
 
-        object_name = "VENKATA_RAMANA_MOTORS__c"
-        sf_object = sf.__getattr__(object_name)
-
-        schema = sf_object.describe()
-        valid_fields = {field["name"] for field in schema["fields"]}
+        # Mapping mode and entry_type to Salesforce object and field
+        object_name = None
+        field_name = None
+        product_field_name = "Product_Name__c"  # Correct field for product name in the object
+        model_field_name = "Modal_Name__c"  # Correct field for model name in the object
 
-        filtered_record = {k: v for k, v in data.items() if k in valid_fields and v is not None}
-        sf_object.create(filtered_record)
-        return f"Sales data successfully added to {object_name}."
-    except Exception as e:
-        return f"Error adding sales data to VENKATA_RAMANA_MOTORS__c: {str(e)}"
+        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"
+                product_field_name = "Product_Name__c"
+                model_field_name = "Modal_Name__c"
+                field_name = "Quantity_Sold__c"
+            elif entry_type == "Non-Sales":
+                object_name = "Un_Billable__c"
+                product_field_name = "Product_Name__c"
+                model_field_name = "Model_Name__c"
+                field_name = "Sold_Out__c"
 
-# Function to handle non-sales entry
-def add_non_sales_entry(data):
-    try:
-        sf = Salesforce(
-            username=SALESFORCE_USERNAME,
-            password=SALESFORCE_PASSWORD,
-            security_token=SALESFORCE_SECURITY_TOKEN,
-        )
+        if not object_name or not field_name:
+            return "Invalid mode or entry type."
 
-        object_name = "UNBILLING_DATA__c"
+        # 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"]}
 
-        # Add TotalQuantity__c to the data for non-sales
-        if "TotalQuantity__c" in valid_fields and "Quantity__c" in data:
-            data["TotalQuantity__c"] = data["Quantity__c"]
-
-        filtered_record = {k: v for k, v in data.items() if k in valid_fields and v is not None}
-        sf_object.create(filtered_record)
-        return f"Non-sales data successfully added to {object_name}."
-    except Exception as e:
-        return f"Error adding non-sales data to UNBILLING_DATA__c: {str(e)}"
-
-# Function to handle exit mode - sales
-def subtract_stock_from_inventory(data):
-    try:
-        sf = Salesforce(
-            username=SALESFORCE_USERNAME,
-            password=SALESFORCE_PASSWORD,
-            security_token=SALESFORCE_SECURITY_TOKEN,
-        )
-
-        object_name = "Inventory_Management__c"
-        sf_object = sf.__getattr__(object_name)
-
-        product_name = data.get("Productname__c")
-        quantity = data.get("Quantity__c", 0)
+        # Extract product name or model number
+        product_name = match_product_name(extracted_text)
+        attributes = extract_attributes(extracted_text)
 
         if not product_name:
-            return "Product name is missing in the data. Cannot update stock."
-
-        # Query existing stock record
-        query = f"SELECT Id, Quantity_Sold__c FROM {object_name} WHERE Product_Name__c = '{product_name}' LIMIT 1"
-        response = sf.query(query)
-
-        if not response["records"]:
-            return f"No stock found for product '{product_name}'. Cannot update stock."
-
-        record_id = response["records"][0]["Id"]
-        current_quantity_sold = response["records"][0].get("Quantity_Sold__c", 0)
-
-        # Update the quantity sold
-        updated_quantity_sold = current_quantity_sold + quantity
+            return "Product name could not be matched from the extracted text."
 
-        sf_object.update(record_id, {"Quantity_Sold__c": updated_quantity_sold})
-
-        return f"Stock updated successfully in exit mode. Quantity sold for product '{product_name}': {updated_quantity_sold}."
-    except Exception as e:
-        return f"Error updating stock in Inventory_Management__c: {str(e)}"
-
-# Function to handle exit mode - non-sales
-def subtract_stock_from_non_sales(data):
-    try:
-        sf = Salesforce(
-            username=SALESFORCE_USERNAME,
-            password=SALESFORCE_PASSWORD,
-            security_token=SALESFORCE_SECURITY_TOKEN,
-        )
-
-        object_name = "Un_Billable__c"
-        sf_object = sf.__getattr__(object_name)
-
-        schema = sf_object.describe()
-        valid_fields = {field["name"] for field in schema["fields"]}
-
-        # Add Sold_Out__c to the data for non-sales exit
-        if "Sold_Out__c" in valid_fields and "Quantity__c" in data:
-            data["Sold_Out__c"] = data["Quantity__c"]
-
-        filtered_record = {k: v for k, v in data.items() if k in valid_fields and v is not None}
-        sf_object.create(filtered_record)
-        return f"Non-sales data successfully updated in {object_name}."
-    except Exception as e:
-        return f"Error updating non-sales data in Un_Billable__c: {str(e)}"
-
-# Unified function for processing
-def process_image(image, quantity, mode, entry_type):
-    try:
-        extracted_text = extract_text(image)  # Extract text from image
-        attributes = find_attributes(extracted_text)  # Find attributes from extracted text
-
-        if "Productname__c" not in attributes or not attributes["Productname__c"]:
-            attributes["Productname__c"] = match_product_name(extracted_text)
-
-        attributes["Quantity__c"] = sanitize_numeric(quantity)
-
-        if not attributes:
-            return f"Extracted Text:\n{extracted_text}\n\nNo attributes found in the image.", None
-
-        numbered_output = "\n".join(
-            [f"{key.replace('__c', '').replace('_', ' ').title()}: {value}" for key, value in attributes.items()]
-        )
+        attributes["Product name"] = product_name
 
-        file_path = save_to_excel(attributes)
+        if mode == "Exit":
+            query = f"SELECT Id, {field_name} FROM {object_name} WHERE {product_field_name} = '{product_name}' OR {model_field_name} = '{attributes.get('Model Name', '')}' LIMIT 1"
+            response = sf.query(query)
 
-        if mode == "Entry":
-            if entry_type == "Sales":
-                message = add_sales_entry(attributes)
-            elif entry_type == "Non-Sales":
-                message = add_non_sales_entry(attributes)
-            else:
-                message = "Invalid entry type. Please select Sales or Non-Sales."
-        elif mode == "Exit":
-            if entry_type == "Sales":
-                message = subtract_stock_from_inventory(attributes)
-            elif entry_type == "Non-Sales":
-                message = subtract_stock_from_non_sales(attributes)
+            if response["records"]:
+                record_id = response["records"][0]["Id"]
+                updated_quantity = quantity  # Overwrite the quantity, don't add
+                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:
-                message = "Invalid entry type. Please select Sales or Non-Sales."
+                return f"No matching record found for product '{product_name}' in {object_name}."
         else:
-            message = "Invalid mode. Please select Entry or Exit."
-
-        return f"Extracted Text:\n{extracted_text}\n\n{numbered_output}\n\n{message}", file_path
+            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 during processing: {str(e)}", None
+        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():