Create db.py

db.py

@@ -0,0 +1,23 @@
+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"]