Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import os
 from paddleocr import PaddleOCR
 from PIL import Image
@@ -13,9 +12,9 @@ from fuzzywuzzy import process
 import plotly.graph_objects as go
 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",
+    "Product name": "Productname__c",
     "Colour": "Colour__c",
     "Motortype": "Motortype__c",
     "Frequency": "Frequency__c",
@@ -74,43 +73,15 @@ ATTRIBUTE_MAPPING = {
     "SRnumber": "SRnumber__c",
     "TypeOfEndUse": "TypeOfEndUse__c",
     "Model Name": "Model_Name_Number__c",
-    "coolingmethod": "coolingmethod__c" 
+    "coolingmethod": "coolingmethod__c"
 }
-       
-# List of product names to match
-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"
-                ]
 
-#📌 Salesforce credentials
+# Salesforce credentials
 SALESFORCE_USERNAME = "venkatramana@sandbox.com"
 SALESFORCE_PASSWORD = "Venkat12345@"
 SALESFORCE_SECURITY_TOKEN = "GhcJJmjBEefdnukJoz4CAQlR"
 
-#🧠 Initialize PaddleOCR
+# Initialize PaddleOCR
 ocr = PaddleOCR(use_angle_cls=True, lang='en')
 
 # Function to extract text using PaddleOCR
@@ -150,90 +121,6 @@ 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):
-    try:
-        sf = Salesforce(
-            username=SALESFORCE_USERNAME,
-            password=SALESFORCE_PASSWORD,
-            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
-
-        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"
-
-        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"]}
-
-        # Extract product name or model number
-        product_name = match_product_name(extracted_text)
-        attributes = extract_attributes(extracted_text)
-
-        if not product_name:
-            return "Product name could not be matched from the extracted text."
-
-        attributes["Product name"] = product_name
-
-        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 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}."
-            else:
-                return f"No matching record found for product '{product_name}' in {object_name}."
-        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)}"
-
-# Function to pull data from Salesforce MotorDataAPI
-def pull_data_from_motor_api():
-    try:
-        sf = Salesforce(
-            username=SALESFORCE_USERNAME,
-            password=SALESFORCE_PASSWORD,
-            security_token=SALESFORCE_SECURITY_TOKEN
-        )
-        motor_data = sf.apexecute("MotorDataAPI/", method="GET")
-        return motor_data  # API returns the list of records
-    except Exception as e:
-        return f"Error pulling data from MotorDataAPI: {str(e)}"
-
 # Function to pull structured data from Salesforce and display as a table
 def pull_data_from_salesforce():
     try:
@@ -242,29 +129,34 @@ def pull_data_from_salesforce():
             password=SALESFORCE_PASSWORD,
             security_token=SALESFORCE_SECURITY_TOKEN
         )
-       
+
         query = "SELECT Product_Name__c, Modal_Name__c, Current_Stocks__c FROM Inventory_Management__c LIMIT 100"
         response = sf.query_all(query)
-       
+
         records = response.get("records", [])
         if not records:
-            return "No data found in Salesforce.", None, None, None
-       
+            return "No data found in Salesforce.", None
+
         df = pd.DataFrame(records)
         df = df.drop(columns=['attributes'], errors='ignore')
-       
+
         # Rename columns for better readability
         df.rename(columns={
             "Product_Name__c": "Product Name",
             "Modal_Name__c": "Model Name",
             "Current_Stocks__c": "Current Stocks"
         }, inplace=True)
-       
-        excel_path = "salesforce_data.xlsx"
-        df.to_excel(excel_path, index=False)
-       
-        # Generate interactive bar graph using Plotly with Tooltip, Hover Effect, and Zooming
+
+        return "Data successfully retrieved.", df
+    except Exception as e:
+        return f"Error fetching data: {str(e)}", None
+
+# Function to generate the graph from the table data
+def generate_graph_from_table(df):
+    if df is not None:
         fig = go.Figure()
+
+        # Create the bar graph using the table data
         fig.add_trace(go.Bar(
             x=df['Product Name'],
             y=df['Current Stocks'],
@@ -274,7 +166,7 @@ def pull_data_from_salesforce():
             text=df['Current Stocks'],
             textposition='outside'
         ))
-       
+
         fig.update_layout(
             title="Current Stocks of Products",
             xaxis=dict(title="Product Name", tickangle=-45),
@@ -283,61 +175,90 @@ def pull_data_from_salesforce():
             dragmode='zoom',
             showlegend=False
         )
-       
-        return "Data successfully retrieved.", df, excel_path, fig
-    except Exception as e:
-        return f"Error fetching data: {str(e)}", None, None, None
+        return fig
+    else:
+        return None
 
-# Unified function to handle image processing and Salesforce interaction
-def process_image(image, mode, entry_type, quantity):
-    extracted_text = extract_text(image)
-    if not extracted_text:
-        return "No text detected in the image.", None
+# Function to update Salesforce records based on mode (Entry/Exit) and type (Sales/Non-Sales)
+def interact_with_salesforce(mode, entry_type, quantity, product_name):
+    try:
+        sf = Salesforce(
+            username=SALESFORCE_USERNAME,
+            password=SALESFORCE_PASSWORD,
+            security_token=SALESFORCE_SECURITY_TOKEN
+        )
 
-    product_name = match_product_name(extracted_text)
-    attributes = extract_attributes(extracted_text)
-    if product_name:
-        attributes["Product name"] = product_name
+        # Define the object and field names based on mode and entry type
+        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"
+                field_name = "Quantity_Sold__c"
+            elif entry_type == "Non-Sales":
+                object_name = "Un_Billable__c"
+                field_name = "Sold_Out__c"
 
-    # Interact with Salesforce
-    message = interact_with_salesforce(mode, entry_type, quantity, extracted_text)
+        # Query the product in Salesforce
+        query = f"SELECT Id, {field_name} FROM {object_name} WHERE Product_Name__c = '{product_name}' LIMIT 1"
+        response = sf.query(query)
 
-    numbered_output = "\n".join([f"{key}: {value}" for key, value in attributes.items()])
-    return f"Extracted Text:\n{extracted_text}\n\nAttributes and Values:\n{numbered_output}", message
+        if response["records"]:
+            record_id = response["records"][0]["Id"]
+            current_quantity = response["records"][0].get(field_name, 0) or 0
+
+            # Update or add the quantity based on mode
+            if mode == "Exit" and entry_type == "Sales":
+                updated_quantity = max(0, current_quantity - quantity)  # Subtract quantity on sales exit
+            else:
+                updated_quantity = current_quantity + quantity  # Add quantity on entry
+
+            # Update or create record in Salesforce
+            sf.__getattr__(object_name).update(record_id, {field_name: updated_quantity})
+            return f"Updated record for {product_name}. New {field_name}: {updated_quantity}."
+        else:
+            return f"No matching record found for product '{product_name}'."
+    except Exception as e:
+        return f"Error interacting with Salesforce: {str(e)}"
 
 # Gradio Interface
 def app():
+    status, df = pull_data_from_salesforce()
+    graph = generate_graph_from_table(df) if df is not None else None
+    
     return gr.TabbedInterface([
         gr.Interface(
             fn=process_image,
             inputs=[
-                gr.Image(type="numpy", label="📄ᴜᴘʟᴏᴀᴅ ɪᴍᴀɢᴇ"),
-                gr.Radio(label="📌Mode", choices=["ᴇɴᴛʀʏ", "ᴇxɪᴛ"], value="Entry"),
-                gr.Radio(label="📦ᴇɴᴛʀʏ ᴛʏᴘᴇ", choices=["ꜱᴀʟᴇꜱ", "ɴᴏɴ-ꜱᴀʟᴇꜱ"], value="Sales"),
-                gr.Number(label="🔢Quantity", value=1, interactive=False),
+                gr.Image(type="numpy", label="📄 Upload Image"),
+                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),
             ],
             outputs=[
-                gr.Text(label="ɪᴍᴀɢᴇ ᴅᴀᴛᴀ ᴠɪᴇᴡᴇʀ"),
-                gr.Text(label="📝ＲＥＳＵＬＴ")
+                gr.Text(label="📝 Extracted Image Data"),
+                gr.Text(label="🚀 Result")
             ],
-            title="🏢𝑽𝑬𝑵𝑲𝑨𝑻𝑨𝑹𝑨𝑴𝑨𝑵𝑨 𝑴𝑶𝑻𝑶𝑹𝑺",
-           description="📦𝐈𝐍𝐕𝐄𝐍𝐓𝐎𝐑𝐘 𝐌𝐀𝐍𝐀𝐆𝐄𝐌𝐄𝐍𝐓"
+            title="🏢 Inventory Management",
+            description="📦 Inventory Management System"
         ),
         gr.Interface(
-            fn=pull_data_from_salesforce,
+            fn=lambda: (status, df, graph),
             inputs=[],
             outputs=[
                 gr.Text(label="Status"),
-                gr.Dataframe(label="📦ꜱᴀʟᴇꜱꜰᴏʀᴄᴇ ᴅᴀᴛᴀ ᴛᴀʙʟᴇ"),
-                gr.File(label="ᴅᴏᴡɴʟᴏᴀᴅ ꜱᴀʟᴇꜱꜰᴏʀᴄᴇ ᴅᴀᴛᴀ"),
-                 gr.Plot(label="📉ꜱᴛᴏᴄᴋ ᴅɪꜱᴛʀɪʙᴜᴛɪᴏɴ ʙᴀʀ ɢʀᴀᴘʜ")
-               
+                gr.Dataframe(label="📦 Salesforce Data Table"),
+                gr.Plot(label="📉 Stock Distribution Bar Graph")
             ],
-            title="📈ꜱᴀʟᴇꜱꜰᴏʀᴄᴇ ᴅᴀᴛᴀ ᴇxᴘᴏʀᴛ",
+            title="📊 Salesforce Data Export",
             description="View, visualize (zoom-in/out), and download Salesforce data (Product Name, Model Name, Current Stocks)."
         )
-    ], ["📥OCR Processing", "📊Salesforce Data Export"])
+    ], ["📥 OCR Processing", "📊 Salesforce Data Export"])
 
 if __name__ == "__main__":
     app().launch(share=True)
-