db_config.py

import os
from pymongo import MongoClient
from dotenv import load_dotenv

load_dotenv()

uri = os.getenv("MONGODB_URI")
client = MongoClient(uri)
db = client["mercury"]

env = os.getenv("FLASK_ENV")

essential_prefix="prod"
if env == "development":
    collection_prefix = os.getenv("DEV_USER")

# MongoDB collections with environment-specific names
embeddings_collection = db["topic_embedding_cache"]
cluster_results_collection = db["cluster_results"]
sections_collection = db["sections"]
essential_contents_collection = db[f"{essential_prefix}_essential_contents"]

def setup_database_indexes():        
    # Creating indexes for embeddings_collection
    embeddings_collection.create_index("topic")
    
    # Creating indexes for cluster_results_collection
    cluster_results_collection.create_index("cluster_name")
    cluster_results_collection.create_index("representative_topic.topic")    

setup_database_indexes()