Update app.py

app.py

@@ -6,14 +6,16 @@ from langchain.embeddings import HuggingFaceEmbeddings
 from langchain.vectorstores import FAISS
 from langchain.llms import CTransformers
 from langchain.chains import ConversationalRetrievalChain
-from dl_hf_model import dl_hf_model
 from ctransformers import AutoModelForCausalLM
 from langchain_g4f import G4FLLM
 from g4f import Provider, models
+import unicodedata
 import requests
 # Define the path for generated embeddings
 DB_FAISS_PATH = 'vectorstore/db_faiss'
-
+def is_japanese_character(character):
+    return 'CJK UNIFIED' in unicodedata.name(character, '')
+    
 # Load the model of choice
 def load_llm():
     # url = "https://huggingface.co/TheBloke/Llama-2-7B-Chat-GGML/blob/main/llama-2-7b-chat.ggmlv3.q4_K_M.bin"  # 2.87G
@@ -50,7 +52,7 @@ hide_streamlit_style = """
 st.markdown(hide_streamlit_style, unsafe_allow_html=True) 
 
 # Set the title for the Streamlit app
-st.title("Coloring Anime ChatBot")
+st.title("Zendo美女チャットボックス")
 
 csv_url = "https://huggingface.co/spaces/uyen13/chatgirl2/raw/main/testchatdata.csv"
 # csv_url="https://docs.google.com/uc?export=download&id=1fQ2v2n9zQcoi6JoOU3lCBDHRt3a1PmaE"
@@ -83,10 +85,21 @@ llm = load_llm()
 
 # Create a conversational chain
 chain = ConversationalRetrievalChain.from_llm(llm=llm, retriever=db.as_retriever())
+# Initialize spaCy with the Japanese model
+# nlp = spacy.load("ja_core_news_sm")
 
 # Function for conversational chat
 def conversational_chat(query):
+    query = "提供されたデータに基づいて,"+query
     result = chain({"question": query, "chat_history": st.session_state['history']})
+    i = 0
+    while i < len(result["answer"]):
+        character = input_string[i]
+        if is_japanese_character(character):
+            break
+        else:
+            result = chain({"question": query, "chat_history": st.session_state['history']})
+        i += 1
     st.session_state['history'].append((query, result["answer"]))
     return result["answer"]
 
@@ -96,10 +109,9 @@ if 'history' not in st.session_state:
 
 # Initialize messages
 if 'generated' not in st.session_state:
-    st.session_state['generated'] = ["Hello ! Ask me about this page like coloring book,how to buy ... 🤗"]
-
+    st.session_state['generated'] = ["こんにちは！zendo美女です。何かお探しですか？... 🤗"]
 if 'past' not in st.session_state:
-    st.session_state['past'] = ["your chat here"]
+    st.session_state['past'] = ["チャットはここから"]
 
 # Create containers for chat history and user input
 response_container = st.container()
@@ -108,7 +120,7 @@ container = st.container()
 # User input form
 with container:
     with st.form(key='my_form', clear_on_submit=True):
-        user_input = st.text_input("ChatBox", placeholder="Ask anything...  ", key='input')
+        user_input = st.text_input("ChatBox", placeholder="質問をご記入ください...  ", key='input')
         submit_button = st.form_submit_button(label='Send')
 
     if submit_button and user_input:
@@ -121,4 +133,5 @@ if st.session_state['generated']:
     with response_container:
         for i in range(len(st.session_state['generated'])):
             message(st.session_state["past"][i], is_user=True, key=str(i) + '_user', avatar_style="big-smile")
-            message(st.session_state["generated"][i], key=str(i), avatar_style="thumbs")
+            message(st.session_state["generated"][i], key=str(i), avatar_style="thumbs")
+