Update app.py

app.py

@@ -10,7 +10,6 @@ from langchain.retrievers.document_compressors import (
     CrossEncoderReranker,
     FlashrankRerank,
 )
-from langchain_core.vectorstores import VectorStore
 from langchain.storage import LocalFileStore
 from langchain_community.cross_encoders import HuggingFaceCrossEncoder
 from langchain_community.document_loaders.generic import GenericLoader
@@ -21,6 +20,7 @@ from langchain_community.retrievers import BM25Retriever
 from langchain_community.vectorstores import FAISS
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.runnables import RunnableLambda, RunnablePassthrough
+from langchain_core.vectorstores import VectorStore
 from langchain_huggingface import HuggingFaceEmbeddings
 from langchain_text_splitters import Language, RecursiveCharacterTextSplitter
 
@@ -120,43 +120,49 @@ def setup_retrievers_and_chain(
     return rag_chain
 
 
-def main():
-    st.title("Code QA Bot")
+def sidebar_content():
+    st.sidebar.title("사용 가이드")
+    st.sidebar.info(
+        """
+    1. 왼쪽 텍스트 영역에 질문을 입력하세요.
+    2. '답변 생성' 버튼을 클릭하세요.
+    3. 답변이 아래에 표시됩니다.
+    4. 새로운 질문을 하려면 '답변 초기화' 버튼을 사용하세요.
+    """
+    )
 
-    # Initialize session state for project folder and answer
-    if "project_folder" not in st.session_state:
-        st.session_state.project_folder = ""
-    if "answer" not in st.session_state:
+    if st.sidebar.button("답변 초기화", key="reset"):
         st.session_state.answer = ""
+        st.experimental_rerun()
 
-    # 프로젝트 경로 입력 받기
-    project_folder = st.text_input(
-        "Enter the project folder path:", value=st.session_state.project_folder
-    )
-    st.session_state.project_folder = project_folder
 
-    if project_folder:
-        # 프로젝트 경로가 입력되면 벡터 스토어와 체인 설정
-        db = setup_embeddings_and_db(project_folder)
-        rag_chain = setup_retrievers_and_chain(db, project_folder)
+def main():
+    st.set_page_config(page_title="WDS QA 봇", page_icon="🤖", layout="wide")
+
+    sidebar_content()
 
-        # 사용자 질문 입력 받기
-        user_question = st.text_input("Ask a question about the code:")
+    st.title("🤖 WDS QA 봇")
 
-        # Add a button to reset the answer
-        if st.button("Reset Answer"):
-            st.session_state.answer = ""
+    st.subheader("질문하기")
+    user_question = st.text_area("코드에 대해 궁금한 점을 물어보세요:", height=100)
 
+    if st.button("답변 생성", key="generate"):
         if user_question:
-            with st.spinner("Generating answer..."):
+            with st.spinner("답변을 생성 중입니다..."):
+                project_folder = "wds"
+                db = setup_embeddings_and_db(project_folder)
+                rag_chain = setup_retrievers_and_chain(db, project_folder)
                 response = rag_chain.invoke(user_question)
                 st.session_state.answer = response
+        else:
+            st.warning("질문을 입력해주세요.")
+
+    if "answer" in st.session_state and st.session_state.answer:
+        st.subheader("답변")
+        st.markdown(st.session_state.answer)
 
-        # Display the answer
-        if st.session_state.answer:
-            st.write(st.session_state.answer)
-    else:
-        st.warning("Please enter the project folder path to proceed.")
+    st.markdown("---")
+    st.caption("© 2023 WDS QA 봇. 모든 권리 보유.")
 
 
 if __name__ == "__main__":