st.experimental_rerun()을 st.rerun()으로 변경 및 UX 문구 일부 개선

app.py

@@ -251,9 +251,13 @@ def generate_khan_answer(query: str, search_results: List[Dict], client: OpenAI)
         logger.error(f"Error during OpenAI API call: {e}", exc_info=True)
         return "답변을 생성하는 중에 문제가 발생했습니다. OpenAI API 키 또는 서비스 상태를 확인해주세요."
 
-# --- Streamlit 앱 UI ---
+# --- Streamlit 앱 UI (리팩토링) ---
 st.set_page_config(page_title="Khan 멘토 (PM 영상 기반)", layout="wide")
 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 키 확인 및 리소스 초기화 ---
@@ -262,95 +266,143 @@ pc = init_pinecone()
 model = load_embedding_model()
 index = get_pinecone_index(pc, INDEX_NAME)
 
-# --- 사용자 입력 ---
-query = st.text_input("멘토에게 질문할 내용을 입력하세요:", placeholder="예: 신입 PM이 가장 먼저 해야 할 일은 무엇인가요?")
-
-# --- 검색 및 답변 생성 실행 ---
-if st.button("Khan 멘토에게 질문하기"):
-    # Always use top_k=3 for Pinecone search
-    if query and index and model and openai_client:
-        with st.spinner("관련 영상을 찾고 Khan 멘토가 답변을 준비하는 중..."):
-            # 1. Pinecone 검색 (Always use top_k=3)
-            pinecone_results = search(query, top_k=5, _index=index, _model=model)
-
-            # 2. OpenAI 답변 생성
-            khan_answer = generate_khan_answer(query, pinecone_results, openai_client)
-
-        # 3. 결과 표시
-        st.subheader("💡 Khan 멘토의 답변")
-        st.markdown(khan_answer) # 생성된 답변 표시
-
-        # 4. 참고 자료 (Pinecone 검색 결과) 표시
-        if pinecone_results:
-            with st.expander("답변에 참고한 영상 정보 보기"):
-                displayed_urls = set() # Keep track of displayed video URLs
-                # Display up to 3 *unique* results based on URL
-                for i, r in enumerate(pinecone_results):
-                    url = r.get('URL', 'N/A')
-
-                    # Skip if this video URL has already been displayed
-                    if url in displayed_urls or url == 'N/A':
-                        continue
-                    
-                    # Add the URL to the set of displayed URLs
-                    displayed_urls.add(url)
-
-                    # --- Display unique video info --- 
-                    st.markdown(f"--- **참고 자료 {len(displayed_urls)} (유사도: {r['점수']:.4f})** ---") # Use length of set for counter
-                    st.markdown(f"**제목:** {r.get('제목', 'N/A')}")
-                    st.markdown(f"**요약:** {r.get('요약', 'N/A')}")
-
-                    timestamp = r.get('타임스탬프', 'N/A')
-                    is_youtube = url and isinstance(url, str) and ('youtube.com' in url or 'youtu.be' in url)
-                    start_seconds = None # Initialize start_seconds
-
-                    # Try to calculate start_seconds if timestamp is valid
-                    if is_youtube and timestamp and timestamp != 'N/A':
-                        start_seconds = parse_timestamp_to_seconds(timestamp)
-
-                    # Display timestamped link (still useful for user)
-                    if is_youtube and start_seconds is not None:
+# --- 상태 관리 ---
+if 'user_state' not in st.session_state:
+    st.session_state['user_state'] = ''
+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. 공감 메시지 생성
+        empathy_prompt = f"""
+        너는 따뜻하고 공감 능력이 뛰어난 상담가야.
+        아래 사용자의 상황을 듣고, 충분히 감정적으로 공감해주고, 용기를 북돋아주는 말��� 해줘.
+        상황: "{st.session_state['user_state']}"
+        """
+        try:
+            empathy_response = openai_client.chat.completions.create(
+                model="gpt-4o",
+                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']}"
+        """
+        try:
+            example_response = openai_client.chat.completions.create(
+                model="gpt-4o",
+                messages=[{"role": "system", "content": example_prompt}],
+                temperature=0.5,
+            )
+            # 응답에서 질문만 추출 (숫자/기호/줄바꿈 등 정제)
+            import re
+            raw = example_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.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()
+
+# --- 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
+    st.subheader("💡 Khan 멘토의 답변")
+    st.markdown(st.session_state['khan_answer'])
+    # 참고 영상 정보 표시
+    if pinecone_results:
+        with st.expander("답변에 참고한 영상 정보 보기"):
+            displayed_urls = set()
+            for i, r in enumerate(pinecone_results):
+                url = r.get('URL', 'N/A')
+                if url in displayed_urls or url == 'N/A':
+                    continue
+                displayed_urls.add(url)
+                st.markdown(f"--- **참고 자료 {len(displayed_urls)} (유사도: {r['점수']:.4f})** ---")
+                st.markdown(f"**제목:** {r.get('제목', 'N/A')}")
+                st.markdown(f"**요약:** {r.get('요약', 'N/A')}")
+                timestamp = r.get('타임스탬프', 'N/A')
+                is_youtube = url and isinstance(url, str) and ('youtube.com' in url or 'youtu.be' in url)
+                start_seconds = None
+                if is_youtube and timestamp and timestamp != 'N/A':
+                    start_seconds = parse_timestamp_to_seconds(timestamp)
+                if is_youtube and start_seconds is not None:
+                    try:
+                        timestamped_link_url = add_timestamp_to_youtube_url(url, timestamp)
+                        st.markdown(f"**영상 링크 (타임스탬프 포함):** [{timestamped_link_url}]({timestamped_link_url})")
+                    except Exception as e:
+                        logger.error(f"Error creating timestamped URL for link: {e}")
+                        st.markdown(f"**영상 링크 (원본):** [{url}]({url})")
+                elif url != "N/A" and isinstance(url, str) and url.startswith("http"):
+                    st.markdown(f"**URL:** [{url}]({url})")
+                else:
+                    st.markdown(f"**URL:** {url}")
+                if is_youtube and url != "N/A":
+                    col1, col2 = st.columns(2)
+                    with col1:
                         try:
-                            # We still generate timestamped URL for the link text
-                            timestamped_link_url = add_timestamp_to_youtube_url(url, timestamp)
-                            st.markdown(f"**영상 링크 (타임스탬프 포함):** [{timestamped_link_url}]({timestamped_link_url})")
+                            st.video(url, start_time=start_seconds or 0)
                         except Exception as e:
-                            logger.error(f"Error creating timestamped URL for link: {e}")
-                            st.markdown(f"**영상 링크 (원본):** [{url}]({url})") # Fallback link
-                    elif url != "N/A" and isinstance(url, str) and url.startswith("http"):
-                        st.markdown(f"**URL:** [{url}]({url})")
-                    else:
-                        st.markdown(f"**URL:** {url}")
-
-                    # Use st.video with original URL and start_time parameter
-                    if is_youtube and url != "N/A":
-                        # Create columns to control width, place video in the first column (50% width)
-                        col1, col2 = st.columns(2)
-                        with col1:
-                            try:
-                                # Pass original URL and calculated start_seconds to st.video
-                                st.video(url, start_time=start_seconds or 0)
-                            except Exception as e:
-                                st.error(f"비디오({url}) 재생 중 오류 발생: {e}")
-                                # Fallback link uses the original URL here as start_time likely failed
-                                st.markdown(f"[YouTube에서 보기]({url})")
-                    elif url != "N/A": # Try st.video for other potential video URLs (no start_time)
-                        # Create columns for non-YouTube videos as well
-                        col1, col2 = st.columns(2)
-                        with col1:
-                            try:
-                                st.video(url)
-                            except Exception as e:
-                                logger.warning(f"st.video failed for non-YouTube URL {url}: {e}")
-
-                    # Remove the display of original timestamp and type
-                    # st.markdown(f"**타임스탬프 (원본):** {timestamp}")
-                    # st.markdown(f"**내용 타입:** {r.get('타입', 'N/A')}")
-                    
-
-    elif not query:
-        st.warning("질문 내용을 입력해주세요.")
-    # API 키 등 다른 요소 부재 시 에러는 각 init 함수에서 처리됨
-
-st.markdown("---")
-st.caption("Powered by Pinecone, Sentence Transformers, and OpenAI") 
+                            st.error(f"비디오({url}) 재생 중 오류 발생: {e}")
+                            st.markdown(f"[YouTube에서 보기]({url})")
+                elif url != "N/A":
+                    col1, col2 = st.columns(2)
+                    with col1:
+                        try:
+                            st.video(url)
+                        except Exception as e:
+                            logger.warning(f"st.video failed for non-YouTube URL {url}: {e}")
+    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
+        st.rerun()