Update app.py

app.py

@@ -46,40 +46,47 @@ st.markdown('<div class="header">🏆 다른 팀장 고민해결 도와주기
 
 # Load Excel file into DataFrame
 df = pd.read_excel("painpoint2.xlsx")
-# 버튼을 가로로 나열하기 위한 컨테이너
-buttons = st.columns(3)
-experience_levels = ['저연차 팀장', '중간연차 팀장', '고연차 팀장']
-button_state = {"active": None}  # 어떤 버튼이 활성화되었는지 추적
 
-# 세션 상태 초기화
-if 'active_button' not in st.session_state:
-    st.session_state['active_button'] = None
+# 버튼 컨테이너 생성
+button_containers = [st.container() for _ in experience_levels]
+buttons = st.columns(len(experience_levels))
+selected_concern = None
 
 # 버튼 생성 및 이벤트 처리
 for i, experience_level in enumerate(experience_levels):
     with buttons[i]:
-        if st.session_state['active_button'] == experience_level:
-            st.button(experience_level, disabled=True)  # 활성화된 버튼 비활성화 표시
-        elif st.button(experience_level):
-            st.session_state['active_button'] = experience_level  # 선택된 버튼 저장
+        # 버튼 스타일 설정
+        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
             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.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']}")
+
 # 조언 제출 섹션
-if 'selected_concern' in st.session_state and st.session_state['active_button']:
-    st.subheader(f"{st.session_state['active_button']}의 고민에 대한 조언을 작성해주세요.")
-    advice = st.text_area("조언:", height=100)
+if 'selected_concern' in st.session_state and st.session_state.get('active_button'):
+    advice = st.text_area("고민에 대한 조언을 입력해주세요:", height=100)
     if st.button("조언 제출하기"):
         if advice.strip():  # 조언이 비어있지 않은 경우
-            record_advice(st.session_state['active_button'], st.session_state['selected_concern'], advice.strip())
+            record_advice(st.session_state.get('active_button'), 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['active_button'] = None  # 선택된 버튼 초기화
         else:
             st.error("조언을 입력해주세요.")