Update app.py

app.py

@@ -24,26 +24,6 @@ CLIENT_ID = os.getenv('SALESFORCE_CLIENT_ID', 'your_client_id')  # Use environme
 CLIENT_SECRET = os.getenv('SALESFORCE_CLIENT_SECRET', 'your_client_secret')  # Use environment variables
 REDIRECT_URI = os.getenv('SALESFORCE_REDIRECT_URI', 'https://huggingface.co/spaces/Yaswanth56/python_test/oauth/callback')
 
-@app.route('/oauth/callback')
-def oauth_callback():
-    auth_code = request.args.get('code')  # The code returned from Salesforce
-
-    if not auth_code:
-        return jsonify({"error": "No authorization code provided"}), 400
-
-    # Your OAuth logic to handle the authorization code
-    return jsonify({"success": "OAuth callback is working!"})
-
-# Salesforce connection setup
-def get_salesforce_connection():
-    # Fetch Salesforce credentials from environment variables
-    sf = Salesforce(
-        username=os.getenv('SALESFORCE_USERNAME', 'your_username'),
-        password=os.getenv('SALESFORCE_PASSWORD', 'your_password'),
-        security_token=os.getenv('SALESFORCE_SECURITY_TOKEN', 'your_security_token')
-    )
-    return sf
-
 # OAuth flow to redirect to Salesforce login
 @app.route('/login')
 def login():
@@ -71,8 +51,8 @@ def oauth_callback():
 
     if token_response.status_code == 200:
         token_info = token_response.json()
-        access_token = token_info['sSSjyhInIsUohKpG8sHzty2q']  # Correct key to fetch the access token
-        instance_url = token_info['https://na1.salesforce.com']  # Correct key to fetch the instance URL
+        access_token = token_info['access_token']  # Correct key to fetch the access token
+        instance_url = token_info['instance_url']  # Correct key to fetch the instance URL
 
         # Store access token in session for future API requests
         session['access_token'] = access_token
@@ -102,5 +82,15 @@ def dashboard():
     else:
         return jsonify({"error": "Failed to fetch user info from Salesforce"}), 400
 
+# Salesforce connection setup
+def get_salesforce_connection():
+    # Fetch Salesforce credentials from environment variables
+    sf = Salesforce(
+        username=os.getenv('SALESFORCE_USERNAME', 'your_username'),
+        password=os.getenv('SALESFORCE_PASSWORD', 'your_password'),
+        security_token=os.getenv('SALESFORCE_SECURITY_TOKEN', 'your_security_token')
+    )
+    return sf
+
 if __name__ == '__main__':
     app.run(debug=True, host="0.0.0.0", port=7860)  # Make sure it listens on the correct port