arman77mxx commited on
Commit
be2b280
·
verified ·
1 Parent(s): b06cf9e

se cambio a multiples archivos

Browse files
Files changed (1) hide show
  1. app.py +10 -9
app.py CHANGED
@@ -11,14 +11,15 @@ from langchain.chains.question_answering import load_qa_chain
11
  st.set_page_config('preguntaDOC')
12
  st.header("Pregunta a tu PDF")
13
  OPENAI_API_KEY = st.text_input('OpenAI API Key', type='password')
14
- pdf_obj = st.file_uploader("Carga tu documento", type="pdf", on_change=st.cache_resource.clear)
15
 
16
  @st.cache_resource
17
- def create_embeddings(pdf):
18
- pdf_reader = PdfReader(pdf)
19
  text = ""
20
- for page in pdf_reader.pages:
21
- text += page.extract_text()
 
 
22
 
23
  text_splitter = RecursiveCharacterTextSplitter(
24
  chunk_size=800,
@@ -32,9 +33,9 @@ def create_embeddings(pdf):
32
 
33
  return knowledge_base
34
 
35
- if pdf_obj:
36
- knowledge_base = create_embeddings(pdf_obj)
37
- user_question = st.text_input("Haz una pregunta sobre tu PDF:")
38
 
39
  if user_question:
40
  os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
@@ -43,4 +44,4 @@ if pdf_obj:
43
  chain = load_qa_chain(llm, chain_type="stuff")
44
  respuesta = chain.run(input_documents=docs, question=user_question)
45
 
46
- st.write(respuesta)
 
11
  st.set_page_config('preguntaDOC')
12
  st.header("Pregunta a tu PDF")
13
  OPENAI_API_KEY = st.text_input('OpenAI API Key', type='password')
14
+ pdf_files = st.file_uploader("Carga tus documentos", type="pdf", accept_multiple_files=True, on_change=st.cache_resource.clear)
15
 
16
  @st.cache_resource
17
+ def create_embeddings(pdfs):
 
18
  text = ""
19
+ for pdf in pdfs:
20
+ pdf_reader = PdfReader(pdf)
21
+ for page in pdf_reader.pages:
22
+ text += page.extract_text()
23
 
24
  text_splitter = RecursiveCharacterTextSplitter(
25
  chunk_size=800,
 
33
 
34
  return knowledge_base
35
 
36
+ if pdf_files:
37
+ knowledge_base = create_embeddings(pdf_files)
38
+ user_question = st.text_input("Haz una pregunta sobre tus PDFs:")
39
 
40
  if user_question:
41
  os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
 
44
  chain = load_qa_chain(llm, chain_type="stuff")
45
  respuesta = chain.run(input_documents=docs, question=user_question)
46
 
47
+ st.write(respuesta)