chat-faq / dependency.py
AnnasBlackHat's picture
edit, add, login
1fea92e
import streamlit as st
from pymongo import MongoClient
import os
import extra_streamlit_components as stx
@st.cache_resource
def load_db():
db_user = os.environ['DB_USER']
db_pass = os.environ['DB_PASS']
db_uri = f"mongodb+srv://{db_user}:{db_pass}@cluster0.firly.gcp.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0"
client = MongoClient(db_uri)
return client
@st.cache_resource(experimental_allow_widgets=True)
def get_manager():
return stx.CookieManager()
client = load_db()
db = client["chat_support"]
faq_collection = db["faq"]