Update app.py

app.py

@@ -1,4 +1,6 @@
 import os
+from dotenv import load_dotenv
+from pymongo import MongoClient
 # os.system("pip install streamlit pandas xlsxwriter openpyxl pymongo matplotlib seaborn")
 
 import streamlit as st
@@ -10,10 +12,15 @@ import hashlib
 import matplotlib.pyplot as plt
 import seaborn as sns
 
+load_dotenv()
+
+# Get MongoDB URI from environment
+mongo_uri = os.environ.get("MONGODB_URI")
+
 # MongoDB Setup
 try:
     from pymongo import MongoClient
-    client = MongoClient("mongodb+srv://dhruvmangroliya:[email protected]/BTP_DB?retryWrites=true&w=majority")
+    client = MongoClient(mongo_uri)
     db = client['BTP_DB']
     results_collection = db['protein_results']
 except: