Update salesforce_integration.py

salesforce_integration.py

@@ -1,23 +1,27 @@
 
 
+
 from simple_salesforce import Salesforce
 import pandas as pd
+import streamlit as st
 
 def fetch_poles():
+    creds = st.secrets["salesforce"]
     sf = Salesforce(
-        username="[email protected]",
-        password="Vedavathi@04",
-        security_token="jqe4His8AcuFJucZz5NBHfGU"
+        username=creds["[email protected]"],
+        password=creds["Vedavathi@04"],
+        security_token=creds["jqe4His8AcuFJucZz5NBHfGU"],
+        domain=creds {"login"}
     )
 
     query = """
     SELECT Name, Site__c, Solar_Generation__c, Wind_Generation__c,
            Power_Required__c, Power_Sufficient__c, Camera_Status__c,
-           Location_Latitude__c, Location_Longitude__c, RFID_Tag__c, 
-           Alert_Level__c, Tilt_Angle__c, Vibration_Level__c
+           Alert_Level__c, Tilt__c, Vibration__c
     FROM Pole__c
     LIMIT 50
     """
     results = sf.query_all(query)
     df = pd.DataFrame(results['records']).drop(columns='attributes')
     return df
+