Update app.py

app.py

@@ -238,8 +238,8 @@ def pull_data_from_salesforce():
             security_token=SALESFORCE_SECURITY_TOKEN
         )
        
-        query_inventory = "SELECT Product_Name__c,   Current_Stocks__c, soldstock__c FROM Inventory_Management__c LIMIT 100"
-        query_unbillable = "SELECT Product_Name__c,  Current_Stock__c,  soldstock__c FROM Un_Billable__c LIMIT 100"
+        query_inventory = "SELECT Product_Name__c, Current_Stocks__c, soldstock__c FROM Inventory_Management__c LIMIT 100"
+        query_unbillable = "SELECT Product_Name__c,Current_Stock__c, soldstock__c FROM Un_Billable__c LIMIT 100"
         
         response_inventory = sf.query_all(query_inventory)
         response_unbillable = sf.query_all(query_unbillable)
@@ -257,7 +257,6 @@ def pull_data_from_salesforce():
         # Rename columns for better readability
         df_inventory.rename(columns={
             "Product_Name__c": "Product Name (Inventory)",
-            "Modal_Name__c": "Model Name (Inventory)",
             "Current_Stocks__c": "Current Stocks (Inventory)",
             "soldstock__c": "Sold Stock (Inventory)"
         }, inplace=True)
@@ -277,8 +276,9 @@ def pull_data_from_salesforce():
        
         # Generate interactive vertical bar graph using Matplotlib
         fig, ax = plt.subplots(figsize=(12, 8))
-        df_inventory.plot(kind='bar', x="Product Name (Inventory)", y="Current Stocks (Inventory)", ax=ax, legend=False, color='blue', position=0, width=0.4)
-        df_unbillable.plot(kind='bar', x="Product Name (Unbillable)", y="Current Stocks (Unbillable)", ax=ax, legend=False, color='red', position=1, width=0.4)
+        width = 0.4  # Width of the bars
+        df_inventory.plot(kind='bar', x="Product Name (Inventory)", y="Current Stocks (Inventory)", ax=ax, legend=True, color='blue', position=0, width=width)
+        df_unbillable.plot(kind='bar', x="Product Name (Unbillable)", y="Current Stocks (Unbillable)", ax=ax, legend=True, color='red', position=1, width=width)
         ax.set_title("Stock Distribution by Product Name")
         ax.set_xlabel("Product Name")
         ax.set_ylabel("Current Stocks")