Update app.py

app.py

@@ -5,32 +5,29 @@ 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 = "[email protected]"
-SF_PASSWORD = "Vedavathi@04"
-SF_SECURITY_TOKEN = "jqe4His8AcuFJucZz5NBHfGU"
-HF_API_URL = "https://api-inference.huggingface.co/models/your-model"
+# Salesforce + Hugging Face config (fill in your values)
+SF_USERNAME = "your_username"
+SF_PASSWORD = "your_password"
+SF_SECURITY_TOKEN = "your_security_token"
+HF_API_URL = "https://api-inference.huggingface.co/models/your-hf-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)
+    try:
+        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 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.")
+    except Exception as e:
+        st.error(f"Error: {e}")
 
 if __name__ == "__main__":
     main()
-
-
-
-
-