Update app.py

app.py

@@ -43,21 +43,7 @@ display_charts(df)
 # Salesforce Integration (e.g., fetch records from Salesforce)
 def fetch_salesforce_data():
     # Example: Querying Salesforce for related data
-    query = """
-    SELECT 
-        Id, 
-        Name, 
-        Site__c, 
-        Tilt__c, 
-        Vibration__c, 
-        Anomalies__c, 
-        Wind_Gen__c, 
-        Solar_Gen__c, 
-        Alert_Level__c, 
-        Camera_Status__c
-    FROM 
-        Pole__c
-    """
+    query = "SELECT Id, Name, Alert_Level__c, Camera_Status__c FROM Pole__c"
     result = sf.query_all(query)
     return result['records']
 
@@ -67,6 +53,7 @@ sf_data = fetch_salesforce_data()
 # Convert the Salesforce data into a Pandas DataFrame
 sf_df = pd.DataFrame(sf_data)
 
-# Display Salesforce Data in a table format with additional columns
-st.subheader("📊 Enhanced Salesforce Data")
-st.dataframe(sf_df[['Name', 'Site__c', 'Tilt__c', 'Vibration__c', 'Anomalies__c', 'Wind_Gen__c', 'Solar_Gen__c', 'Alert_Level__c', 'Camera_Status__c']], use_container_width=True)
+# Display Salesforce Data in a table format
+st.subheader("📊 Salesforce Data")
+st.dataframe(sf_df[['Name', 'Alert_Level__c', 'Camera_Status__c']], use_container_width=True)
+