Refactor Dockerfile to set MongoDB URI as an environment variable during runtime, enhancing security. Update Streamlit app to reduce cache duration and add display options for data presentation, improving user experience.

Dockerfile

@@ -14,13 +14,13 @@ COPY src/ ./src/
 
 RUN pip3 install -r requirements.txt
 
-# Expose the secret SECRET_EXAMPLE at buildtime and use its value as git remote URL
-RUN --mount=type=secret,id=mongo_uri,mode=0444,required=true \
-    git init && \
-    git remote add origin $(cat /run/secrets/mongo_uri)
+# Set the environment variable from the secret during runtime
+ENV mongo_uri=""
 
 EXPOSE 8501
 
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+# Use the secret as an environment variable during container runtime
+ENTRYPOINT --mount=type=secret,id=mongo_uri,target=/run/secrets/mongo_uri,mode=0444,required=true \
+    sh -c 'export mongo_uri=$(cat /run/secrets/mongo_uri) && streamlit run src/streamlit_app.py --server.port=8501 --server.address=0.0.0.0'

src/streamlit_app.py

@@ -16,18 +16,22 @@ def init_conn():
 
 db = init_conn()
 
-@st.cache_resource(ttl = 300)
+@st.cache_resource(ttl = 60)
 def init_baselines():
-        collection = db["HR_Table"] 
-        cursor = collection.find()
-        raw_display = pd.DataFrame(cursor)
-        raw_display.rename(columns={"Names": "Player"}, inplace = True)
-        raw_frame = raw_display.drop_duplicates(subset='Player')
-
-        return raw_frame
+    collection = db["HR_Table"] 
+    cursor = collection.find()
+    raw_display = pd.DataFrame(cursor)
+    raw_display.rename(columns={"Names": "Player"}, inplace = True)
+    raw_frame = raw_display.drop(columns=['_id']).drop_duplicates(subset='Player')
 
+    return raw_frame
 
 hr_frame = init_baselines()
 st.title("HR Finder Table")
 
-st.dataframe(hr_frame, use_container_width = True, hide_index = True)
+disp_options = st.radio("Display options:" ['Exclude DFS Info', 'Include DFS Info'], key='display_options')
+if disp_options == 'Exclude DFS Info':
+    disp_frame = hr_frame.drop(columns=['Salary', 'Position', 'FD_Position', 'Order'])
+else:
+    disp_frame = hr_frame.copy()
+st.dataframe(disp_frame, use_container_width = True, hide_index = True)