Update app.py

app.py

@@ -150,75 +150,91 @@ 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):
+# 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
-        )
+        sf = authenticate_salesforce()
+        object_name = "Inventory_Management__c"
+        product_name = data.get("Productname__c")
+        quantity = data.get("Quantity__c", 0)
 
-        # 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
+        if not product_name:
+            return "Product name is missing in the data. Cannot update stock."
 
-        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"
+        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)
+        updated_quantity_sold = current_quantity_sold + quantity
+
+        sf.__getattr__(object_name).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)}"
 
-        if not object_name or not field_name:
-            return "Invalid mode or entry type."
+# Function to handle exit mode - non-sales
+def subtract_stock_from_non_sales(data):
+    try:
+        sf = authenticate_salesforce()
+        object_name = "Un_Billable__c"
+        product_name = data.get("Productname__c")
+        quantity = data.get("Quantity__c", 0)
+
+        if not product_name or not quantity:
+            return "Missing product name or quantity. Cannot process non-sales update."
 
-        # Get valid fields for the specified Salesforce object
-        sf_object = sf.__getattr__(object_name)
-        schema = sf_object.describe()
+        schema = sf.__getattr__(object_name).describe()
         valid_fields = {field["name"] for field in schema["fields"]}
 
-        # Extract product name or model number
-        product_name = match_product_name(extracted_text)
-        attributes = extract_attributes(extracted_text)
+        if "Sold_Out__c" in valid_fields:
+            data["Sold_Out__c"] = quantity
 
-        if not product_name:
-            return "Product name could not be matched from the extracted text."
+        filtered_record = {k: v for k, v in data.items() if k in valid_fields and v is not None}
+        sf.__getattr__(object_name).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)}"
 
-        attributes["Product name"] = product_name
+# Unified function for processing
+def process_image(image, quantity, mode, entry_type):
+    try:
+        extracted_text = extract_text(image)
+        attributes = find_attributes(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 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 not attributes:
+            return f"Extracted Text:\n{extracted_text}\n\nNo attributes found in the image.", None
 
-            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}."
+        file_path = save_to_excel(attributes)
+
+        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)
             else:
-                return f"No matching record found for product '{product_name}' in {object_name}."
+                message = "Invalid entry type. Please select Sales or Non-Sales."
         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}."
+            message = "Invalid mode. Please select Entry or Exit."
 
+        return f"Extracted Text:\n{extracted_text}\n\n{message}", file_path
     except Exception as e:
-        return f"❌ Error interacting with Salesforce: {str(e)}"
+        return f"Error during processing: {str(e)}", None
 
 # Function to pull structured data from Salesforce and display as a table
 def pull_data_from_salesforce():