8. 실습 ⑤ — Streamlit 웹 화면 (선택)

# ==============================================================
# 15차시. PPT 15차시_기본RAG앱만들기. 코드 폴더는 src/rag_app
# PPT: 15차시 문서 질의응답
# 역할: Streamlit 채팅 화면. 질문을 rag.ask에 넘기고 말풍선과 참고 페이지를 쌓습니다.
# 읽는 키: st.session_state.history, ask()의 answer, sources, cited
# 쓰는 키: 브라우저 화면. history에 (질문, 결과)를 추가
# 그래프:
#   chat_input
#     |
#     v
#   ask(question) --> history
#     |
#     v
#   사용자 말풍선 / 답변 말풍선 / 참고 페이지 / 인용 경고
# --------------------------------------------------------------
# session_state.history가 없으면 빈 리스트로 시작합니다. 새로고침 전까지 대화가 남습니다.
# 출처 페이지 번호는 중복을 빼고 정렬해 '참고: 파일명 N페이지'로 보여 줍니다.
# 파일 맨 아래 주석의 실행 명령은 streamlit run web.py 입니다.
# 서버를 띄우지 않았습니다.
# ==============================================================

# streamlit은 파이썬 코드만으로 웹 화면을 만드는 라이브러리입니다. streamlit run web.py 로 실행합니다.
import streamlit as st

# 검색과 생성은 rag.py의 ask 하나에 맡깁니다. 이 파일은 화면만 그립니다.
from rag import ask

# 브라우저 탭에 보일 제목과 아이콘을 정합니다.
st.set_page_config(
    page_title="문서 Q&A",
    page_icon="■"
)

# 화면 맨 위의 큰 제목입니다.
st.title("■ 문서 기반 질의응답")

# Streamlit은 입력이 있을 때마다 파일을 처음부터 다시 실행합니다. 대화 기록은 session_state에 두어야 사라지지 않습니다.
if "history" not in st.session_state:
    st.session_state.history = []

# 화면 아래 채팅 입력칸입니다. 아직 입력이 없으면 None입니다.
question = st.chat_input(
    "궁금한 것을 물어보세요"
)

# 새 질문이 들어온 실행에서만 ask를 부릅니다.
if question:

    # 답변을 만드는 동안 로딩 메시지를 표시합니다.
    with st.spinner("자료를 찾는 중..."):

        # RAG 시스템에 질문을 전달합니다.
        result = ask(question)

    # 질문과 결과를 대화 기록에 저장합니다.
    st.session_state.history.append(
        (question, result)
    )

# 지금까지 쌓인 대화를 처음부터 다시 그립니다.
for question, result in st.session_state.history:

    # 사용자 말풍선을 먼저, 답변 말풍선을 다음에 그립니다.
    with st.chat_message("user"):
        st.write(question)

    with st.chat_message("assistant"):
        st.write(result["answer"])

        # 출처가 있을 때만 참고 페이지를 붙입니다. 같은 페이지는 한 번만 넣습니다.
        if result["sources"]:
            pages = []

            for source in result["sources"]:
                page = source["page"]
                
                if page not in pages:
                    pages.append(page)
            
            # 페이지 번호를 작은 수부터 정렬합니다.
            pages.sort()

            # 숫자 페이지를 문자열로 변환합니다.
            page_texts = []

            for page in pages:
                page_texts.append(str(page))

            # 1, 2, 3 형태의 문자열로 만듭니다.
            page_text = ", ".join(page_texts)

            # 첫 번째 검색 결과에서 파일 이름을 가져옵니다.
            filename = result["sources"][0]["file"]

            # 참고 문서와 페이지 정보를 출력합니다.
            st.caption(
                f"※ 참고: {filename} "
                f"{page_text}페이지"
            )

# streamlit run web.py