Update src/streamlit_app.py

src/streamlit_app.py

@@ -38,6 +38,7 @@ st.set_page_config(
     initial_sidebar_state="collapsed",
     layout="centered",
 )
+initialize_session_state()
 
 if not check_password():
     st.stop()
@@ -53,7 +54,6 @@ wx_client = setup_watsonxai_client(
 
 # ----- ----- MAIN APP
 def main():
-    initialize_session_state()
     st.subheader(f"{bot_name} {bot_icon}")
 
     if display_chat_history:
@@ -84,7 +84,7 @@ def main():
                 params=params,
                 model_id=model_id,
             )
-            print(messages)
+
             text_output = generate_response(stream_generator, stream=stream_outputs)
 
             # Stream the response with typewriter effect