Update app.py

app.py

@@ -17,9 +17,9 @@ client = gspread.authorize(creds)
 sheet = client.open("generation_streamlit_0302").sheet1
 
 # 사용자 조언 데이터를 Google Sheets 문서에 추가하는 함수
-def record_advice(concern, advice):
+def record_advice(experience, concern, advice):
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    sheet.append_row([timestamp, concern, advice])
+    sheet.append_row([timestamp, experience, concern, advice])
 
 # 스타일링을 위한 CSS
 st.markdown("""
@@ -46,30 +46,40 @@ st.markdown('<div class="header">🏆 다른 팀장 고민해결 도와주기
 
 # Load Excel file into DataFrame
 df = pd.read_excel("painpoint2.xlsx")
-
-# 버튼을 통한 연차 선택 및 고민 표시
+# 버튼을 가로로 나열하기 위한 컨테이너
+buttons = st.columns(3)
 experience_levels = ['저연차 팀장', '중간연차 팀장', '고연차 팀장']
-concerns = {experience: df[experience].dropna() for experience in experience_levels}
+button_state = {"active": None}  # 어떤 버튼이 활성화되었는지 추적
+
+# 세션 상태 초기화
+if 'active_button' not in st.session_state:
+    st.session_state['active_button'] = None
 
 # 버튼 생성 및 이벤트 처리
-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"{experience}에 해당하는 고민이 없습니다.")
+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  # 선택된 버튼 저장
+            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:
-    st.subheader(f"{st.session_state['experience']}의 고민에 대한 조언을 작성해주세요.")
+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 st.button("조언 제출하기"):
         if advice.strip():  # 조언이 비어있지 않은 경우
-            record_advice(st.session_state['selected_concern'], advice.strip())
+            record_advice(st.session_state['active_button'], st.session_state['selected_concern'], advice.strip())
             st.success("조언이 성공적으로 제출되었습니다!")
-            del st.session_state['selected_concern']  # 제출 후 선택된 고민 초기화
+            # 제출 후 세션 상태 초기화
+            st.session_state['selected_concern'] = None
+            st.session_state['active_button'] = None
         else:
             st.error("조언을 입력해주세요.")