Update app.py

app.py

@@ -45,31 +45,29 @@ def local_query(query, context):
 def run_query(btn, history, query):
 
 
+    file_name = btn.name 
    
-      
-    # file_name = btn.name 
-   
-    # loader = PDFMinerLoader(file_name)
-    # doc = loader.load()
+    loader = PDFMinerLoader(file_name)
+    doc = loader.load()
 
-    # text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
-    # texts = text_splitter.split_documents(doc)
+    text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
+    texts = text_splitter.split_documents(doc)
 
-    # texts = [i.page_content for i in texts]
+    texts = [i.page_content for i in texts]
 
-    # doc_emb = st_model.encode(texts)
-    # doc_emb = doc_emb.tolist()
+    doc_emb = st_model.encode(texts)
+    doc_emb = doc_emb.tolist()
 
-    # ids = [str(uuid.uuid1()) for _ in doc_emb]
+    ids = [str(uuid.uuid1()) for _ in doc_emb]
 
-    # client = chromadb.Client()
-    # collection = client.create_collection("test_db") 
+    client = chromadb.Client()
+    collection = client.create_collection("test_db") 
     
-    # collection.add(
-    #     embeddings=doc_emb,
-    #     documents=texts,
-    #     ids=ids
-    # )
+    collection.add(
+        embeddings=doc_emb,
+        documents=texts,
+        ids=ids
+    )