Update app.py

app.py

@@ -237,9 +237,9 @@ def pull_data_from_salesforce(data_type):
         )
         
         if data_type == "Inventory":
-            query = "SELECT Productname__c, Current_Stocks__c, soldstock__c FROM Inventory_Management__c LIMIT 100"
+            query = "SELECT Productname__c,Model__c, H_p__c, Stage__c, Current_Stocks__c, soldstock__c FROM Inventory_Management__c LIMIT 100"
         else:
-            query = "SELECT Productname__c, Current_Stock__c, soldstock__c FROM Un_Billable__c LIMIT 100"
+            query = "SELECT Productname__c, Model__c, H_p__c, Stage__c, Current_Stock__c, soldstock__c FROM Un_Billable__c LIMIT 100"
         
         response = sf.query_all(query)
         records = response.get("records", [])
@@ -253,6 +253,9 @@ def pull_data_from_salesforce(data_type):
         # Rename columns for better readability
         df.rename(columns={
             "Productname__c": "Product Name",
+            "Model__c": "Model",
+            "H_p__c": "H.P",
+            "Stage__c": "Stage",
             "Current_Stocks__c": "Current Stocks",
             "Current_Stock__c": "Current Stocks",
             "soldstock__c": "Sold Stock"