Update main.py

main.py

@@ -66,12 +66,22 @@ if "current_page" not in st.session_state:
 #             result = rag_pipeline.query(prompt, run_evaluation=False)
 #             # append result to messages and st.rerun()
 # --- MONGODB CONNECTION ---
-@st.cache_resource 
+@st.cache_resource
 def init_connection():
     if not MONGO_URL:
-        st.error("MONGO_URL not found in .env file.")
+        st.warning("MONGO_URL not set. Running without chat history.")
+        return None
+    try:
+        client = MongoClient(
+            MONGO_URL,
+            tlsCAFile=certifi.where(),
+            serverSelectionTimeoutMS=5000,
+        )
+        client.admin.command("ping")
+        return client
+    except Exception as e:
+        st.warning(f"MongoDB unavailable. Running without chat history. ({e})")
         return None
-    return MongoClient(MONGO_URL, tlsCAFile=certifi.where())
 
 client = init_connection()
 if client:
@@ -80,10 +90,14 @@ if client:
 
 # --- HELPER FUNCTIONS FOR MEMORY ---
 def get_all_past_chats():
-    if not client: return {}
-    cursor = collection.find({}, {"chat_id": 1, "messages": 1})
-    return {doc["chat_id"]: doc["messages"] for doc in cursor}
-
+    if not client:
+        return {}
+    try:
+        cursor = collection.find({}, {"chat_id": 1, "messages": 1})
+        return {doc["chat_id"]: doc["messages"] for doc in cursor}
+    except Exception as e:
+        st.warning(f"Could not load chat history. ({e})")
+        return {}
 def save_chat(chat_id, messages):
     if not client: return
     collection.update_one(