Duas funções: uma cria o DB e outra Delete o DB

functions.py

@@ -0,0 +1,67 @@
+#!/usr/bin/env python3
+"""
+Data Scientist.: Dr. Eddy Giusepe Chirinos Isidro
+
+Execução deste script
+=====================
+$ python functions.py
+"""
+from langchain_community.document_loaders import PyPDFLoader
+from langchain_text_splitters import RecursiveCharacterTextSplitter
+from langchain_community.embeddings.sentence_transformer import (SentenceTransformerEmbeddings,
+                                                                )
+from langchain_chroma import Chroma
+
+import warnings
+warnings.filterwarnings('ignore')
+
+import shutil
+import os
+
+
+# Criando o Database:
+def create_db():
+
+    loader = PyPDFLoader("files/CHATGPT Granularity_clustering.pdf")
+    pages = loader.load()
+    #print(len(pages))
+
+    text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000,
+                                                   chunk_overlap=100,
+                                                   length_function=len,
+                                                   is_separator_regex=False
+                                                  )
+    
+    chunks = text_splitter.split_documents(pages)
+    #print(len(chunks))
+
+    ids = [str(i) for i in range(1, len(chunks) + 1)]
+    #print(ids)
+
+    # Crie a função de Embeddings de código aberto:
+    embedding_function = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2",
+                                                       model_kwargs = {'device': 'cpu'}
+                                                      )
+
+    # Crie o banco de dados Chroma com IDs:
+    Chroma.from_documents(documents=chunks,
+                          embedding=embedding_function,
+                          persist_directory="./chroma_db",
+                          ids=ids
+                         )              
+    
+
+# Excluindo o banco de dados (Database):
+def delete_persisted_db():
+    if "chroma_db" in os.listdir():
+        shutil.rmtree("chroma_db")
+        print(f"Banco de dados excluído e seu conteúdo.")
+    else:
+        raise FileNotFoundError("Banco de dados não encontrado.")
+    
+
+
+# Testando o funcionamento deste script:
+if __name__ == "__main__":
+    create_db()
+