Update app.py

app.py

@@ -25,7 +25,8 @@ client = chromadb.Client()
 collection = client.create_collection("test_db") 
 
 
-def get_context(query_text, collection):
+def get_context(query_text):
+    
     query_emb = st_model.encode(query_text)
     query_response = collection.query(query_embeddings=query_emb.tolist(), n_results=4)
     context = query_response['documents'][0][0]
@@ -45,38 +46,13 @@ def local_query(query, context):
  
     return tokenizer.batch_decode(outputs, skip_special_tokens=True)
 
-def generate_langchain(btn):
 
-    file_name = btn.name 
    
-    loader = PDFMinerLoader(file_name)
-    doc = loader.load()
-
-    text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
-    texts = text_splitter.split_documents(doc)
-
-    texts = [i.page_content for i in texts]
-
-    doc_emb = st_model.encode(texts)
-    doc_emb = doc_emb.tolist()
-
-    ids = [str(uuid.uuid1()) for _ in doc_emb]
-
-    client = chromadb.Client()
-    collection = client.create_collection("test_db") 
-    
-    collection.add(
-        embeddings=doc_emb,
-        documents=texts,
-        ids=ids
-    )
-
-    return collection
+   
 
 def run_query(btn, history, query):
 
-    # context = get_context(query, collection)
-    context  = 'My name is damla'
+    context = get_context(query)
     
     print('calling local query')
     result = local_query(query, context)