Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 from processor import DataProcessor
 from llm_handler import DDoSInference
 
+
 class DDoSResistanceHelper:
     def __init__(self):
         # Set up Streamlit page configuration
@@ -46,17 +47,27 @@ class DDoSResistanceHelper:
 
     def setup_session_state(self):
         """Initialize Streamlit session state variables."""
-        st.write("Initializing session state...")
-        session_keys = ['chat_history', 'current_file', 'analysis_complete', 'selected_theme', 'preview_data']
+        session_keys = [
+            'chat_history',
+            'current_file',
+            'analysis_complete',
+            'selected_theme',
+            'preview_data'
+        ]
+
         for key in session_keys:
             if key not in st.session_state:
-                st.session_state[key] = [] if key == 'chat_history' else None
-        st.write("Session state initialized.")
+                if key == 'chat_history':
+                    st.session_state[key] = []
+                elif key == 'selected_theme':
+                    st.session_state[key] = 'Light'  # Default theme
+                else:
+                    st.session_state[key] = None
 
     def apply_theme(self):
         """Apply selected theme to Streamlit app."""
-        theme_name = st.session_state.get('selected_theme', 'Light')
-        theme = self.themes[theme_name]
+        theme_name = st.session_state.get('selected_theme', 'Light')  # Default to 'Light' if not set
+        theme = self.themes.get(theme_name, self.themes['Light'])  # Fallback to 'Light' if theme invalid
         st.markdown(f"""
         <style>
         .stApp {{
@@ -68,8 +79,10 @@ class DDoSResistanceHelper:
 
     def render_theme_selector(self):
         """Render theme selection icons on the top-right corner."""
-        st.sidebar.empty()
-        st.write(f'<div style="position: absolute; top: 15px; right: 15px;">', unsafe_allow_html=True)
+        st.write(
+            '<div style="position: absolute; top: 15px; right: 15px;">',
+            unsafe_allow_html=True
+        )
         if st.button("🌞 Light", key="light_theme"):
             st.session_state.selected_theme = 'Light'
             self.apply_theme()
@@ -134,6 +147,7 @@ class DDoSResistanceHelper:
         self.render_analysis_interface()
         self.render_chat_interface()
 
+
 if __name__ == "__main__":
     app = DDoSResistanceHelper()
     app.main()