chatbytes commited on
Commit
9b3282f
·
verified ·
1 Parent(s): 30e878c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -32,6 +32,10 @@ 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)
@@ -39,14 +43,11 @@ def text_extract(file):
39
  qa = RetrievalQA.from_chain_type(
40
  llm=llm, chain_type="stuff", retriever=retriever, return_source_documents=True
41
  )
42
- result=qa.invoke("hi")
43
  except:
44
- result="hi"
45
-
46
 
47
- return result['result']
48
- def text_extract1(file):
49
- return 'hi';
50
 
51
 
52
 
@@ -62,7 +63,7 @@ with gr.Blocks() as demo:
62
  with gr.Column():
63
  input_file=gr.File(label="Upload PDF", file_count="single")
64
  submit_btn=gr.Button("Submit")
65
- submit_btn.click(text_extract1, [input_file], [user_input])
66
  #send_btn.click(chatbot_response,[user_input,state],[chatbot, state])
67
 
68
  if __name__ == "__main__":
 
32
  page = pdf_reader.pages[page_num]
33
  text += page.extract_text()
34
  text_splitter=text_splitter_function(text);
35
+ result=response_from_bot(text_splitter)
36
+ print(result)
37
+ return result
38
+ def response_from_bot(text_splitter):
39
  db = FAISS.from_texts(text_splitter, embeddings);
40
  retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 2})
41
  llm=GooglePalm(google_api_key=google_api)
 
43
  qa = RetrievalQA.from_chain_type(
44
  llm=llm, chain_type="stuff", retriever=retriever, return_source_documents=True
45
  )
46
+ result=qa.invoke("what should i eat on monday")
47
  except:
48
+ result="hir"
 
49
 
50
+ return result['result'];
 
 
51
 
52
 
53
 
 
63
  with gr.Column():
64
  input_file=gr.File(label="Upload PDF", file_count="single")
65
  submit_btn=gr.Button("Submit")
66
+ submit_btn.click(text_extract, [input_file], [user_input])
67
  #send_btn.click(chatbot_response,[user_input,state],[chatbot, state])
68
 
69
  if __name__ == "__main__":