Update app.py

app.py

@@ -94,6 +94,18 @@ def get_conversation_chain(vectorstore):
     )
     return conversation_chain
 
+# 사용자 입력을 처리하는 함수입니다.
+from transformers import pipeline
+
+# Hugging Face의 question-answering 모델을 로드합니다.
+model_name = "bert-base-uncased"
+qa_model = pipeline("question-answering", model=model_name)
+
+def generate_answer(question, context):
+    # 질문과 문맥을 입력으로 받아 답변을 생성합니다.
+    answer = qa_model(question=question, context=context)
+    return answer["answer"]
+
 # 사용자 입력을 처리하는 함수입니다.
 def handle_userinput(user_question):
     # 대화 체인을 사용하여 사용자 질문에 대한 응답을 생성합니다.
@@ -103,11 +115,15 @@ def handle_userinput(user_question):
 
     for i, message in enumerate(st.session_state.chat_history):
         if i % 2 == 0:
-            st.write(user_template.replace(
-                "{{MSG}}", message.content), unsafe_allow_html=True)
+            st.write(user_template.replace("{{MSG}}", message.content), unsafe_allow_html=True)
         else:
-            st.write(bot_template.replace(
-                "{{MSG}}", message.content), unsafe_allow_html=True)
+            st.write(bot_template.replace("{{MSG}}", message.content), unsafe_allow_html=True)
+
+        # 질문에 대한 답변을 생성하여 출력합니다.
+        if i % 2 == 1 and i > 0:
+            prev_user_message = st.session_state.chat_history[i-1].content
+            answer = generate_answer(message.content, prev_user_message)
+            st.write(bot_template.replace("{{MSG}}", answer), unsafe_allow_html=True)
 
 
 def main():