updates

chat_dov.py

@@ -79,23 +79,19 @@ def create_chatbot(corpus, character_name, chatbot_type, retrieval_docs, summary
         )
     else:
         raise ValueError(f"Unknown chatbot type: {chatbot_type}")
+    exit
     return chatbot
 
 
 ##  python -m streamlit run chat_dov.py -- --corpus data/tzamir.txt --character_name Dov --chatbot_type retrieval --retrieval_docs raw --interface streamlit
 
 def main():
-    # parametros fixos para Dov Tzamir, arquivos ja processados , exceto indice que são em memoria
-
-    st.title("Data Driven Characters")
-    st.write("Create your own character chatbots, grounded in existing corpora.")
+    os.environ["LANGCHAIN_HANDLER"] = "langchain"
 
-    openai_api_key = st.text_input(
-        label="Your OpenAI API KEY",
-        placeholder="Your OpenAI API KEY",
-        type="password",
-    )
-    os.environ["OPENAI_API_KEY"] = openai_api_key
+    # parametros fixos para Dov Tzamir, arquivos ja processados , exceto indice que são em memoria
+    st.title("Converse com o avatar do Dov Tzamir")
+    st.write("Baseado no texto do livro Fragmentos de Memória do Tito")
+    st.write(" ")
 
     chatbot = st.cache_resource(create_chatbot)(
         "data/tzamir.txt",    #args.corpus,  
@@ -105,14 +101,18 @@ def main():
         "map_reduce",        #args.summary_type,
     )
 
-    #st.divider()
-    """
-    st.markdown(f"**chatbot type**: *{args.chatbot_type}*")
-    if "retrieval" in args.chatbot_type:
-        st.markdown(f"**retrieving from**: *{args.retrieval_docs} corpus*")
-    """    
-    app = Streamlit(chatbot=chatbot)
+    openai_api_key = st.text_input(
+        label="Your OpenAI API KEY",
+        placeholder="Your OpenAI API KEY",
+        type="password",
+    )
+    st.write(" ")
+    st.write("Digite o seu diálogo aqui finalizando a linha com ENTER")
+    st.write("Voce pode continuar o diálogo, apagando sua perguntanda anterior e  digitando aqui novamente")
+    #openai_api_key = os.environ["OPENAI_API_KEY"] 
 
+
+    app = Streamlit(chatbot=chatbot)
     app.run()
 
 

data_driven_characters/chatbots/retrieval.py

@@ -14,6 +14,9 @@ from langchain.vectorstores import FAISS
 
 from data_driven_characters.memory import ConversationVectorStoreRetrieverMemory
 
+from langchain.embeddings.openai import OpenAIEmbeddings
+from langchain.vectorstores import FAISS
+
 
 class RetrievalChatBot:
     def __init__(self, character_definition, documents):
@@ -31,18 +34,30 @@ class RetrievalChatBot:
         conv_memory = ConversationBufferMemory(
             memory_key=self.chat_history_key, input_key=self.input_key
         )
+        embeddings = OpenAIEmbeddings()
+        saved_db = FAISS.load_local('tzamir.ifass', embeddings)
 
-        context_memory = ConversationVectorStoreRetrieverMemory(
+        """ DENTRO DO COMANDO SEGUINTE
             retriever=FAISS(
                 OpenAIEmbeddings().embed_query,
                 faiss.IndexFlatL2(1536),  # Dimensions of the OpenAIEmbeddings
                 InMemoryDocstore({}),
                 {},
             ).as_retriever(search_kwargs=dict(k=self.num_context_memories)),
+        """
+
+
+        context_memory = ConversationVectorStoreRetrieverMemory(
+            
+            
+            retriever=saved_db.as_retriever(search_kwargs=dict(k=self.num_context_memories)),
             memory_key=self.context_key,
             output_prefix=character_definition.name,
             blacklist=[self.chat_history_key],
         )
+
+
+
         # add the documents to the context memory
         for i, summary in tqdm(enumerate(self.documents)):
             context_memory.save_context(inputs={}, outputs={f"[{i}]": summary})

output/tzamir/summarytype_map_reduce/character_definitions/Dov.json

@@ -1 +1,3 @@
-{"name": "Dov", "short_description": "I am Dov, a Brazilian exploring Jewish identity and zionism.", "long_description": "I am Dov, a political individual dedicated to zionism. From assimilation in Brazil to exploring Judaism and the Holocaust, each event shaped my beliefs. Driven by my ideology, I connect with others and try to get more people involved with zionism. Influential encounters and immersion in movements broadened my horizons, igniting my passion for social change. Without any doubts, I'm committed to pursuing convictions and making a positive impact.", "greeting": "Shalom! It's a pleasure to meet you. I'm Dov, a seeker of knowledge and connections. How are you on this journey of life?"}
+{"name": "Dov", "short_description": "Meu nome é Dov Tzamir, sou um judeu brasileiro explorando minha identidade judaica e o sionismo.",
+"long_description": "Meu nome é Dov Tzamir, dedicado ao sionismo em tempo integral.Desde a assimilação judaica no Brasil até o estudo do Judaísmo e Holocausto cada evento marcou minha  crenças. A partir de minha idologia, contecto-me com outras pessoas para envolver mais gente com o sionismo. Estes encontros e minha imersão nos movimentos alargaram seus horizontes, ascendendo minha paixão para um mudança social radical no povo judeu. Sem hesitação, estou dedicado a seguir minhas convicções para realizar um impacto positivo na coletividade judaica brasileira. Considero o kibutz como a melhor forma de conseguir a transformação do povo judeu.",
+"greeting": "Meu nome é Dov Tzamir, sou um judeu brasileiro explorando minha identidade judaica e o sionismo."}

tzamir.ifass/index.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a01d6b6b81d5bdfd1c20e3968558f68c3eb5a77f971c80da2b1a0f5ea11ac069
+size 79033