Update src/streamlit_app.py

src/streamlit_app.py

@@ -341,31 +341,19 @@ def convert_image_to_bytes(img):
 def main():
     st.set_page_config("XANE - AI Assistant", layout="centered")
     
-    # Initialize session ID - this creates unique ID per browser session
+    # Initialize session ID if not already set
     if "xane_id" not in st.session_state:
         st.session_state.xane_id = str(uuid.uuid4())
-        st.session_state.first_run = True
-    
-    # Clear any existing messages when new session starts
-    if st.session_state.get("first_run", True):
-        if "messages" in st.session_state:
-            del st.session_state.messages
-        if "chat_sessions" in st.session_state:
-            del st.session_state.chat_sessions
-        st.session_state.first_run = False
-    
     
     # Sidebar navigation
     st.sidebar.title("Navigation")
     page = st.sidebar.radio("Go to", ["Chatbot", "Image Generator"], key="nav_radio")
 
     if page == "Chatbot":
-        # Clear image generator session state when switching to chatbot
         if "generated_image" in st.session_state:
             del st.session_state.generated_image
         chatbot_page()
     elif page == "Image Generator":
-        # Clear chat session state when switching to image generator
         if "messages" in st.session_state:
             del st.session_state.messages
         image_generator_page()