Update app.py

app.py

@@ -21,6 +21,7 @@ def record_advice(concern, advice):
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     sheet.append_row([timestamp, concern, advice])
 
+# 스타일링을 위한 CSS
 st.markdown("""
     <style>
     .header {
@@ -40,73 +41,40 @@ st.markdown("""
     </style>
     """, unsafe_allow_html=True)
 
-# Display the header
+# 헤더 표시
 st.markdown('<div class="header">🏆 다른 팀장 고민해결 도와주기 🏆 </div>', unsafe_allow_html=True)
 
-st.markdown("""
-    <div class="description-box">
-        <p> How to use </br> 1. 좌측에서 조언을 제공할 팀장의 연차를 선택해주세요. </br> 2. [고민 확인]버튼을 누르면 아래에 고민 내용과 조언 작성창이 뜹니다. </br> 3. 다시 한번 [고민 확인]버튼을 누르면 새로운 고민이 나옵니다 </br> </p>
-    </div>
-    """, unsafe_allow_html=True)
-
 # Load Excel file into DataFrame
 df = pd.read_excel("painpoint2.xlsx")
 
-# Define experience level options
-seniority_options = ['저연차 팀장', '중간연차 팀장', '고연차 팀장']
-
-def set_background():
-    st.markdown(
-        """
-        <style>
-        .stApp {
-            background-image: url('URL_TO_YOUR_BACKGROUND_IMAGE');
-            background-size: cover;
-        }
-        </style>
-        """,
-        unsafe_allow_html=True
-    )
-
-set_background()
-
-st.sidebar.markdown("#\n" * 5) 
-# Add radio button to select experience level
-st.sidebar.subheader("조언해줄 연차 선택하기:")
-selected_experience = st.sidebar.radio(" ", seniority_options)
-submit_button_left = st.sidebar.button("고민 확인")
-# Initialize session state for button click
-if 'button_clicked' not in st.session_state:
-    st.session_state['button_clicked'] = False
-
-# Modify the button click to update session state
-if submit_button_left:
-    st.session_state['button_clicked'] = True
-
-# Display random rows based on selected experience level
-if submit_button_left:
-    st.subheader("선택된 고민은?")
-    selected_row = df[selected_experience].sample(n=1).iloc[0]
-    st.session_state['selected_concern'] = selected_row['고민']  # 여기서 '고민'은 DataFrame의 해당 열 이름입니다.
-    with st.expander("고민 내용보기", expanded=True):
-        st.markdown(selected_row['고민'])  # 세션 상태에 저장된 고민 내용을 사용자에게 보여줍니다.
-
-    # Add advice section
-    st.subheader("고민 조언해주기")
-    st.write("다른 팀장님들의 고민에 대해 조언해주시겠어요?")
-    advice = st.text_area(" ", height=100)
-    submit_button_right = st.button("조언 제출하기")
-
-        # 조언 제출 버튼 처리
-    if submit_button_right:
-        if not st.session_state.get('submitted', False):  # 'submitted' 상태가 False인 경우에만 실행
-            user_advice = advice.strip()  # 사용자 입력에서 공백 제거
-            if user_advice:  # 사용자가 실제로 조언을 입력했는지 확인
-                selected_concern = st.session_state.get('selected_concern', '고민 없음')  # 세션 상태에서 선택된 고민 내용을 가져옴
-                record_advice(selected_concern, user_advice)  # Google Sheets에 기록
-                st.session_state['submitted'] = True  # 'submitted' 상태를 True로 설정하여 중복 제출 방지
-                st.success("조언이 성공적으로 제출되었습니다!")
-            else:
-                st.error("조언을 입력해주세요.")  # 사용자가 조언을 입력하지 않은 경우 에러 메시지 출력
+# 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} 고민 확인")
+
+# 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]
+            st.session_state['selected_concern'] = selected_concern
+            st.markdown(f"### 선택된 고민: {selected_concern}")
+        else:
+            st.error(f"{selected_experience}에 해당하는 고민이 없습니다.")
+    else:
+        st.error("선택한 연차에 해당하는 열이 데이터에 없습니다.")
+
+# 조언 제출 섹션
+if 'selected_concern' in st.session_state:
+    advice = st.text_area("고민에 대한 조언을 입력해주세요:", height=100)
+    submit_advice_button = st.button("조언 제출하기")
+
+    if submit_advice_button:
+        if advice.strip():  # 조언이 비어있지 않은 경우
+            record_advice(st.session_state['selected_concern'], advice.strip())
+            st.success("조언이 성공적으로 제출되었습니다!")
+            del st.session_state['selected_concern']  # 제출 후 선택된 고민 초기화
         else:
-            st.warning("이미 제출되었습니다.")  # 이미 제출된 경우 경고 메시지 출력
+            st.error("조언을 입력해주세요.")