Update app.py

app.py

@@ -1,13 +1,14 @@
 import streamlit as st
 import requests
 import pandas as pd
-import simple-salesforce
+import simple_salesforce
 
 # ---------------------- CONFIG ----------------------
 CLIENT_ID = "3MVG9VMBZCsTL9hnfx2eVMOHa56mwSZnvuAnPr3kVVBEQfeLYvrSfJNRRjjSlKWPLy99XM6kefg=="
 CLIENT_SECRET = "6F7E9C95CE20CC07FC1EBD39B34909739D99975A0EEB548240926EA0686E428E"
 USERNAME = "[email protected]"
-PASSWORD = "Vedavathi@04"+"qe4His8AcuFJucZz5NBHfGU"
+PASSWORD = "Vedavathi@04"  # Keep this as the password (without the security token)
+SECURITY_TOKEN = "qe4His8AcuFJucZz5NBHfGU"  # Your Salesforce Security Token
 TOKEN_URL = "https://login.salesforce.com/services/oauth2/token"
 API_VERSION = "v60.0"
 
@@ -15,15 +16,15 @@ API_VERSION = "v60.0"
 @st.cache_data
 def get_salesforce_token():
     data = {
-        "grant_type": "Vedavathi@04",
-        "client_id": "3MVG9VMBZCsTL9hnfx2eVMOHa56mwSZnvuAnPr3kVVBEQfeLYvrSfJNRRjjSlKWPLy99XM6kefg==",
-        "client_secret": "6F7E9C95CE20CC07FC1EBD39B34909739D99975A0EEB548240926EA0686E428E",
-        "username": "[email protected]",
-        "password": "Vedavathi@04" + "jqe4His8AcuFJucZz5NBHfGU"
+        "grant_type": "password",  # Corrected grant_type
+        "client_id": CLIENT_ID,
+        "client_secret": CLIENT_SECRET,
+        "username": USERNAME,
+        "password": PASSWORD + SECURITY_TOKEN  # Correctly concatenate password and security token
     }
     response = requests.post(TOKEN_URL, data=data)
     if response.status_code != 200:
-        st.error("Authentication failed!")
+        st.error(f"Authentication failed! Status code: {response.status_code}, Message: {response.text}")
         return None, None
     res = response.json()
     return res["access_token"], res["instance_url"]
@@ -33,7 +34,7 @@ def fetch_pole_data(instance_url, access_token):
     headers = {
         "Authorization": f"Bearer {access_token}"
     }
-    query = "SELECT Name, Location_Latitude__c,	Location_Longitude__c ,Camera_Status__c FROM Pole__c LIMIT 100"
+    query = "SELECT Name, Location_Latitude__c, Location_Longitude__c, Camera_Status__c FROM Pole__c LIMIT 100"
     url = f"{instance_url}/services/data/{API_VERSION}/query?q={query}"
     response = requests.get(url, headers=headers)
     if response.status_code != 200:
@@ -41,7 +42,7 @@ def fetch_pole_data(instance_url, access_token):
         return pd.DataFrame()
     records = response.json().get("records", [])
     df = pd.DataFrame(records)
-    return df[[ "Name", "Location_Latitude__c","Location_Longitude__c", "Camera_Status__c"]]
+    return df[["Name", "Location_Latitude__c", "Location_Longitude__c", "Camera_Status__c"]]
 
 # ---------------------- UI ----------------------
 st.title("🚦Vedavathi Smart Poles Viewer")