Update database/connections.py

database/connections.py

@@ -6,14 +6,23 @@ import streamlit as st
 
 @st.cache_resource
 def get_mongo_client():
-
-    uri = cnf.MONGO_URI
-    dbname = cnf.MONGO_DB
-    client = MongoClient(uri)
-    db = client[dbname]
-    db.command("ping")
-    logger.info("MongoDB connection established successfully")
-    return db
+    try:
+        uri = cnf.MONGO_URI
+        dbname = cnf.MONGO_DB
+        
+        # Check if required environment variables are set
+        if not uri or not dbname:
+            logger.warning("MongoDB configuration not found. Running without database.")
+            return None
+            
+        client = MongoClient(uri)
+        db = client[dbname]
+        db.command("ping")
+        logger.info("MongoDB connection established successfully")
+        return db
+    except Exception as e:
+        logger.error(f"MongoDB connection failed: {e}")
+        return None
 
 def get_results_collection():
     db = get_mongo_client()