Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import streamlit as st
 from components.dashboard import Dashboard
 from PurpleTeamIDS import analyze_security_log, API_TOKEN
@@ -12,15 +13,15 @@ st.set_page_config(
 )
 
 def main():
-    # Get API token from environment or sidebar
+    # Get API token from environment or sidebar securely
     api_token = st.sidebar.text_input(
         "Hugging Face API Token",
-        value=API_TOKEN,
+        value=os.getenv("HUGGING_FACE_API_TOKEN", API_TOKEN),
         type="password",
         key="huggingface_api_token"
     )
 
-    if api_token == "your_huggingface_api_token_here":
+    if not api_token or api_token == "your_huggingface_api_token_here":
         st.warning("Please enter your Hugging Face API token in the sidebar.")
         st.stop()
 
@@ -28,8 +29,12 @@ def main():
     dashboard = Dashboard()
 
     # Load custom CSS
-    with open("styles/custom.css") as f:
-        st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
+    css_path = "styles/custom.css"
+    try:
+        with open(css_path) as f:
+            st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
+    except FileNotFoundError:
+        st.error(f"CSS file not found: {css_path}")
 
     # Render main content
     dashboard.render_main_content()
@@ -50,5 +55,4 @@ def main():
         st.experimental_rerun()
 
 if __name__ == "__main__":
-    main()
-    
+    main()