Update app.py

app.py

@@ -47,34 +47,29 @@ st.markdown('<div class="header">🏆 다른 팀장 고민해결 도와주기
 # Load Excel file into DataFrame
 df = pd.read_excel("painpoint2.xlsx")
 
-# Add radio button to select experience level and display concerns
+# 버튼을 통한 연차 선택 및 고민 표시
 experience_levels = ['저연차 팀장', '중간연차 팀장', '고연차 팀장']
-selected_experience = st.radio("조언해줄 연차 선택하기:", experience_levels)
-submit_button = st.button(f"{selected_experience} 고민 확인")
+concerns = {experience: df[experience].dropna() for experience in experience_levels}
 
-# Display random concern based on selected experience level
-if submit_button:
-    if selected_experience in df.columns:
-        # 해당 연차 열에서 무작위로 고민을 선택
-        concerns = df[selected_experience].dropna()
-        if not concerns.empty:
-            selected_concern = concerns.sample(n=1).iloc[0]
+# 버튼 생성 및 이벤트 처리
+for experience in experience_levels:
+    if st.button(experience):
+        if not concerns[experience].empty:
+            selected_concern = concerns[experience].sample(n=1).iloc[0]
             st.session_state['selected_concern'] = selected_concern
+            st.session_state['experience'] = experience  # 경험 레벨 저장
             st.markdown(f"### 선택된 고민: {selected_concern}")
         else:
-            st.error(f"{selected_experience}에 해당하는 고민이 없습니다.")
-    else:
-        st.error("선택한 연차에 해당하는 열이 데이터에 없습니다.")
+            st.error(f"{experience}에 해당하는 고민이 없습니다.")
 
 # 조언 제출 섹션
 if 'selected_concern' in st.session_state:
-    advice = st.text_area("고민에 대한 조언을 입력해주세요:", height=100)
-    submit_advice_button = st.button("조언 제출하기")
-
-    if submit_advice_button:
+    st.subheader(f"{st.session_state['experience']}의 고민에 대한 조언을 작성해주세요.")
+    advice = st.text_area("조언:", height=100)
+    if st.button("조언 제출하기"):
         if advice.strip():  # 조언이 비어있지 않은 경우
             record_advice(st.session_state['selected_concern'], advice.strip())
             st.success("조언이 성공적으로 제출되었습니다!")
             del st.session_state['selected_concern']  # 제출 후 선택된 고민 초기화
         else:
-            st.error("조언을 입력해주세요.")
+            st.error("조언을 입력해주세요.")