chatbytes commited on
Commit
e6e4b49
·
verified ·
1 Parent(s): 25b948d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -9
app.py CHANGED
@@ -22,6 +22,11 @@ def text_splitter_function(text):
22
  texts = text_splitter.split_text(text)
23
  return texts;
24
 
 
 
 
 
 
25
  def text_extract(file):
26
  pdf_reader = PyPDF2.PdfReader(file.name)
27
  # Get the number of pages
@@ -32,16 +37,16 @@ def text_extract(file):
32
  page = pdf_reader.pages[page_num]
33
  text += page.extract_text()
34
  text_splitter=text_splitter_function(text);
35
- db = FAISS.from_texts(text_splitter, embeddings);
36
- retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 2})
37
- llm=GooglePalm(google_api_key=google_api)
38
- qa = RetrievalQA.from_chain_type(
39
- llm=llm, chain_type="stuff", retriever=retriever, return_source_documents=True
40
- )
41
- result=qa.invoke("where is tajmahal")
42
 
43
- return 'hi'
44
-
 
 
 
 
 
 
 
45
 
46
 
47
 
 
22
  texts = text_splitter.split_text(text)
23
  return texts;
24
 
25
+
26
+ def helper(text_splitter):
27
+ db = FAISS.from_texts(text_splitter, embeddings);
28
+ return 'hi';
29
+
30
  def text_extract(file):
31
  pdf_reader = PyPDF2.PdfReader(file.name)
32
  # Get the number of pages
 
37
  page = pdf_reader.pages[page_num]
38
  text += page.extract_text()
39
  text_splitter=text_splitter_function(text);
 
 
 
 
 
 
 
40
 
41
+ result=helper(text_splitter);
42
+ return result
43
+ # db = FAISS.from_texts(text_splitter, embeddings);
44
+ # retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 2})
45
+ # llm=GooglePalm(google_api_key=google_api)
46
+ # qa = RetrievalQA.from_chain_type(
47
+ # llm=llm, chain_type="stuff", retriever=retriever, return_source_documents=True
48
+ # )
49
+ # result=qa.invoke("where is tajmahal")
50
 
51
 
52