Update app.py

app.py

@@ -55,11 +55,12 @@ selected_concern = None
 # 버튼 생성 및 이벤트 처리
 for i, experience_level in enumerate(experience_levels):
     with buttons[i]:
-        # 버튼이 이미 선택되었는지 여부에 따라 다른 메시지 표시
         if st.session_state.get(experience_level):
             st.button(experience_level, disabled=True)  # 활성화된 버튼 비활성화 표시
         elif st.button(experience_level):
-            st.session_state[experience_level] = True  # 버튼 상태를 저장
+            # 선택된 experience_level을 저장
+            st.session_state['experience'] = experience_level  # 이 부분이 추가됨
+            st.session_state[experience_level] = True
             with st.spinner("고민을 랜덤으로 뽑는 중입니다. 잠시 기다려주세요..."):
                 concerns = df[experience_level].dropna()
                 if not concerns.empty: