db.py

import streamlit as st
from pymongo import MongoClient
from dotenv import load_dotenv
import os

load_dotenv()

@st.cache_resource
def get_db():
    mongo_uri = os.getenv("MONGO_URI")

    if not mongo_uri:
        raise RuntimeError("MongoDB URI not configured")

    client = MongoClient(
        mongo_uri,
        serverSelectionTimeoutMS=5000,
        connectTimeoutMS=5000,
        tls=True
    )

    client.admin.command("ping")
    return client["final_chatbot_talks"]