Update app.py

app.py

@@ -122,13 +122,13 @@ with col2:
     )
     
 with col3:
-   prompt_selection = st.selectbox(
+   if "prompt_selectbox" not in st.session_state:
+     st.session_state["prompt_selectbox"] = st.selectbox(
        "Choose a prompt:",
        options=list(prompts.keys()),
        format_func=lambda x: x,
-       index=0,
-       key="prompt_selectbox"
-   )
+       index=0
+     )
 
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
@@ -179,8 +179,8 @@ if prompt := st.chat_input("Enter your prompt here..."):
             {"role": "assistant", "content": combined_response}
         )
         
-if prompt := prompts.get(prompt_selection):
-    st.session_state.prompt_selectbox = "none"
+if prompt := prompts.get(st.session_state["prompt_selectbox"]):
+    st.session_state["prompt_selectbox"].option("none")
     
     st.session_state.messages.append({"role": "user", "content": prompt})
     with st.chat_message("user", avatar="❓"):