공감 비서 프롬프트 및 설문조사 링크 위치 개선

app.py

@@ -278,7 +278,7 @@ A: 단기적으로는 맞습니다. 성과 없이 퇴사하면 이력서에 남
         logger.error(f"Error during OpenAI API call: {e}", exc_info=True)
         return "답변을 생성하는 중에 문제가 발생했습니다. OpenAI API 키 또는 서비스 상태를 확인해주세요."
 
-# --- Streamlit 앱 UI (사이드바 메뉴 추가) ---
+# --- Streamlit 앱 UI (Khan 멘토 단일 루프 구조) ---
 st.set_page_config(page_title="Khan 멘토 (PM 영상 기반)", layout="wide")
 
 # --- 사이드바 메뉴 ---
@@ -286,16 +286,18 @@ menu = st.sidebar.radio(
     "기능 선택",
     ("Khan 멘토에게 상담하기", "상사에게 잘보이기")
 )
+# 사이드바 맨 아래에 설문조사 링크
+st.sidebar.markdown('<hr style="margin:1em 0;">', unsafe_allow_html=True)
+st.sidebar.markdown(
+    '<a href="https://forms.gle/SUqrGBT3dktSB7v26" target="_blank" style="display:inline-block; background:#f9e79f; color:#1a237e; font-weight:bold; padding:0.5em 1.2em; border-radius:8px; text-decoration:none; font-size:1.1em; margin-bottom:16px;">📝 서비스 사용성 설문조사 참여하기</a>',
+    unsafe_allow_html=True
+)
 
 openai_client = init_openai_client()
 
 if menu == "Khan 멘토에게 상담하기":
     st.title("✨ Khan 멘토에게 질문하기")
-    st.markdown(
-        '<a href="https://forms.gle/SUqrGBT3dktSB7v26" target="_blank" style="display:inline-block; background:#f9e79f; color:#1a237e; font-weight:bold; padding:0.5em 1.2em; border-radius:8px; text-decoration:none; font-size:1.1em; margin-bottom:16px;">📝 서비스 사용성 설문조사 참여하기</a>',
-        unsafe_allow_html=True
-    )
-    st.markdown("PM 관련 영상 내용을 기반으로 Khan 멘토가 답변해 드립니다.")
+    
 
     # --- API 키 확인 및 리소스 초기화 ---
     pc = init_pinecone()
@@ -303,99 +305,76 @@ if menu == "Khan 멘토에게 상담하기":
     index = get_pinecone_index(pc, INDEX_NAME)
 
     # --- 상태 관리 ---
-    if 'user_state' not in st.session_state:
-        st.session_state['user_state'] = ''
+    if 'user_question' not in st.session_state:
+        st.session_state['user_question'] = ''
     if 'empathy_message' not in st.session_state:
         st.session_state['empathy_message'] = ''
-    if 'example_questions' not in st.session_state:
-        st.session_state['example_questions'] = []
-    if 'selected_question' not in st.session_state:
-        st.session_state['selected_question'] = ''
     if 'khan_answer' not in st.session_state:
         st.session_state['khan_answer'] = ''
-    if 'step' not in st.session_state:
-        st.session_state['step'] = 1
-
-    # --- 1단계: 사용자 상태 입력 ---
-    if st.session_state['step'] == 1:
-        user_state = st.text_area("지금 어떤 상황이신가요? 고민을 자유롭게 적어주세요.", value=st.session_state['user_state'])
-        if st.button("나의 이야기 들려주기 ✍️"):
-            st.session_state['user_state'] = user_state
-            # 2단계로 이동
-            st.session_state['step'] = 2
-            st.rerun()
-
-    # --- 2단계: 공감 메시지 + 예시 질문 생성 ---
-    if st.session_state['step'] == 2:
-        with st.spinner("어떤 상황인지 고민해볼께요..."):
-            # 1. 공감 메시지 생성
+    if 'pinecone_results' not in st.session_state:
+        st.session_state['pinecone_results'] = []
+    if 'extra_questions' not in st.session_state:
+        st.session_state['extra_questions'] = []
+    if 'current_input' not in st.session_state:
+        st.session_state['current_input'] = ''
+
+    # --- 질문 입력 및 답변 생성 ---
+    st.markdown("#### 당신의 고민을 알려 주세요!")
+    user_q = st.text_input("나의 고민은...", value=st.session_state['current_input'], key="main_input")
+    if st.button("질문하기", key="main_ask") or (user_q and st.session_state['user_question'] != user_q):
+        st.session_state['user_question'] = user_q
+        st.session_state['current_input'] = user_q
+        # 1. 공감 비서 메시지
+        with st.spinner("생각중..."):
             empathy_prompt = f"""
-            너는 따뜻하고 공감 능력이 뛰어난 상담가야.
-            아래 사용자의 상황을 듣고, 충분히 감정적으로 공감해주고, 용기를 북돋아주는 말을 해줘.
-            부를 때는 '당신은', '당신이'처럼 친근한 말투로 부르고, 물음표는 사용하지 않아.
-            상황: "{st.session_state['user_state']}"
+            너는 따뜻하고 친절한 비서야. 아래 사용자의 질문을 듣고, 감정적으로 충분하게 1~2문장으로 공감해주되 질문에 대한 답변은 하지마, 마지막에 '칸 멘토의 생각을 들어볼까요?'라고 안내해줘. \n질문: "{user_q}"
             """
             try:
                 empathy_response = openai_client.chat.completions.create(
-                    model="gpt-4o",
+                    model="gpt-4o-mini",
                     messages=[{"role": "system", "content": empathy_prompt}],
                     temperature=0.7,
                 )
                 st.session_state['empathy_message'] = empathy_response.choices[0].message.content.strip()
             except Exception as e:
                 st.session_state['empathy_message'] = f"공감 메시지 생성 중 오류: {e}"
-            # 2. 예시 질문 생성
-            example_prompt = f"""
-            아래 상황에서 Khan 멘토에게 할 수 있는 구체적이고 실용적인 질문 3~4가지를 한국어로 만들어줘.\n각 질문은 한 문장으로, 실제로 도움이 될 만한 내용이어야 해.\n상황: "{st.session_state['user_state']}"
-            """
+        # 2. Pinecone 검색 및 Khan 멘토 답변
+        with st.spinner("Khan 멘토가 뜸을 들이며..."):
+            pinecone_results = search(user_q, top_k=5, _index=index, _model=model)
+            st.session_state['pinecone_results'] = pinecone_results
+            khan_answer = generate_khan_answer(user_q, pinecone_results, openai_client)
+            st.session_state['khan_answer'] = khan_answer
+        # 3. 추가 질문 생성
+        with st.spinner("추가 질문을 생성하는 중..."):
+            extra_prompt = (
+                f"아래 질문에서 유사하게 궁금할 수 있는 추가 질문 3~4개를 한국어로 만들어줘. 지나치게 세부적인 툴에 대한 얘기보다는 프로덕트, 프로젝트, 리더십에 대한 일반적인 질문으로 만들어. 질문: \"{st.session_state['user_question']}"
+            )
             try:
-                example_response = openai_client.chat.completions.create(
-                    model="gpt-4o",
-                    messages=[{"role": "system", "content": example_prompt}],
-                    temperature=0.5,
+                extra_response = openai_client.chat.completions.create(
+                    model="gpt-4o-mini",
+                    messages=[{"role": "system", "content": extra_prompt}],
+                    temperature=0.5
                 )
-                # 응답에서 질문만 추출 (숫자/기호/줄바꿈 등 정제)
                 import re
-                raw = example_response.choices[0].message.content.strip()
+                raw = extra_response.choices[0].message.content.strip()
                 questions = re.findall(r'\d+\.\s*(.+)', raw)
                 if not questions:
-                    # 숫자 없이 줄바꿈만 있을 경우
                     questions = [q.strip('-• ').strip() for q in raw.split('\n') if q.strip()]
-                st.session_state['example_questions'] = questions[:4]
-            except Exception as e:
-                st.session_state['example_questions'] = [f"예시 질문 생성 중 오류: {e}"]
-            # 3단계로 이동
-            st.session_state['step'] = 3
-            st.rerun()
-
-    # --- 3단계: 공감 메시지 + 예시 질문 버튼/직접입력 + Khan 답변 ---
-    if st.session_state['step'] == 3:
-        st.success(st.session_state['empathy_message'])
-        st.markdown("#### 이런 질문을 해볼 수 있어요!")
-        cols = st.columns(len(st.session_state['example_questions']))
-        for i, q in enumerate(st.session_state['example_questions']):
-            if cols[i].button(q):
-                st.session_state['selected_question'] = q
-                st.session_state['step'] = 4
+                st.session_state['extra_questions'] = questions[:4]
                 st.rerun()
-        st.markdown("---")
-        user_q = st.text_input("직접 궁금한 점을 입력해도 좋아요!", value=st.session_state['selected_question'])
-        if st.button("Khan 멘토에게 질문하기"):
-            st.session_state['selected_question'] = user_q
-            st.session_state['step'] = 4
-            st.rerun()
+            except Exception as e:
+                st.session_state['extra_questions'] = [f"추가 질문 생성 중 오류: {e}"]
+        st.rerun()
 
-    # --- 4단계: Khan 멘토 답변 ---
-    if st.session_state['step'] == 4:
-        with st.spinner("Khan 멘토가 답변을 준비하는 중..."):
-            pinecone_results = search(st.session_state['selected_question'], top_k=5, _index=index, _model=model)
-            khan_answer = generate_khan_answer(st.session_state['selected_question'], pinecone_results, openai_client)
-            st.session_state['khan_answer'] = khan_answer
+    # --- 답변 및 추가질문 UI ---
+    if st.session_state['user_question']:
+        st.info(st.session_state['empathy_message'])
         st.subheader("💡 Khan 멘토의 답변")
         st.markdown(st.session_state['khan_answer'])
         # 참고 영상 정보 표시
+        pinecone_results = st.session_state['pinecone_results']
         if pinecone_results:
-            with st.expander("답변에 참고한 영상 정보 보기"):
+            with st.expander("답변에 참고한 영상 정보 보기", expanded=True):
                 displayed_urls = set()
                 for i, r in enumerate(pinecone_results):
                     url = r.get('URL', 'N/A')
@@ -437,25 +416,49 @@ if menu == "Khan 멘토에게 상담하기":
                             except Exception as e:
                                 logger.warning(f"st.video failed for non-YouTube URL {url}: {e}")
         st.markdown("---")
-        # --- 예시 질문 버튼/직접입력 다시 노출 (후속 질문) ---
-        st.markdown("#### 추가로 궁금한 점이 있으신가요? 아래 예시 질문을 클릭하거나 직접 입력해보세요!")
-        cols = st.columns(len(st.session_state['example_questions']))
-        for i, q in enumerate(st.session_state['example_questions']):
-            if cols[i].button(q, key=f"followup_{i}"):
-                st.session_state['selected_question'] = q
-                st.session_state['step'] = 4
-                st.rerun()
-        user_q = st.text_input("직접 궁금한 점을 입력해도 좋아요! (후속 질문)", value="", key="followup_input")
-        if st.button("Khan 멘토에게 추가 질문하기", key="followup_btn"):
-            st.session_state['selected_question'] = user_q
-            st.session_state['step'] = 4
+        # --- 추가 질문 생성 타이밍 제어 ---
+        if 'extra_questions_ready' not in st.session_state or not st.session_state['extra_questions_ready']:
+            # 답변이 렌더링된 후에만 spinner 돌리기
+            st.session_state['extra_questions_ready'] = True
             st.rerun()
+        elif not st.session_state['extra_questions']:
+            # 백그라운드에서 추가 질문 생성 (스피너 없이)
+            extra_prompt = (
+                f"아래 질문에서 유사하게 궁금할 수 있는 추가 질문 3~4개를 한국어로 만들어줘. 지나치게 세부적인 툴에 대한 얘기보다는 프로덕트, 프로젝트, 리더십에 대한 일반적인 질문으로 만들어. 질문: \"{st.session_state['user_question']}"
+            )
+            try:
+                extra_response = openai_client.chat.completions.create(
+                    model="gpt-4o-mini",
+                    messages=[{"role": "system", "content": extra_prompt}],
+                    temperature=0.5
+                )
+                import re
+                raw = extra_response.choices[0].message.content.strip()
+                questions = re.findall(r'\d+\.\s*(.+)', raw)
+                if not questions:
+                    questions = [q.strip('-• ').strip() for q in raw.split('\n') if q.strip()]
+                st.session_state['extra_questions'] = questions[:4]
+                st.rerun()
+            except Exception as e:
+                st.session_state['extra_questions'] = [f"추가 질문 생성 중 오류: {e}"]
+        else:
+            st.markdown("#### 추가로 궁금한 점이 있으신가요? 아래 예시 질문을 클릭하거나 직접 입력해보세요!")
+            cols = st.columns(len(st.session_state['extra_questions']))
+            for i, q in enumerate(st.session_state['extra_questions']):
+                if cols[i].button(q, key=f"extra_{i}"):
+                    st.session_state['current_input'] = q
+                    st.session_state['user_question'] = ''
+                    st.rerun()
+            user_extra = st.text_input("직접 추가 질문 입력", value="", key="extra_input")
+            if st.button("추가 질문하기", key="extra_btn"):
+                st.session_state['current_input'] = user_extra
+                st.session_state['user_question'] = ''
+                st.rerun()
         st.markdown("---")
         st.caption("Powered by Pinecone, Sentence Transformers, and OpenAI")
-        # 다시 처음으로 돌아가기 버튼
-        if st.button("처음으로 돌아가기"):
-            for k in ['user_state','empathy_message','example_questions','selected_question','khan_answer','step']:
-                st.session_state[k] = '' if k != 'step' else 1
+        if st.button("다른 고민 상담하기"):
+            for k in ['user_question','empathy_message','khan_answer','pinecone_results','extra_questions','current_input','extra_questions_ready']:
+                st.session_state[k] = ''
             st.rerun()
 else:
     st.title("👔 상사에게 잘보이기: 맞춤 보고문 만들기")