Update app.py

app.py

@@ -47,46 +47,43 @@ st.markdown('<div class="header">🏆 다른 팀장 고민해결 도와주기
 # Load Excel file into DataFrame
 df = pd.read_excel("painpoint2.xlsx")
 
-# 버튼 컨테이너 생성
-button_containers = [st.container() for _ in experience_levels]
+# 연차 선택을 위한 버튼 선언
+experience_levels = ['저연차 팀장', '중간연차 팀장', '고연차 팀장']
 buttons = st.columns(len(experience_levels))
 selected_concern = None
 
 # 버튼 생성 및 이벤트 처리
 for i, experience_level in enumerate(experience_levels):
     with buttons[i]:
-        # 버튼 스타일 설정
-        button_style = f"border-radius:10px; border:1px solid #FFF; padding:10px; margin:3px; width:100%;"
-        if st.session_state.get('active_button') == experience_level:
-            # 선택된 버튼에 색상 적용
-            button_style += " background-color:#0d6efd; color:#FFF;"
-        else:
-            # 기본 버튼 스타일
-            button_style += " background-color:#FFF; color:#333;"
-        
-        # 버튼 렌더링
-        if st.button(experience_level, key=f"button_{i}", help=f"Select {experience_level}", args=(experience_level,), kwargs={'style': button_style}):
-            st.session_state['active_button'] = experience_level
+        # 버튼이 이미 선택되었는지 여부에 따라 다른 메시지 표시
+        if st.session_state.get(experience_level):
+            st.button(experience_level, disabled=True)
+        elif st.button(experience_level):
+            st.session_state[experience_level] = True  # 버튼 상태를 저장
             concerns = df[experience_level].dropna()
             if not concerns.empty:
                 selected_concern = concerns.sample(n=1).iloc[0]
                 st.session_state['selected_concern'] = selected_concern
+                st.session_state['experience'] = experience_level  # 경험 레벨 저장
+                st.markdown(f"### 선택된 고민: {selected_concern}")
             else:
                 st.error(f"{experience_level}에 해당하는 고민이 없습니다.")
 
 # 선택된 고민 내용을 전체 컨테이너에 표시
 if 'selected_concern' in st.session_state:
-    with st.container():
-        st.markdown(f"### 선택된 고민: {st.session_state['selected_concern']}")
+    st.markdown(f"### 선택된 고민: {st.session_state['selected_concern']}")
 
 # 조언 제출 섹션
-if 'selected_concern' in st.session_state and st.session_state.get('active_button'):
+if 'selected_concern' in st.session_state and st.session_state.get('experience'):
     advice = st.text_area("고민에 대한 조언을 입력해주세요:", height=100)
     if st.button("조언 제출하기"):
         if advice.strip():  # 조언이 비어있지 않은 경우
-            record_advice(st.session_state.get('active_button'), st.session_state['selected_concern'], advice.strip())
+            record_advice(st.session_state['experience'], st.session_state['selected_concern'], advice.strip())
             st.success("조언이 성공적으로 제출되었습니다!")
-            st.session_state['selected_concern'] = None  # 제출 후 선택된 고민 초기화
-            st.session_state['active_button'] = None  # 선택된 버튼 초기화
+            # 제출 후 세션 상태 초기화
+            st.session_state['selected_concern'] = None
+            st.session_state['experience'] = None
+            for experience_level in experience_levels:
+                st.session_state[experience_level] = False  # 모든 버튼 상태 초기화
         else:
             st.error("조언을 입력해주세요.")