Update app.py

app.py

@@ -13,7 +13,7 @@ import kaleido  # Ensure kaleido is imported
 
 # Attribute mappings: readable names to Salesforce API names
 ATTRIBUTE_MAPPING = {
-    "Productname": "Productname__c",
+    "Product name": "Productname__c",
     "Colour": "Colour__c",
     "Motortype": "Motortype__c",
     "Frequency": "Frequency__c",
@@ -161,6 +161,8 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
         # 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":
@@ -169,40 +171,6 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
             elif entry_type == "Non-Sales":
                 object_name = "UNBILLING_DATA__c"
                 field_name = "TotalQuantity__c"
-
-            # Get valid fields from Salesforce object schema
-            sf_object = sf.__getattr__(object_name)
-            schema = sf_object.describe()
-            valid_fields = {field["name"] for field in schema["fields"]}
-
-            # Extract attributes from the extracted text
-            attributes = extract_attributes(extracted_text)
-
-                # Explicitly extract Product Name from the attributes
-            product_name = attributes.get("Productname", "").strip()  # Extract 'Productname' from the text
-            if product_name:
-                # Ensure that the Product Name is mapped to the right Salesforce field
-                if entry_type == "Sales":
-                    attributes["Product_Name__c"] = product_name  # For VENKATA_RAMANA_MOTORS__c
-                elif entry_type == "Non-Sales":
-                    attributes["Productname__c"] = product_name  # For UNBILLING_DATA__c
-                
-             # Convert extracted keys to match Salesforce API field names
-            mapped_attributes = {}
-            for key, value in attributes.items():
-                field_name_sf = key.replace(" ", "_") + "__c"  # Convert to Salesforce format
-                if field_name_sf in valid_fields:
-                    mapped_attributes[field_name_sf] = value  # Only keep valid fields
-
-            mapped_attributes[field_name] = quantity  # Ensure Quantity is added
-
-            if not mapped_attributes:
-                return "No valid attributes found to export."
-
-             # Creating a new record with all valid attributes
-            sf_object.create(mapped_attributes)
-            return f"✅ Record created in {object_name} with extracted valid attributes and Quantity: {quantity}."
-
         elif mode == "Exit":
             if entry_type == "Sales":
                 object_name = "Inventory_Management__c"
@@ -211,21 +179,49 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
                 object_name = "Un_Billable__c"
                 field_name = "Sold_Out__c"
 
-            # Extract product name
-            product_name = match_product_name(extracted_text)
-            if not product_name:
-                return "Product name could not be matched from the extracted text."
+        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"]}
 
-            query = f"SELECT Id, {field_name} FROM {object_name} WHERE Product_Name__c = '{product_name}' LIMIT 1"
+        # 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
-                sf.__getattr__(object_name).update(record_id, {field_name: updated_quantity})
-                return f"✅ Updated record for product '{product_name}' in {object_name}. New {field_name}: {updated_quantity}."
+                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}' in {object_name}."
+                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)}"
@@ -298,7 +294,7 @@ def app():
         gr.Interface(
             fn=process_image,
             inputs=[
-                gr.Image(type="numpy", label="📄 Upload Image"),
+                gr.Image(type="numpy", label="📄ᴜᴘʟᴏᴀᴅ ɪᴍᴀɢᴇ"),
                 gr.Dropdown(label="📌 Mode", choices=["Entry", "Exit"], value="Entry"),
                 gr.Radio(label="📦 Entry Type", choices=["Sales", "Non-Sales"], value="Sales"),
                 gr.Number(label="🔢 Quantity", value=1, interactive=True),
@@ -307,8 +303,8 @@ def app():
                 gr.Text(label="📝 Extracted Image Data"),
                 gr.Text(label="🚀 Result")
             ],
-            title="🏢 Inventory Management",
-            description="📦 Inventory Management System"
+            title="🏢 𝑽𝑬𝑵𝑲𝑨𝑻𝑨𝑹𝑨𝑴𝑨𝑵𝑨 𝑴𝑶𝑻𝑶𝑹𝑺",
+            description="📦 𝐈𝐍𝐕𝐄𝐍𝐓𝐎𝐑𝐘 𝐌𝐀𝐍𝐀𝐆𝐄𝐌𝐄𝐍𝐓"
         ),
         gr.Interface(
             fn=pull_data_from_salesforce,
@@ -325,5 +321,4 @@ def app():
     ], ["📥 OCR Processing", "📊 Salesforce Data Export"])
 
 if __name__ == "__main__":
-    app().launch(share=True)
-
+    app().launch(share=True)