Upload Chatbot.py

pages/Chatbot.py

@@ -1,27 +1,8 @@
 
 import streamlit as st
-import os
-import pickle
-import faiss
 import logging
-
-from multiprocessing import Lock
-from multiprocessing.managers import BaseManager
-from llama_index.callbacks import CallbackManager, LlamaDebugHandler
-from llama_index import VectorStoreIndex, Document,Prompt, SimpleDirectoryReader, ServiceContext, StorageContext, load_index_from_storage
-from llama_index.chat_engine import CondenseQuestionChatEngine;
-from llama_index.node_parser import SimpleNodeParser
-from llama_index.langchain_helpers.text_splitter import TokenTextSplitter
-from llama_index.constants import DEFAULT_CHUNK_OVERLAP
-from llama_index.response_synthesizers import get_response_synthesizer
-from llama_index.vector_stores.faiss import FaissVectorStore
-from llama_index.graph_stores import SimpleGraphStore
-from llama_index.storage.docstore import SimpleDocumentStore
-from llama_index.storage.index_store import SimpleIndexStore
-import tiktoken
-from logging import getLogger, StreamHandler, Formatter
-
 import common
+from llama_index import Prompt
 
 index_name = "./data/storage"
 pkl_name = "./data/stored_documents.pkl"
@@ -39,13 +20,6 @@ custom_prompt = Prompt("""\
   Search query:
 """)
 
-# # list of (human_message, ai_message) tuples
-custom_chat_history = [
-    (
-        'こんにちは、アシスタント。これから質問に答えて下さい。',
-        'こんにちは。了解しました。'
-    )
-]
 chat_history = []
 
 logging.basicConfig(level=logging.INFO)
@@ -55,6 +29,12 @@ logger.debug("調査用ログ")
 common.check_login()
 
 st.title("💬 Chatbot")
+if st.button("リセット",use_container_width=True):
+    st.session_state.chat_engine.reset()
+    st.session_state.messages = [{"role": "assistant", "content": "お困りごとはございますか？"}]
+    st.experimental_rerun()
+    logger.info("reset")
+
 if "messages" not in st.session_state:
     st.session_state["messages"] = [{"role": "assistant", "content": "お困りごとはございますか？"}]
 
@@ -65,13 +45,6 @@ if prompt := st.chat_input():
     st.session_state.messages.append({"role": "user", "content": prompt})
     st.chat_message("user").write(prompt)
     response = st.session_state.chat_engine.chat(prompt)
-    # response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=st.session_state.messages)
     msg = str(response)
     st.session_state.messages.append({"role": "assistant", "content": msg})
     st.chat_message("assistant").write(msg)
-
-if st.button("リセット",use_container_width=True):
-    st.session_state.chat_engine.reset()
-    st.session_state.messages = [{"role": "assistant", "content": "お困りごとはございますか？"}]
-    logger.info("reset")
-