File size: 578 Bytes
1fea92e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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"]