app.py

from flask import Flask, request, jsonify
from transformers import BertTokenizerFast, BertForTokenClassification, pipeline
from simple_salesforce import Salesforce
import os
from dotenv import load_dotenv  # Import dotenv to load the .env file

# Load environment variables from .env file
load_dotenv()

app = Flask(__name__)

# Load Salesforce credentials from .env file using os.getenv()
SF_USERNAME = os.getenv("smart.scorer@sathkrutha.com")
SF_PASSWORD = os.getenv("Vendor@1234")
SF_SECURITY_TOKEN = os.getenv("XcBErW59i7jONEwwjNTXMlIKO")

# Authenticate with Salesforce
sf = Salesforce(username=SF_USERNAME, password=SF_PASSWORD, security_token=SF_SECURITY_TOKEN)

# Load the pre-trained BERT model for NER (you can fine-tune this if needed)
tokenizer = BertTokenizerFast.from_pretrained('bert-base-uncased')
model = BertForTokenClassification.from_pretrained('bert-base-uncased')
nlp = pipeline("ner", model=model, tokenizer=tokenizer)

# Your existing Flask route and logic here...

if __name__ == '__main__':
    app.run(debug=True)