Upload src/app.py with huggingface_hub

src/app.py

@@ -1,32 +1,39 @@
 import streamlit as st
 from pathlib import Path
- 
+import streamlit.components.v1 as components
+import html
+from langchain_community.document_loaders import TextLoader, CSVLoader
+
+from openai import OpenAI
+import uuid
+import tempfile
 from langchain_community.document_loaders import PyMuPDFLoader
 from langchain_text_splitters import RecursiveCharacterTextSplitter
 from langchain_openai import OpenAIEmbeddings, ChatOpenAI
 from langchain_community.vectorstores import FAISS
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.output_parsers import StrOutputParser
- 
+
 st.set_page_config(
     page_title="AIVLE 학습도우미",
     page_icon="🤖",
     layout="wide"
 )
- 
+
+# PDF_PATH = Path("Docs/AIVLE_School_백서_통합본_최종.pdf")
+
 BASE_DIR = Path(__file__).resolve().parent
 PDF_PATH = BASE_DIR / "AIVLE_School_백서_통합본_최종.pdf"
- 
-# =========================
-# CSS
-# =========================
+client = OpenAI()
+
+
 st.markdown("""
 <style>
 .main .block-container {
     padding-top: 2rem;
     max-width: 1100px;
 }
- 
+
 .hero-box {
     background: #d9f3f2;
     border-radius: 22px;
@@ -36,46 +43,35 @@ st.markdown("""
     justify-content: space-between;
     align-items: center;
 }
- 
+
 .hero-title {
     font-size: 30px;
     font-weight: 800;
     color: #111827;
     margin-bottom: 10px;
 }
- 
+
 .hero-sub {
     font-size: 15px;
     color: #6b7280;
 }
- 
+
 .robot {
     font-size: 72px;
 }
- 
+
 .section-title {
     font-size: 17px;
     font-weight: 800;
     margin-bottom: 12px;
 }
- 
-.question-card {
-    background: white;
-    border: 1px solid #e5e7eb;
-    border-radius: 14px;
-    padding: 16px 18px;
-    font-size: 15px;
-    font-weight: 600;
-    color: #374151;
-    box-shadow: 0 2px 8px rgba(0,0,0,0.03);
-}
- 
+
 .chat-wrap {
     border-top: 1px solid #e5e7eb;
     margin-top: 28px;
     padding-top: 26px;
 }
- 
+
 .user-bubble {
     background: #d9f3f2;
     padding: 14px 18px;
@@ -86,7 +82,7 @@ st.markdown("""
     margin-bottom: 8px;
     font-weight: 500;
 }
- 
+
 .assistant-card {
     background: white;
     border: 1px solid #e5e7eb;
@@ -96,50 +92,59 @@ st.markdown("""
     max-width: 780px;
     box-shadow: 0 2px 10px rgba(0,0,0,0.04);
 }
- 
+
 .assistant-name {
     font-size: 14px;
     color: #6b7280;
     font-weight: 700;
     margin-bottom: 8px;
 }
- 
-.time-text {
-    color: #9ca3af;
-    font-size: 12px;
-    text-align: right;
-}
- 
-.input-box {
-    border: 1px solid #e5e7eb;
-    border-radius: 18px;
-    padding: 16px;
-    margin-top: 30px;
-    background: white;
-}
- 
-.side-history {
-    background: #f9fafb;
-    border-radius: 12px;
-    padding: 10px 12px;
-    margin-bottom: 8px;
-    font-size: 14px;
-}
 </style>
 """, unsafe_allow_html=True)
- 
+
 # =========================
 # 세션 상태
 # =========================
-if "messages" not in st.session_state:
-    st.session_state.messages = []
- 
-if "chat_history_titles" not in st.session_state:
-    st.session_state.chat_history_titles = []
- 
+if "conversations" not in st.session_state:
+    first_id = str(uuid.uuid4())
+    st.session_state.conversations = {
+        first_id: {
+            "title": "새 대화",
+            "messages": []
+        }
+    }
+    st.session_state.current_chat_id = first_id
+
+if "current_chat_id" not in st.session_state:
+    st.session_state.current_chat_id = list(st.session_state.conversations.keys())[0]
+
 if "show_faq" not in st.session_state:
     st.session_state.show_faq = False
- 
+
+if "faq_category" not in st.session_state:
+    st.session_state.faq_category = "전체"
+
+if "logged_in" not in st.session_state:
+    st.session_state.logged_in = False
+
+if "user_name" not in st.session_state:
+    st.session_state.user_name = ""
+
+if "show_login" not in st.session_state:
+    st.session_state.show_login = False
+if "recommended_questions" not in st.session_state:
+    st.session_state.recommended_questions = [
+        "미니 프로젝트는 어떻게 진행되나요?",
+        "프로젝트 주제는 어떻게 정하나요?",
+        "팀 프로젝트에서 무엇을 기록해야 하나요?",
+        "머신러닝 모델 성능은 어떻게 평가하나요?"
+    ]
+if "show_help" not in st.session_state:
+    st.session_state.show_help = False
+
+if "uploaded_retriever" not in st.session_state:
+    st.session_state.uploaded_retriever = None
+
 # =========================
 # RAG
 # =========================
@@ -148,106 +153,275 @@ def build_rag_chain():
     if not PDF_PATH.exists():
         st.error(f"PDF 파일을 찾을 수 없습니다: {PDF_PATH}")
         st.stop()
- 
+
     loader = PyMuPDFLoader(str(PDF_PATH))
     docs = loader.load()
- 
+
     splitter = RecursiveCharacterTextSplitter(
         chunk_size=1200,
         chunk_overlap=200,
         separators=["\n\n", "\n", ".", " ", ""]
     )
     chunks = splitter.split_documents(docs)
- 
+
     embedding = OpenAIEmbeddings(model="text-embedding-3-small")
     vectorstore = FAISS.from_documents(chunks, embedding)
-    retriever = vectorstore.as_retriever(search_kwargs={"k": 8})
- 
+    retriever = vectorstore.as_retriever(search_kwargs={"k": 3})
+
     llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
- 
+
     prompt = ChatPromptTemplate.from_template("""
 당신은 AIVLE School 백서 PDF 내용을 기반으로 답변하는 학습도우미 챗봇입니다.
- 
+
 규칙:
 1. 반드시 [문서 내용]에 근거해서 답변하세요.
 2. 문서에서 확인되지 않는 내용은 "백서에서 확인되지 않습니다."라고 답하세요.
 3. 답변은 한국어로 친절하고 간결하게 작성하세요.
- 
+
 [문서 내용]
 {context}
- 
+
 [질문]
 {question}
 """)
- 
+
     def format_docs(docs):
         return "\n\n".join(doc.page_content for doc in docs)
- 
+
     chain = prompt | llm | StrOutputParser()
- 
+
     return retriever, chain, format_docs
- 
+
 retriever, rag_chain, format_docs = build_rag_chain()
- 
+
+def build_uploaded_retriever(uploaded_file):
+
+    suffix = Path(uploaded_file.name).suffix
+
+    with tempfile.NamedTemporaryFile(
+        delete=False,
+        suffix=suffix
+    ) as tmp:
+
+        tmp.write(uploaded_file.getvalue())
+        tmp_path = tmp.name
+
+    if suffix == ".pdf":
+        loader = PyMuPDFLoader(tmp_path)
+
+    elif suffix == ".txt":
+        loader = TextLoader(
+            tmp_path,
+            encoding="utf-8"
+        )
+
+    elif suffix == ".csv":
+        loader = CSVLoader(tmp_path)
+
+    else:
+        return None
+
+    docs = loader.load()
+
+    splitter = RecursiveCharacterTextSplitter(
+        chunk_size=1000,
+        chunk_overlap=150
+    )
+
+    chunks = splitter.split_documents(docs)
+
+    embedding = OpenAIEmbeddings(
+        model="text-embedding-3-small"
+    )
+
+    vectorstore = FAISS.from_documents(
+        chunks,
+        embedding
+    )
+
+    return vectorstore.as_retriever(
+        search_kwargs={"k": 3}
+    )
+
+
+
+
+def generate_recommend_questions(user_question):
+    docs = retriever.invoke(user_question)
+    context = format_docs(docs)
+
+    recommend_prompt = ChatPromptTemplate.from_template("""
+당신은 AIVLE School 백서 내용을 기반으로 '추천 질문'만 만드는 도우미입니다.
+
+규칙:
+1. 답변을 절대 작성하지 마세요.
+2. 설명도 절대 작성하지 마세요.
+3. 추천 질문만 4개 작성하세요.
+4. 각 줄은 반드시 물음표(?)로 끝나는 질문 문장이어야 합니다.
+5. 번호, 불릿, 따옴표, 접두어를 쓰지 마세요.
+6. 백서에 근거한 질문만 만드세요.
+7. 사용자의 질문과 직접 관련된 질문만 생성하세요.
+
+[백서 내용]
+{context}
+
+[사용자 질문]
+{question}
+""")
+
+    recommend_chain = (
+        recommend_prompt
+        | ChatOpenAI(model="gpt-4o-mini", temperature=0.2)
+        | StrOutputParser()
+    )
+
+    result = recommend_chain.invoke({
+        "context": context,
+        "question": user_question
+    })
+
+    questions = []
+
+    for line in result.split("\n"):
+        line = line.strip()
+
+        if not line:
+            continue
+
+        if not line.endswith("?"):
+            continue
+
+        line = line.lstrip("-•0123456789. ").strip()
+
+        questions.append(line)
+
+    return questions[:4]
+
+
+def copy_button(text, key):
+    safe_text = html.escape(text).replace("\n", "\\n").replace("'", "\\'")
+
+    components.html(
+        f"""
+        <button onclick="navigator.clipboard.writeText('{safe_text}')"
+            style="
+                margin-top:8px;
+                padding:6px 12px;
+                border-radius:8px;
+                border:1px solid #d1d5db;
+                background:white;
+                cursor:pointer;
+                font-size:13px;
+            ">
+            📋 복사
+        </button>
+        """,
+        height=40
+    )
+
+def generate_tts(text, filename):
+    speech_file = f"temp/{filename}.mp3"
+
+    Path("temp").mkdir(exist_ok=True)
+
+    response = client.audio.speech.create(
+        model="gpt-4o-mini-tts",
+        voice="nova",
+        input=text
+    )
+
+    response.stream_to_file(speech_file)
+
+    return speech_file
+
+def transcribe_audio(audio_file):
+    transcript = client.audio.transcriptions.create(
+        model="whisper-1",
+        file=audio_file
+    )
+
+    return transcript.text
+
 # =========================
 # 사이드바
 # =========================
 with st.sidebar:
+
+    st.image(
+        "images/에이블학습도우미 로고.png",
+        width=270
+    )
+
     st.markdown("## AIVLE")
     st.markdown("### 학습도우미 챗봇")
     st.caption("에이블스쿨의 학습 여정을 함께하는 AI 도우미입니다.")
- 
+
     if st.button("＋ 새 대화", use_container_width=True):
-        st.session_state.messages = []
+        new_id = str(uuid.uuid4())
+        st.session_state.conversations[new_id] = {
+            "title": "새 대화",
+            "messages": []
+        }
+        st.session_state.current_chat_id = new_id
         st.rerun()
- 
+
     st.divider()
- 
+
     st.button("🏠 홈", use_container_width=True)
+
     if st.button("❔ FAQ", use_container_width=True):
         st.session_state.show_faq = True
- 
+        st.rerun()
+
     st.divider()
- 
+
     st.markdown("### 파일 관리")
-    st.file_uploader("파일 업로드", type=["pdf", "txt", "csv"])
-    st.button("📁 업로드 목록", use_container_width=True)
- 
+
+    uploaded_file = st.file_uploader(
+        "파일 업로드",
+        type=["pdf", "txt", "csv"]
+    )
+
+    if uploaded_file is not None:
+
+        st.session_state.uploaded_retriever = (
+            build_uploaded_retriever(uploaded_file)
+        )
+
+        st.success(
+            "업로드한 파일을 챗봇 참고 자료로 추가했습니다."
+        )
+
+    st.button(
+        "📁 업로드 목록",
+        use_container_width=True
+    )
+
     st.divider()
- 
+
     st.markdown("### 대화 기록")
-    if st.session_state.chat_history_titles:
-        for title in reversed(st.session_state.chat_history_titles[-5:]):
-            st.markdown(f'<div class="side-history">{title}</div>', unsafe_allow_html=True)
-    else:
-        st.caption("아직 대화 기록이 없습니다.")
- 
-    # 여기부터 로그인 코드도 사이드바 안
-    if "logged_in" not in st.session_state:
-        st.session_state.logged_in = False
- 
-    if "user_name" not in st.session_state:
-        st.session_state.user_name = ""
- 
-    if "show_login" not in st.session_state:
-        st.session_state.show_login = False
- 
+    for chat_id, chat in reversed(list(st.session_state.conversations.items())):
+        title = chat["title"]
+        if st.button(f"💬 {title}", key=f"chat_{chat_id}", use_container_width=True):
+            st.session_state.current_chat_id = chat_id
+            st.session_state.show_faq = False
+            st.rerun()
+
     st.divider()
- 
+
     if not st.session_state.logged_in:
         st.markdown("### 👤 계정")
- 
+
         if st.button("로그인", use_container_width=True):
             st.session_state.show_login = not st.session_state.show_login
- 
+
         if st.session_state.show_login:
             with st.form("sidebar_login_form", clear_on_submit=False):
                 login_name = st.text_input("이름", key="login_name_input")
                 login_id = st.text_input("아이디", key="login_id_input")
                 login_pw = st.text_input("비밀번호", type="password", key="login_pw_input")
- 
                 login_btn = st.form_submit_button("로그인", use_container_width=True)
- 
+
             if login_btn:
                 if login_name.strip() and login_id.strip() and login_pw.strip():
                     st.session_state.logged_in = True
@@ -256,219 +430,194 @@ with st.sidebar:
                     st.rerun()
                 else:
                     st.warning("이름, 아이디, 비밀번호를 모두 입력하세요.")
- 
     else:
         st.markdown(f"👤 **{st.session_state.user_name}**")
-        st.caption("에이블스쿨 7기")
- 
+        st.caption("에이블스쿨 9기")
+
         if st.button("로그아웃", use_container_width=True):
             st.session_state.logged_in = False
             st.session_state.user_name = ""
             st.session_state.show_login = False
             st.rerun()
- 
- 
+
+    st.divider()
+
+    st.markdown("### 도움이 필요하신가요?")
+
+    if st.button("💬 문의하기", use_container_width=True):
+        st.session_state.show_help = True
+
+
+
+
+# =========================
+# FAQ 데이터
+# =========================
+faq_data = {
+    "전체": [
+        ("미니 프로젝트는 어떻게 진행되나요?", "미니 프로젝트는 주제 선정, 기획, 구현, 발표, 피드백 단계로 진행됩니다."),
+        ("미니 프로젝트 팀은 어떻게 구성되나요?", "팀은 과정 운영 방식에 따라 구성됩니다."),
+        ("프로젝트 주제는 어떻게 정하나요?", "문제 정의, 데이터 활용 가능성, 구현 가능성을 고려해 정합니다."),
+        ("머신러닝 모델 성능은 어떻게 평가하나요?", "Accuracy, Precision, Recall, F1 Score 등을 사용합니다."),
+        ("데이터 전처리는 왜 중요한가요?", "데이터 품질이 분석 결과와 모델 성능에 직접적인 영향을 주기 때문입니다.")
+    ],
+    "학습 과정": [
+        ("학습은 어떤 방식으로 진행되나요?", "이론 학습, 실습, 프로젝트, 피드백 중심으로 진행됩니다."),
+        ("복습은 어떻게 하면 좋나요?", "실습 코드 재작성, 개념 요약, 오류 해결 과정을 반복하는 것이 좋습니다.")
+    ],
+    "프로젝트": [
+        ("미니 프로젝트는 어떻게 진행되나요?", "주제 선정, 기획, 구현, 발표, 피드백 단계로 진행됩니다."),
+        ("프로젝트 주제는 어떻게 정하나요?", "문제 정의와 구현 가능성을 중심으로 정합니다."),
+        ("팀 프로젝트에서 무엇을 기록해야 하나요?", "문제 정의, 데이터 근거, 본인 역할, 의사결정 과정, 산출물, 한계와 개선 방향을 꾸준히 기록해야 합니다.")
+    ],
+    "KDT": [
+        ("교육 수료 기준은 어떻게 되나요?", "총 훈련일수의 일정 비율 이상 출석해야 수료가 가능하며, 훈련 수준 유지를 위해 100% 참여가 권장됩니다."),
+        ("중도 포기 시 내일배움카드 차감액이 있나요?", "국민내일배움카드 훈련과정 중도포기 시 회차에 따라 차감 기준이 적용될 수 있습니다. 세부 금액과 기준은 고용센터 안내를 확인해야 합니다."),
+        ("훈련장려금은 무엇인가요?", "장기간 국비 과정을 수강하는 교육생의 경제적 부담을 줄이기 위해 교통비와 식비 명목으로 제공되는 지원금입니다."),
+        ("훈련장려금 지급 기준은 어떻게 되나요?", "단위기간 출석률, 고용 상태, 소득 조건 등 지급 기준을 충족해야 수령할 수 있습니다. 개인별 상황에 따라 달라질 수 있습니다.")
+    ],
+    "시스템/접속": [
+        ("접속 오류가 나면 어떻게 하나요?", "인터넷 연결, 계정 정보, 브라우저 캐시를 먼저 확인합니다."),
+        ("파일 업로드가 안 되면 어떻게 하나요?", "파일 형식과 용량 제한을 확인해야 합니다.")
+    ],
+    "기타": [
+        ("문의는 어디로 하나요?", "운영진 또는 지정된 문의 채널을 통해 문의하면 됩니다.")
+    ]
+}
+
 # =========================
 # FAQ 화면
 # =========================
 if st.session_state.show_faq:
- 
-    st.markdown("""
-    <style>
-    .faq-box-wrap {
-        border: 2px solid #8b5cf6;
-        border-radius: 12px;
-        background: white;
-        padding: 24px;
-        margin-bottom: 24px;
-    }
- 
-    .faq-title {
-        font-size: 26px;
-        font-weight: 800;
-    }
- 
-    .faq-sub {
-        color: #6b7280;
-        font-size: 15px;
-        margin-top: 4px;
-    }
- 
-    .faq-category {
-        background: #e0f7f5;
-        color: #0f766e;
-        padding: 12px;
-        border-radius: 12px;
-        font-weight: 700;
-        margin-bottom: 10px;
-    }
- 
-    .faq-category-normal {
-        padding: 12px;
-        font-weight: 700;
-        color: #374151;
-        margin-bottom: 10px;
-    }
-    </style>
-    """, unsafe_allow_html=True)
- 
-    st.markdown('<div class="faq-box-wrap">', unsafe_allow_html=True)
- 
-    col_title, col_close = st.columns([10, 1])
- 
-    with col_title:
-        st.markdown(
-            '<div class="faq-title">❔ FAQ</div>',
-            unsafe_allow_html=True
-        )
- 
-        st.markdown(
-            '<div class="faq-sub">자주 묻는 질문을 확인해보세요.</div>',
-            unsafe_allow_html=True
-        )
- 
-    with col_close:
-        if st.button("✕", key="close_faq"):
-            st.session_state.show_faq = False
-            st.rerun()
- 
+    st.markdown("## ❔ FAQ")
+    st.caption("자주 묻는 질문을 확인해보세요.")
+
+    if st.button("✕ FAQ 닫기"):
+        st.session_state.show_faq = False
+        st.rerun()
+
     left, right = st.columns([1, 3])
- 
+
     with left:
-        st.markdown("**카테고리**")
- 
-        st.markdown(
-            '<div class="faq-category">전체</div>',
-            unsafe_allow_html=True
-        )
- 
-        st.markdown(
-            '<div class="faq-category-normal">학습 과정</div>',
-            unsafe_allow_html=True
-        )
- 
-        st.markdown(
-            '<div class="faq-category-normal">미니 프로젝트</div>',
-            unsafe_allow_html=True
-        )
- 
-        st.markdown(
-            '<div class="faq-category-normal">AI/데이터</div>',
-            unsafe_allow_html=True
-        )
- 
-        st.markdown(
-            '<div class="faq-category-normal">시스템/접속</div>',
-            unsafe_allow_html=True
-        )
- 
-        st.markdown(
-            '<div class="faq-category-normal">기타</div>',
-            unsafe_allow_html=True
-        )
- 
+        st.markdown("### 카테고리")
+        for category in faq_data.keys():
+            if st.button(category, key=f"faq_{category}", use_container_width=True):
+                st.session_state.faq_category = category
+                st.rerun()
+
     with right:
-        st.markdown("**전체**")
- 
-        with st.expander(
-            "미니 프로젝트는 어떻게 진행되나요?",
-            expanded=True
-        ):
-            st.write(
-                "미니 프로젝트는 주제 선정, 기획, 구현, 발표, 피드백 단계로 진행됩니다."
-            )
- 
-        with st.expander(
-            "머신러닝 모델 성능은 어떻게 평가하나요?",
-            expanded=True
-        ):
-            st.write(
-                "Accuracy, Precision, Recall, F1 Score 등을 사용합니다."
-            )
- 
-    st.markdown('</div>', unsafe_allow_html=True)
- 
- 
- 
+        selected = st.session_state.faq_category
+        st.markdown(f"### {selected}")
+
+        for question, answer in faq_data[selected]:
+            with st.expander(question):
+                st.write(answer)
+
+    st.stop()
+
 # =========================
 # 메인 화면 헤더
 # =========================
-st.markdown("""
-<div class="hero-box">
-    <div>
-        <div class="hero-title">안녕하세요! 😊<br>무엇을 도와드릴까요?</div>
-        <div class="hero-sub">학습, 프로젝트, AI, 데이터 분석 등 무엇이든 물어보세요!</div>
-    </div>
-    <div class="robot">🤖</div>
-</div>
-""", unsafe_allow_html=True)
- 
+
+st.image(
+    "images/이미지.png",
+    use_container_width=True
+)
+
 # =========================
 # 추천 질문
 # =========================
 st.markdown('<div class="section-title">추천 질문</div>', unsafe_allow_html=True)
- 
+
+recommended = st.session_state.recommended_questions
+
 q1, q2 = st.columns(2)
- 
+
 with q1:
-    if st.button("📌 미니 프로젝트는 어떻게 진행되나요?", use_container_width=True):
-        st.session_state.pending_question = "미니 프로젝트는 어떻게 진행되나요?"
-        st.rerun()
- 
-    if st.button("🐍 Python에서 리스트 컴프리헨션이 뭐예요?", use_container_width=True):
-        st.session_state.pending_question = "Python에서 리스트 컴프리헨션이 뭐예요?"
-        st.rerun()
- 
+    for idx in [0, 1]:
+        if idx < len(recommended):
+            if st.button(
+                recommended[idx],
+                key=f"rec_{idx}",
+                use_container_width=True
+            ):
+                st.session_state.pending_question = recommended[idx]
+                st.rerun()
+
 with q2:
-    if st.button("👥 팀 프로젝트 협업 팁을 알려주세요.", use_container_width=True):
-        st.session_state.pending_question = "팀 프로젝트 협업 팁을 알려주세요."
-        st.rerun()
- 
-    if st.button("📊 머신러닝 모델 성능은 어떻게 평가하나요?", use_container_width=True):
-        st.session_state.pending_question = "머신러닝 모델 성능은 어떻게 평가하나요?"
-        st.rerun()
- 
+    for idx in [2, 3]:
+        if idx < len(recommended):
+            if st.button(
+                recommended[idx],
+                key=f"rec_{idx}",
+                use_container_width=True
+            ):
+                st.session_state.pending_question = recommended[idx]
+                st.rerun()
+
 # =========================
 # 질문 처리 함수
 # =========================
 def answer_question(question):
-    st.session_state.messages.append({"role": "user", "content": question})
- 
-    title = question[:22] + "..." if len(question) > 22 else question
-    if title not in st.session_state.chat_history_titles:
-        st.session_state.chat_history_titles.append(title)
- 
+    chat_id = st.session_state.current_chat_id
+    chat = st.session_state.conversations[chat_id]
+
+    chat["messages"].append({"role": "user", "content": question})
+
+    if chat["title"] == "새 대화":
+        chat["title"] = question[:18] + "..." if len(question) > 18 else question
+
+
+
     docs = retriever.invoke(question)
-    context = format_docs(docs)
- 
+
+    all_docs = list(docs)
+
+    if st.session_state.uploaded_retriever is not None:
+
+        uploaded_docs = (
+            st.session_state.uploaded_retriever.invoke(question)
+        )
+
+        all_docs.extend(uploaded_docs)
+
+    context = format_docs(all_docs)
+
     answer = rag_chain.invoke({
         "context": context,
         "question": question
     })
- 
-    st.session_state.messages.append({"role": "assistant", "content": answer})
- 
+
+    chat["messages"].append({"role": "assistant", "content": answer})
+
+    st.session_state.recommended_questions = generate_recommend_questions(question)
+
 # 추천 질문 클릭 처리
 if "pending_question" in st.session_state:
     question = st.session_state.pending_question
     del st.session_state.pending_question
     answer_question(question)
     st.rerun()
- 
+
 # =========================
 # 채팅 영역
 # =========================
+current_chat = st.session_state.conversations[st.session_state.current_chat_id]
+messages = current_chat["messages"]
+
 st.markdown('<div class="chat-wrap">', unsafe_allow_html=True)
- 
-if not st.session_state.messages:
+
+if not messages:
     st.markdown("""
-    <div class="assistant-name">🤖 AIVLE 도우미</div>
-    <div class="assistant-card">
-        안녕하세요! AIVLE 백서를 기반으로 학습과 프로젝트 관련 질문에 답변드릴게요.
+    <div style="min-height: 560px;">
+        <div class="assistant-name">🤖 AIVLE 도우미</div>
+        <div class="assistant-card">
+            안녕하세요! AIVLE 백서를 기반으로 학습과 프로젝트 관련 질문에 답변드릴게요.
+        </div>
     </div>
     """, unsafe_allow_html=True)
- 
-for msg in st.session_state.messages:
+
+for idx, msg in enumerate(messages):
     if msg["role"] == "user":
         st.markdown(
             f'<div class="user-bubble">{msg["content"]}</div>',
@@ -482,14 +631,60 @@ for msg in st.session_state.messages:
             """,
             unsafe_allow_html=True
         )
- 
+
+
+
+        copy_button(msg["content"], key=f"copy_{idx}")
+
+
+        if st.button("🔊 음성으로 듣기", key=f"tts_{idx}"):
+
+            audio_path = generate_tts(
+                msg["content"],
+                f"audio_{idx}"
+            )
+
+            st.audio(audio_path)
+
+
+
 st.markdown('</div>', unsafe_allow_html=True)
- 
+
 # =========================
 # 입력 영역
 # =========================
-user_input = st.chat_input("메시지를 입력하세요...")
- 
+col1, col2 = st.columns([5, 1])
+
+with col1:
+    user_input = st.chat_input("메시지를 입력하세요...")
+
+with col2:
+    audio_file = st.audio_input(
+        "🎤 음성 질문",
+        label_visibility="collapsed"
+    )
+
+# 텍스트 질문
 if user_input:
     answer_question(user_input)
     st.rerun()
+
+# 음성 질문
+if audio_file is not None:
+
+    with st.spinner("음성을 분석하는 중입니다..."):
+
+        question = transcribe_audio(audio_file)
+
+        current_chat = st.session_state.conversations[
+            st.session_state.current_chat_id
+        ]
+
+        current_chat["messages"].append({
+            "role": "user",
+            "content": f"🎤 {question}"
+        })
+
+        answer_question(question)
+
+    st.rerun()