Update app.py

app.py

@@ -153,6 +153,72 @@ def filter_valid_attributes(attributes, valid_fields):
 #📊 Function to interact with Salesforce based on mode and type
 
 
+import os
+from paddleocr import PaddleOCR
+from PIL import Image
+import gradio as gr
+import requests
+import re
+from simple_salesforce import Salesforce
+import pandas as pd
+import matplotlib.pyplot as plt
+from io import BytesIO
+from fuzzywuzzy import process
+import kaleido  # Ensure kaleido is imported
+
+# Salesforce credentials
+SALESFORCE_USERNAME = "venkatramana@sandbox.com"
+SALESFORCE_PASSWORD = "Venkat12345@"
+SALESFORCE_SECURITY_TOKEN = "GhcJJmjBEefdnukJoz4CAQlR"
+
+# Initialize PaddleOCR
+ocr = PaddleOCR(use_angle_cls=True, lang='en')
+
+# 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])
+    return "\n".join(extracted_text)
+
+# Function to match product name using fuzzy matching
+def match_product_name(extracted_text):
+    best_match = None
+    best_score = 0
+    PRODUCT_NAMES = [
+        "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"
+    ]
+    for line in extracted_text.split("\n"):
+        match, score = process.extractOne(line, PRODUCT_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
+
 # Function to interact with Salesforce based on mode and type
 def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
     try:
@@ -174,6 +240,16 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
             elif entry_type == "Non-Sales":
                 object_name = "UNBILLING_DATA__c"
                 field_name = "TotalQuantity__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."
+
+            # Creating a new record
+            sf.__getattr__(object_name).create({product_field_name: product_name, field_name: quantity})
+            return f"✅ Record created in {object_name} with Product: '{product_name}', Quantity: {quantity}."
+
         elif mode == "Exit":
             if entry_type == "Sales":
                 object_name = "Inventory_Management__c"
@@ -182,16 +258,11 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
                 object_name = "Un_Billable__c"
                 field_name = "Sold_Out__c"
 
-        if not object_name or not field_name:
-            return "Invalid mode or entry type."
-
-        # 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."
+            # 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."
 
-        # Handling "Exit" Mode (Updating Records)
-        if mode == "Exit":
             query = f"SELECT Id, {field_name} FROM {object_name} WHERE {product_field_name} = '{product_name}' LIMIT 1"
             response = sf.query(query)
 
@@ -203,15 +274,10 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
             else:
                 return f"❌ No matching record found for product '{product_name}' in {object_name}."
 
-        # Handling "Entry" Mode (Creating Records)
-        else:
-            attributes = {product_field_name: product_name, field_name: quantity}
-            sf.__getattr__(object_name).create(attributes)
-            return f"✅ Data successfully exported to Salesforce object {object_name}."
-
     except Exception as e:
         return f"❌ Error interacting with Salesforce: {str(e)}"
-     # Function to pull structured data from Salesforce and display as a table
+
+    # Function to pull structured data from Salesforce and display as a table
     def pull_data_from_salesforce():
         try:
             sf = Salesforce(