Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,6 +3,7 @@ import uuid
 from datetime import datetime
 import json
 import os
+import re
 from model import (
     orchestrator_chat,
     fetch_medical_evidence,
@@ -15,7 +16,7 @@ st.set_page_config(
     page_title="Medical AI Assistant",
     page_icon=None,
     layout="wide",
-    initial_sidebar_state="collapsed"
+    initial_sidebar_state="expanded"  # Keep sidebar expanded to show the toggle
 )
 
 # Custom CSS for styling with purple->teal gradient and dark mode
@@ -45,6 +46,12 @@ st.markdown("""
         background-color: #121212;
     }
     
+    /* Sidebar styling */
+    section[data-testid="stSidebar"] {
+        background-color: #1e1e24;
+        border-right: 1px solid rgba(122, 95, 255, 0.2);
+    }
+    
     /* Chat container */
     .stChatFloatingInputContainer {
         border-radius: 16px;
@@ -140,52 +147,6 @@ st.markdown("""
         color: rgba(255, 107, 107, 0.8);
     }
     
-    /* New toggle switch */
-    .toggle-container {
-        display: flex;
-        align-items: center;
-        margin-bottom: 1rem;
-        background-color: #1e1e24;
-        padding: 8px 16px;
-        border-radius: 12px;
-    }
-    .toggle-label {
-        margin-right: 10px;
-        font-weight: 500;
-        color: #f0f0f0;
-    }
-    
-    /* ChatGPT-like toggle style */
-    .chatgpt-toggle {
-        display: flex;
-        align-items: center;
-        justify-content: flex-end;
-        background-color: rgba(32, 33, 35, 0.5);
-        padding: 8px 16px;
-        border-radius: 8px;
-        margin-bottom: 10px;
-        border-top: 1px solid rgba(255, 255, 255, 0.1);
-    }
-    
-    .chatgpt-toggle .stToggle>label {
-        background: rgba(122, 95, 255, 0.2);
-        padding: 4px 10px;
-        font-size: 0.8rem;
-    }
-    
-    /* Fixed bottom controls */
-    .controls-container {
-        position: fixed;
-        bottom: 80px;
-        left: 0;
-        right: 0;
-        z-index: 1000;
-        padding: 10px 50px;
-        background-color: rgba(18, 18, 18, 0.7);
-        backdrop-filter: blur(10px);
-        border-top: 1px solid rgba(255, 255, 255, 0.1);
-    }
-    
     /* Legal disclaimer (small and muted) */
     .footer-text {
         font-size: 0.7rem;
@@ -195,6 +156,7 @@ st.markdown("""
         position: fixed;
         bottom: 0;
         width: 100%;
+        z-index: 99;
     }
 </style>
 """, unsafe_allow_html=True)
@@ -207,6 +169,31 @@ if 'consultation_id' not in st.session_state:
 if 'use_rag' not in st.session_state:
     st.session_state.use_rag = True
 
+# Helper function to check if explanation has meaningful content
+def has_meaningful_content(text):
+    if not text:
+        return False
+    
+    # Check if the text is just equal signs or other separators
+    stripped_text = text.strip()
+    if re.match(r'^[=\-_*]+$', stripped_text.replace('\n', '')):
+        return False
+    
+    # Check if the text only contains "## REASONING" with no actual content
+    if "## REASONING" in stripped_text and len(stripped_text) < 20:
+        return False
+        
+    return True
+
+# Put the toggle in the sidebar - this is the most reliable approach
+with st.sidebar:
+    st.header("Settings")
+    st.session_state.use_rag = st.toggle("Database Search", value=st.session_state.use_rag, 
+                                       help="Toggle to enable or disable medical database search")
+
+# Main app area
+st.title("Medical AI Assistant")
+
 # Display chat history
 for message in st.session_state.history:
     if message["role"] == "user":
@@ -215,32 +202,14 @@ for message in st.session_state.history:
     else:
         with st.chat_message("assistant"):
             st.markdown(message["content"])
-            # Only display the explanation in an expander if it exists
-            if message.get("explanation"):
+            # Only display the explanation in an expander if it exists AND has actual content
+            if message.get("explanation") and has_meaningful_content(message.get("explanation")):
                 with st.expander("Show Reasoning"):
                     st.markdown(message["explanation"])
                     if message.get("evidence"):
                         st.markdown("---")
                         st.markdown("**Legend:** 🔓 = Open Access (full text available)")
 
-# Create a container at the bottom for the toggle and input
-with st.container():
-    # Add spacing
-    st.markdown("<div style='height: 60px'></div>", unsafe_allow_html=True)
-    
-    # Create a custom container for the toggle at the bottom
-    st.markdown("<div class='controls-container'>", unsafe_allow_html=True)
-    
-    # ChatGPT-style toggle
-    col1, col2 = st.columns([3, 1])
-    with col2:
-        st.markdown("<div class='chatgpt-toggle'>", unsafe_allow_html=True)
-        st.session_state.use_rag = st.toggle("Database Search", value=st.session_state.use_rag, 
-                                             help="Toggle to enable or disable medical database search")
-        st.markdown("</div>", unsafe_allow_html=True)
-    
-    st.markdown("</div>", unsafe_allow_html=True)
-
 # Chat input
 if prompt := st.chat_input("Describe your symptoms or ask a medical question..."):
     # Add user message to chat
@@ -270,8 +239,8 @@ if prompt := st.chat_input("Describe your symptoms or ask a medical question..."
         # Display response
         st.markdown(reply)
         
-        # Add explanation in an expander ONLY if explanation is not empty
-        if explanation:
+        # Add explanation in an expander ONLY if explanation has meaningful content
+        if explanation and has_meaningful_content(explanation):
             with st.expander("Show Reasoning"):
                 st.markdown(explanation)
                 
@@ -287,6 +256,9 @@ if prompt := st.chat_input("Describe your symptoms or ask a medical question..."
         "explanation": explanation,
         "evidence": evidence if evidence else []
     })
+    
+    # Force refresh to update the UI
+    st.rerun()
 
 # Small, unobtrusive legal disclaimer 
 st.markdown("""