Upload 4 files

app.py

@@ -19,10 +19,12 @@ from llama_index.graph_stores import SimpleGraphStore
 from llama_index.storage.docstore import SimpleDocumentStore
 from llama_index.storage.index_store import SimpleIndexStore
 import tiktoken
+import streamlit_authenticator as stauth
+import yaml
 from logging import getLogger, StreamHandler, Formatter
 
 index_name = "./data/storage"
-pkl_name = "/data/stored_documents.pkl"
+pkl_name = "./data/stored_documents.pkl"
 
 custom_prompt = Prompt("""\
   以下はこれまでの会話履歴と、ドキュメントを検索して回答する必要がある、ユーザーからの会話文です。
@@ -37,41 +39,12 @@ custom_prompt = Prompt("""\
   Search query:
 """)
 
-# # list of (human_message, ai_message) tuples
-custom_chat_history = [
-    (
-        'こんにちは、アシスタント。これから質問に答えて下さい。',
-        'こんにちは。了解しました。'
-    )
-]
 chat_history = []
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger("__name__")
 logger.debug("調査用ログ")
 
-
-st.title("💬 Chatbot")
-if "messages" not in st.session_state:
-    st.session_state["messages"] = [{"role": "assistant", "content": "お困りごとはございますか？"}]
-
-for msg in st.session_state.messages:
-    st.chat_message(msg["role"]).write(msg["content"])
-
-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")
-
 def initialize_index():
     logger.info("initialize_index start")
     text_splitter = TokenTextSplitter(separator="。", chunk_size=1500
@@ -125,9 +98,32 @@ def initialize_index():
         else:
             st.session_state.stored_docs=list()
 
-if __name__ == "__main__":
-    # init the global index
-    logger.info("main start")
-    if "chat_engine" not in st.session_state:
-      initialize_index()
-      logger.info("initializing index...")
+with open('config.yaml') as file:
+    config = yaml.load(file, Loader=yaml.SafeLoader)
+
+authenticator = stauth.Authenticate(
+    config['credentials'],
+    config['cookie']['name'],
+    config['cookie']['key'],
+    config['cookie']['expiry_days'],
+    config['preauthorized'],
+)
+
+name, authentication_status, username = authenticator.login('Login', 'main')
+
+
+if 'authentication_status' not in st.session_state:
+    st.session_state['authentication_status'] = None
+
+if st.session_state["authentication_status"]:
+    authenticator.logout('Logout', 'main')
+    st.write(f'ログインに成功しました')
+    initialize_index()
+		# ここにログイン後の処理を書く。
+elif st.session_state["authentication_status"] is False:
+    st.error('ユーザ名またはパスワードが間違っています')
+elif st.session_state["authentication_status"] is None:
+    st.warning('ユーザ名やパスワードを入力してください')
+
+
+

common.py

@@ -0,0 +1,15 @@
+# common.py
+import extra_streamlit_components as stx
+import streamlit as st
+import logging
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger("__name__")
+logger.debug("調査用ログ")
+
+#ログインの確認
+def check_login():
+    value = stx.CookieManager().get(cookie='some_cookie_name')
+    if value is None:
+        st.warning("**ログインしてください**")
+        st.stop()

config.yaml

@@ -0,0 +1,13 @@
+cookie:
+  expiry_days: 10
+  key: some_signature_keys
+  name: some_cookie_name
+credentials:
+  usernames:
+    stylez:
+      email: [email protected]
+      name: test user
+      password: $2b$12$LUwDsoPgN7T3v1xK42rGR.yCiDauGBH4fYeia7CN84F2v8tY3IEs2
+preauthorized:
+  emails:
+  - [email protected]

requirements.txt

@@ -6,4 +6,6 @@ anthropic
 llama-index==0.7.4
 pypdf==3.9.0
 faiss-cpu==1.7.4
-html2text
+html2text
+streamlit-authenticator
+extra_streamlit_components