Update app.py

app.py

@@ -11,44 +11,117 @@ from io import BytesIO
 from fuzzywuzzy import process
 import kaleido  # Ensure kaleido is imported
 
-# ✅ Attribute mappings: readable names to Salesforce API names
+# Attribute mappings: readable names to Salesforce API names
 ATTRIBUTE_MAPPING = {
     "Product name": "Productname__c",
-    "Type": "Type__c",
-    "model name": "Model Name",
+    "Colour": "Colour__c",
+    "Motortype": "Motortype__c",
+    "Frequency": "Frequency__c",
+    "Grossweight": "Grossweight__c",
+    "Ratio": "Ratio__c",
+    "MotorFrame": "Motorframe__c",
+    "Model": "Model__c",
+    "Speed": "Speed__c",
+    "Quantity": "Quantity__c",
     "Voltage": "Voltage__c",
-    "Outlet": "Outlet__c",
+    "Material": "Material__c",
+    "Type": "Type__c",
+    "Horsepower": "Horsepower__c",
+    "Consignee": "Consignee__c",
+    "LOT": "LOT__c",
     "Stage": "Stage__c",
-    "H.p": "H_p__c",
-    "Model": "Model__c"
+    "Outlet": "Outlet__c",
+    "Serialnumber": "Serialnumber__c",
+    "HeadSize": "Headsize__c",
+    "Deliverysize": "Deliverysize__c",
+    "Phase": "Phase__c",
+    "Size": "Size__c",
+    "MRP": "MRP__c",
+    "Usebefore": "Usebefore__c",
+    "Height": "Height__c",
+    "MaximumDischarge Flow": "Maximumdischargeflow__c",
+    "DischargeRange": "Dischargeflow__c",
+    "Assembledby": "Manufacturer__c",
+    "Manufacturedate": "Manufacturedate__c",
+    "Companyname": "Companyname__c",
+    "Customercarenumber": "Customercarenumber__c",
+    "SellerAddress": "Selleraddress__c",
+    "Selleremail": "Selleremail__c",
+    "GSTIN": "GSTIN__c",
+    "Totalamount": "Totalamount__c",
+    "Paymentstatus": "Paymentstatus__c",
+    "Paymentmethod": "Paymentstatus__c",
+    "Invoicedate": "Manufacturedate__c",
+    "Warranty": "Warranty__c",
+    "Brand": "Brand__c",
+    "Motorhorsepower": "Motorhorsepower__c",
+    "Power": "Power__c",
+    "Motorphase": "Motorphase__c",
+    "Enginetype": "Enginetype__c",
+    "Tankcapacity": "Tankcapacity__c",
+    "Head": "Head__c",
+    "Usage/Application": "Usage_Application__c",
+    "Volts": "volts__c",
+    "Hertz": "Hertz__c",
+    "Frame": "frame__c",
+    "Mounting": "Mounting__c",
+    "Tollfreenumber": "Tollfreenumber__c",
+    "Pipesize": "Pipesize__c",
+    "Manufacturer": "Manufacturer__c",
+    "Office": "Office__c",
+    "SRnumber": "SRnumber__c",
+    "TypeOfEndUse": "TypeOfEndUse__c",
+    "Model Name": "Model_Name_Number__c",
+    "coolingmethod": "coolingmethod__c"
 }
 
-# ✅ List of product names to match
+# List of product names to match
 PRODUCT_NAMES = [
-    "CG COMMERCIAL MOTORS", "Fusion", "Agroking", "Openwell"
+    "Centrifugal mono block pump", "SINGLE PHASE MOTOR STARTER", "EasyPact EZC 100",
+    "Openwell Submersible Pumpset", "Electric Motor", "Self Priming Pump",
+    "Control panel for single phase submersible pumps", "MOTOR", "Submersible pump set",
+    "Fusion submersible pump set", "DCT", "Shock proof water proof", "CG COMMERCIAL MOTORS", "Fusion",
+    "control panel for single phase submerisible pumps",
+    "single phase digital starter dry run and timer panel", "5HP AV1 XL Kirloskar Pump",
+    "Phase stainless steel submersible pump", "Submersible pump", "WB15X",
+    "Vtype self priming pump", "SP SHINE DISC", "havells submersible pump",
+    "Havells open well Submersible pump", "Bertolini pump CK3 90pp",
+    "WPA 772 Water Pump Assy", "bertolini TTL triplex high pressure plunger pumps",
+    "Generic plunger high pressure pump", "Apple Normal, Banana",
+    "Cast Iron KSb centrifugal pump", "5.5kw Water Pump",
+    "KSB reliable i line centrifuged pumps", "Apple Normal, Orange, Banana",
+    "Positive API 6745 hydraulic diaphragm pump", "1/2 inch Fuel Hose Pipe", "Kirloskar Water Pump",
+    "Rotodel motor pump", "PVC Electrical Insulation Materials",
+    "Electric kirloskar domestic water pump", "Electrical Insulation Materials",
+    "sellowell motor pump", "bhupathi submersible pump set",
+    "Flowshine Submersible pump set", "Index submersible pump",
+    "Wintoss Plastic Electric Switch Board", "Electric 18 watt ujagar cooler pump",
+    "Generator Service", "LG WM FHT1207ZWL, LG REF GL-S292RSCY",
+    "Water tank, Filters, Water Pump", "MS Control Submersible Panel",
+    "Centrifugal Monoblock Pumps", "Electric Motor with Pump BodyBlue and White",
+    "Various Repair and Maintenance Parts", "Earthmax Pump",
+    "Water Tank, Filters, Water Pump", "Centrifugal Water Pump for Agriculture",
+    "mono block pumps"
 ]
 
-# ✅ List of model names to match
-MODEL_NAMES = [
-    "V9", "V4", "V3", "V7", "V6"
-]
+# Salesforce credentials
+SALESFORCE_USERNAME = "venkatramana@sandbox.com"
+SALESFORCE_PASSWORD = "Venkat12345@"
+SALESFORCE_SECURITY_TOKEN = "GhcJJmjBEefdnukJoz4CAQlR"
+
+# Initialize PaddleOCR
+ocr = PaddleOCR(use_angle_cls=True, lang='en')
 
-def extract_text(image_path):
-    """
-    Extracts text from an image using PaddleOCR.
-    """
-    result = ocr.ocr(image_path)
+# Function to extract text using PaddleOCR
+def extract_text(image):
+    result = ocr.ocr(image)
     extracted_text = []
-    
     for line in result[0]:
-        extracted_text.append(line[1][0])  # Extract detected text
-        
+        extracted_text.append(line[1][0])
     return "\n".join(extracted_text)
 
+# Function to match product name using fuzzy matching
 def match_product_name(extracted_text):
-    """
-    Uses fuzzy matching to find the best product name match from extracted text.
-    """
     best_match = None
     best_score = 0
 
@@ -60,45 +133,6 @@ def match_product_name(extracted_text):
 
     return best_match if best_score >= 70 else None  # Threshold of 70 for a match
 
-def match_model_name(extracted_text):
-    """
-    Uses fuzzy matching to find the best model name match from extracted text.
-    """
-    best_match = None
-    best_score = 0
-
-    for line in extracted_text.split("\n"):
-        match, score = process.extractOne(line, MODEL_NAMES)
-        if score > best_score:
-            best_match = match
-            best_score = score
-
-    return best_match if best_score >= 70 else None  # Threshold of 70 for a match
-
-def extract_attributes(extracted_text):
-    """
-    Extracts key attributes like Model, H.P, Stage, Voltage, etc., from the extracted text.
-    """
-    attributes = {}
-    
-    for text in extracted_text.split("\n"):
-        if "Model" in text:
-            attributes["Model"] = text.split("Model")[-1].strip(" :")
-        elif "H.P" in text or "HP" in text:
-            attributes["H.P"] = text.split("H.P")[-1].strip(" :")
-        elif "Stage" in text:
-            attributes["Stage"] = text.split("Stage")[-1].strip(" :")
-        elif "Outlet" in text:
-            attributes["Outlet"] = text.split("Outlet")[-1].strip(" :")
-        elif "Voltage" in text:
-            attributes["Voltage"] = text.split("Voltage")[-1].strip(" :")
-        elif "Phase" in text:
-            attributes["Phase"] = text.split("Phase")[-1].strip(" :")
-        elif "Year of Mfg" in text:
-            attributes["Year of Mfg"] = text.split("Year of Mfg")[-1].strip(" :")
-
-    return attributes
-
 # Function to extract attributes and their values
 def extract_attributes(extracted_text):
     attributes = {}