Update app.py

app.py

@@ -149,76 +149,155 @@ 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 interact with Salesforce based on mode and type
-def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
+# Function to handle sales entry
+def add_sales_entry(data):
     try:
         sf = Salesforce(
             username=SALESFORCE_USERNAME,
             password=SALESFORCE_PASSWORD,
-            security_token=SALESFORCE_SECURITY_TOKEN
+            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"  # Correct field for product name in the object
-        model_field_name = "Modal_Name__c"  # Correct field for model name in the object
+        object_name = "VENKATA_RAMANA_MOTORS__c"
+        sf_object = sf.__getattr__(object_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"
-                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"
+        schema = sf_object.describe()
+        valid_fields = {field["name"] for field in schema["fields"]}
+
+        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 not object_name or not field_name:
-            return "Invalid mode or entry type."
+# 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,
+        )
 
-        # Get valid fields for the specified Salesforce object
+        object_name = "UNBILLING_DATA__c"
         sf_object = sf.__getattr__(object_name)
+
         schema = sf_object.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)
+        # 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)
 
         if not product_name:
-            return "Product name could not be matched from the extracted text."
+            return "Product name is missing in the data. Cannot update stock."
 
-        attributes["Product name"] = product_name
+        # 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
 
-        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)
+        sf_object.update(record_id, {"Quantity_Sold__c": updated_quantity_sold})
 
-            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}."
+        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()]
+        )
+
+        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{numbered_output}\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():