from simple_salesforce import Salesforce
import pandas as pd
import streamlit as st

def fetch_poles():
    sf = Salesforce(
        username="greenenergy@vedavathi.com",          
        password="Vedavathi@04",         
        security_token="jqe4His8AcuFJucZz5NBHfGU",      
        domain="login"      # use "login" for production, "test" for sandbox
    )

    query = """
    SELECT Name, Site__c, Solar_Generation__c, Wind_Generation__c,
           Power_Required__c, Power_Sufficient__c, Camera_Status__c,
           Alert_Level__c, Health_Score__c, RFID_Tag__c, Location_Latitude__c,
           Location_Longitude__c
    FROM Pole__c
    LIMIT 50
    """
    results = sf.query_all(query)
    df = pd.DataFrame(results['records']).drop(columns='attributes')
    return df