Update app.py

app.py

@@ -1,12 +1,10 @@
 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
 from fuzzywuzzy import process
+from simple_salesforce import Salesforce
 
 # Attribute mappings: readable names to Salesforce API names
 ATTRIBUTE_MAPPING = {
@@ -144,7 +142,7 @@ def interact_with_salesforce(mode, entry_type, quantity, attributes):
             if entry_type == "Sales":
                 object_name = "VENKATA_RAMANA_MOTORS__c"
                 field_name = "Quantity__c"
-            elif entry_type =="Non-Sales":
+            elif entry_type == "Non-Sales":
                 object_name = "UNBILLING_DATA__c"
                 field_name = "TotalQuantity__c"
         elif mode == "Exit":
@@ -221,44 +219,7 @@ def export_to_salesforce(mode, entry_type, quantity, edited_df):
         # Export to Salesforce
         message = interact_with_salesforce(mode, entry_type, quantity, edited_attributes)
 
-        # Fetch the price from Inventory_Management__c based on attributes
-        try:
-            sf = Salesforce(
-                username=SALESFORCE_USERNAME,
-                password=SALESFORCE_PASSWORD,
-                security_token=SALESFORCE_SECURITY_TOKEN
-            )
-            product_name = edited_attributes.get("Product name", "")
-            model_name = edited_attributes.get("Model Name", "")
-            stage = edited_attributes.get("Stage", "")
-
-            # Build the query
-            query_conditions = []
-            if product_name:
-                query_conditions.append(f"Productname__c = '{product_name}'")
-            if model_name:
-                query_conditions.append(f"Model__c = '{model_name}'")
-            if stage:
-                query_conditions.append(f"Stage__c = '{stage}'")
-
-            if query_conditions:
-                query = f"SELECT Price__c FROM Inventory_Management__c WHERE {' AND '.join(query_conditions)} LIMIT 1"
-                response = sf.query(query)
-
-                if response["records"]:
-                    price = response["records"][0].get("Price__c", None)
-                    if price:
-                        price_message = f"The estimated price for the {product_name} with {model_name} at {stage} is ₹{price:,}."
-                        return f"{message}\n\n{price_message}"
-                    else:
-                        return f"{message}\n\nPrice information not available for the specified product."
-                else:
-                    return f"{message}\n\nNo matching record found for the specified product."
-            else:
-                return f"{message}\n\nInsufficient data to fetch price information."
-
-        except Exception as e:
-            return f"{message}\n\nError fetching price information: {str(e)}"
+        return message
 
     except Exception as e:
         return f"❌ Error exporting to Salesforce: {str(e)}"
@@ -293,4 +254,4 @@ def app():
     return demo
 
 if __name__ == "__main__":
-    app().launch(share=True)
+    app().launch(share=True)