from pymongo import MongoClient

# MongoDB URI with actual credentials
CONNECTION_STRING = "mongodb+srv://transpolymer:transpolymer365@cluster0.2ojpls3.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0"

# Function to get the database
def get_database():
    client = MongoClient(CONNECTION_STRING)
    return client["transpolymerDB"]  # Your database name