Update app.py

app.py

@@ -127,6 +127,7 @@ with col3:
        options=list(prompts.keys()),
        format_func=lambda x: x,
        index=0,
+       key="prompt_selectbox"
    )
 
 # Display chat messages from history on app rerun
@@ -179,7 +180,7 @@ if prompt := st.chat_input("Enter your prompt here..."):
         )
         
 if prompt := prompts.get(prompt_selection):
-    prompt_selection = "none"
+    st.session_state.prompt_selectbox = "none"
     
     st.session_state.messages.append({"role": "user", "content": prompt})
     with st.chat_message("user", avatar="❓"):