fix: KeyError in clear_session + replace deprecated use_container_width

Author: pavanbadempet

frontend/components/charts.py

@@ -40,7 +40,7 @@ def render_radar_chart(input_data: dict):
         margin=dict(l=40, r=40, t=40, b=40)
     )
 
-    st.plotly_chart(fig, use_container_width=True)
+    st.plotly_chart(fig, use_container_width=True)  # Still valid for plotly_chart
 
 def render_trend_chart(records: list, metric_key: str, label: str):
     """
@@ -73,4 +73,4 @@ def render_trend_chart(records: list, metric_key: str, label: str):
     fig = px.line(df, x="Date", y=label, markers=True, title=f"{label} Over Time")
     fig.update_layout(xaxis_title="Checkup Date", yaxis_title=label)
 
-    st.plotly_chart(fig, use_container_width=True)
+    st.plotly_chart(fig, use_container_width=True)  # Still valid for plotly_chart

frontend/components/sidebar.py

@@ -341,7 +341,7 @@ def render_sidebar():
 
         # --- 7. SIGN OUT BUTTON ---
         st.markdown("<div style='height: 0.75rem'></div>", unsafe_allow_html=True)
-        if st.button("🚪 Sign Out", type="secondary", use_container_width=True, key="logout_btn"):
+        if st.button("🚪 Sign Out", type="secondary", width="stretch", key="logout_btn"):
             api.clear_session()
             st.rerun()
 

frontend/utils/api.py

@@ -36,8 +36,17 @@ def load_session() -> Optional[Dict[str, str]]:
 
 def clear_session():
     cm = _get_cookie_manager()
-    cm.delete("auth_token", key="del_token")
-    cm.delete("auth_username", key="del_user")
+    # Check if cookie exists before deleting to avoid KeyError
+    try:
+        if cm.get("auth_token"):
+            cm.delete("auth_token", key="del_token")
+    except Exception:
+        pass
+    try:
+        if cm.get("auth_username"):
+            cm.delete("auth_username", key="del_user")
+    except Exception:
+        pass
     st.session_state.pop('token', None)
     st.session_state.pop('username', None)
 

frontend/views/auth_view.py

@@ -116,7 +116,7 @@ def render_auth_page():
             with st.form("login", border=False):
                 u = st.text_input("Username", placeholder="Enter username", label_visibility="collapsed")
                 p = st.text_input("Password", type="password", placeholder="Enter password", label_visibility="collapsed")
-                if st.form_submit_button("Sign In →", type="primary", use_container_width=True):
+                if st.form_submit_button("Sign In →", type="primary", width="stretch"):
                     if api.login(u, p): st.rerun()
 
         with tab2:
@@ -125,7 +125,7 @@ def render_auth_page():
                 us = st.text_input("Username", placeholder="Choose a username", label_visibility="collapsed")
                 em = st.text_input("Email", placeholder="Your email", label_visibility="collapsed")
                 pw = st.text_input("Password", type="password", placeholder="Create password", label_visibility="collapsed")
-                if st.form_submit_button("Create Account", type="primary", use_container_width=True):
+                if st.form_submit_button("Create Account", type="primary", width="stretch"):
                     if api.signup(us, pw, em, us, "2000-01-01"):  # Use username as name initially
                         if api.login(us, pw): st.rerun()
 

frontend/views/chat_view.py

@@ -236,6 +236,6 @@ def render_chat_page():
         st.markdown("<br>", unsafe_allow_html=True)
         col1, col2, col3 = st.columns([1, 1, 1])
         with col2:
-            if st.button("🗑️ Clear Chat", type="secondary", use_container_width=True):
+            if st.button("🗑️ Clear Chat", type="secondary", width="stretch"):
                 st.session_state.messages = []
                 st.rerun()

frontend/views/pricing_view.py

@@ -118,7 +118,7 @@ def render_pricing_page():
 <div style="background: linear-gradient(180deg, rgba(37, 99, 235, 0.05), rgba(59, 130, 246, 0.02)); border: 2px solid #3B82F6; border-top: none; border-radius: 0 0 16px 16px; padding: 0 2rem 2rem 2rem; text-align: center;">
 """, unsafe_allow_html=True)
 
-        if st.button("Upgrade Now", key="upgrade_pro", type="primary", use_container_width=True):
+        if st.button("Upgrade Now", key="upgrade_pro", type="primary", width="stretch"):
             with st.spinner("Initializing Payment..."):
                 # Create Order (999 INR = 99900 paise)
                 resp = api.create_payment_order(99900, "pro_tier")