Update src/streamlit_app.py

src/streamlit_app.py

@@ -15,7 +15,6 @@ st.set_page_config(page_title="Mindful Decision-Making Coach", page_icon=":balan
 st.title("Mindful Decision-Making Coach")
 st.markdown("_A chatbot for mindful legal decisions, inspired by Ellen Langer_")
 
-
 # Initialize session state
 if "messages" not in st.session_state:
     st.session_state.messages = []
@@ -33,9 +32,8 @@ for msg in st.session_state.messages:
 user_input = None
 if st.session_state.awaiting_first:
     user_input = st.chat_input("Describe your legal decision or dilemma...")
-else:
-    if st.session_state.prompt_idx < len(PROMPTS):
-        user_input = st.chat_input("Your response...")
+elif st.session_state.prompt_idx < len(PROMPTS):
+    user_input = st.chat_input("Your response...")
 
 if user_input:
     if st.session_state.awaiting_first:
@@ -47,10 +45,8 @@ if user_input:
         st.session_state.messages.append({"role": "user", "content": user_input})
         st.session_state.messages.append({"role": "assistant", "content": PROMPTS[st.session_state.prompt_idx]})
         st.session_state.prompt_idx += 1
-
-    # Force Streamlit to rerun by updating query params
-    st.query_params["dummy"] = str(st.session_state.prompt_idx)  # Fixed here
+    # No need for any rerun or query param tricks!
 
 # Final message after all prompts
 if not st.session_state.awaiting_first and st.session_state.prompt_idx >= len(PROMPTS):
-    st.markdown("**Thank you for sharing your decision. I have logged your responses and will remind you to reflect on the outcome in a few weeks.**")
+    st.markdown("**Thank you for sharing your decision. I have logged your responses and will remind you to reflect on the outcome in a few weeks.**")