Update modules/salesforce_connector.py python Copy Edit

modules/salesforce_connector.py python Copy Edit

@@ -1,13 +1,13 @@
+
 from simple_salesforce import Salesforce
 import pandas as pd
 
 def connect_salesforce(username, password, token, domain="login"):
-    sf = Salesforce(username=username, password=password, security_token=token, domain=domain)
-    return sf
+    return Salesforce(username=username, password=password, security_token=token, domain=domain)
 
 def fetch_pole_data(sf):
     query = """
-    SELECT Name, Solar_Gen__c, Wind_Gen__c, Tilt__c, Vibration__c, Camera_Status__c 
+    SELECT Name, Solar_Generation__c, Wind_Generation__c, Tilt__c, Vibration__c, Camera_Status__c 
     FROM Smart_Pole__c
     LIMIT 50
     """