db_func.py

import os
import pymongo

user_name = os.getenv("MONGO_DB_USERNAME")
password = os.getenv("MONGO_DB_PASSWORD")

# Replace these with your MongoDB cluster connection details
mongo_uri = uri = f"mongodb+srv://{user_name}:{password}@cluster0.mrrq3sf.mongodb.net/?retryWrites=true&w=majority"
database_name = "ophthal-llm"
collection_name = "drvai-demo"


def insert_one(document):
    client = pymongo.MongoClient(mongo_uri)
    db = client[database_name]
    collection = db[collection_name]
    collection.insert_one(document)
    client.close()