Update app.py

app.py

@@ -22,8 +22,17 @@ SALESFORCE_OAUTH_URL = "https://login.salesforce.com/services/oauth2/authorize"
 SALESFORCE_TOKEN_URL = "https://login.salesforce.com/services/oauth2/token"
 CLIENT_ID = os.getenv('SALESFORCE_CLIENT_ID', 'your_client_id')  # Use environment variables
 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')
+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():