clear partial session state

st.session_state['count_checked'] = 0
st.session_state['checked_boxes'] = []

app.py

@@ -145,6 +145,8 @@ if st.button('Submit'):
     # # initialize
     # st.session_state.clear()
     # init_session_state()
+    st.session_state['count_checked'] = 0
+    st.session_state['checked_boxes'] = []
 
     # ### call OpenAI text-embedding
     res = openai.Embedding.create(model=EMBEDDING_MODEL, input=[query], api_key=OPENAI_API_KEY)