Update app.py

app.py

@@ -218,33 +218,58 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
 
     except Exception as e:
         return f"❌ Error interacting with Salesforce: {str(e)}"
+import pandas as pd
+import matplotlib.pyplot as plt
+from io import BytesIO
+from PIL import Image
+from simple_salesforce import Salesforce
+
+# 📌 Salesforce Credentials
+SALESFORCE_USERNAME = "venkatramana@sandbox.com"
+SALESFORCE_PASSWORD = "Venkat12345@"
+SALESFORCE_SECURITY_TOKEN = "GhcJJmjBEefdnukJoz4CAQlR"
 
-def pull_data_from_motor_api():
+# Function to Fetch and Display Salesforce Data
+def fetch_salesforce_data():
     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
+
+        query = "SELECT Product_Name__c, Current_Stocks__c FROM Inventory_Management__c LIMIT 100"
+        response = sf.query(query)
+        records = response.get('records', [])
+
+        if not records:
+            return "<p>No Data Found</p>", None
+
+        # Convert to DataFrame
+        df = pd.DataFrame(records)
+        df.rename(columns={'Product_Name__c': 'Product Name', 'Current_Stocks__c': 'Current Stocks'}, inplace=True)
+        df.drop(columns=['attributes'], inplace=True, errors='ignore')
+
+        return df.to_html(index=False), generate_inventory_graph(df)
+
     except Exception as e:
-        return f"Error pulling data from MotorDataAPI: {str(e)}"
-      # Function to generate a bar graph from Salesforce data
-def generate_bar_graph(df):
+        return f"Error fetching data: {str(e)}", None
+
+# Function to Generate Inventory Graph
+def generate_inventory_graph(df):
     try:
-        fig, ax = plt.subplots(figsize=(12, 8))
+        fig, ax = plt.subplots(figsize=(12, 6))
         df.plot(kind='bar', x="Product Name", y="Current Stocks", ax=ax, legend=False)
-        ax.set_title("Stock Distribution by Product Name")
+        ax.set_title("Inventory Stock Distribution")
         ax.set_xlabel("Product Name")
-        ax.set_ylabel("Current Stocks")
+        ax.set_ylabel("Stock Quantity")
         plt.xticks(rotation=45, ha="right", fontsize=10)
         plt.tight_layout()
         buffer = BytesIO()
         plt.savefig(buffer, format="png")
         buffer.seek(0)
-        img = Image.open(buffer)
-        return img
+        return Image.open(buffer)
+
     except Exception as e:
         return None