# app.py

import streamlit as st
from modules.salesforce_connector import connect_salesforce, fetch_pole_data
from modules.huggingface_inference import predict_alerts
from modules.data_view import show_raw_data, show_predictions, show_status

# ---------------------- CONFIG ----------------------
SF_USERNAME = "greenenergy@vedavathi.com"
SF_PASSWORD = "Vedavathi@04"
SF_SECURITY_TOKEN = "jqe4His8AcuFJucZz5NBHfGU"
HF_API_URL = "https://api-inference.huggingface.co/models/your-model"
HF_API_TOKEN = "hf_your_token"

# ---------------------- MAIN ----------------------
def main():
    st.title("📡 Hugging Face + Salesforce Smart Pole Monitor")

    sf = connect_salesforce(SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN)
    df = fetch_pole_data(sf)

    if not df.empty:
        show_raw_data(df)
        df = predict_alerts(df, HF_API_URL, HF_API_TOKEN)
        show_status()
        show_predictions(df)
    else:
        st.warning("No data fetched from Salesforce.")

if __name__ == "__main__":
    main()