Spaces:
Sleeping
Sleeping
Chandranshu Jain
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,6 +9,7 @@ from langchain.chains.question_answering import load_qa_chain
|
|
| 9 |
from langchain.prompts import PromptTemplate
|
| 10 |
from langchain_community.document_loaders import PyPDFLoader
|
| 11 |
from langchain_chroma import Chroma
|
|
|
|
| 12 |
|
| 13 |
st.set_page_config(page_title="Document Genie", layout="wide")
|
| 14 |
|
|
@@ -26,8 +27,20 @@ Follow these simple steps to interact with the chatbot:
|
|
| 26 |
2. **Ask a Question**: After processing the documents, ask any question related to the content of your uploaded documents for a precise answer.
|
| 27 |
""")
|
| 28 |
|
| 29 |
-
def get_pdf(pdf_docs):
|
| 30 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
docs = loader.load()
|
| 32 |
return docs
|
| 33 |
|
|
|
|
| 9 |
from langchain.prompts import PromptTemplate
|
| 10 |
from langchain_community.document_loaders import PyPDFLoader
|
| 11 |
from langchain_chroma import Chroma
|
| 12 |
+
import io
|
| 13 |
|
| 14 |
st.set_page_config(page_title="Document Genie", layout="wide")
|
| 15 |
|
|
|
|
| 27 |
2. **Ask a Question**: After processing the documents, ask any question related to the content of your uploaded documents for a precise answer.
|
| 28 |
""")
|
| 29 |
|
| 30 |
+
#def get_pdf(pdf_docs):
|
| 31 |
+
# loader = PyPDFLoader(pdf_docs)
|
| 32 |
+
# docs = loader.load()
|
| 33 |
+
# return docs
|
| 34 |
+
|
| 35 |
+
def get_pdf(pdf_file):
|
| 36 |
+
# Extract file path from UploadedFile object
|
| 37 |
+
file_path = pdf_file.name
|
| 38 |
+
# Read the content of the uploaded PDF file
|
| 39 |
+
pdf_content = pdf_file.read()
|
| 40 |
+
# Close the file object
|
| 41 |
+
pdf_file.close()
|
| 42 |
+
# Load the PDF content using PyPDFLoader
|
| 43 |
+
loader = PyPDFLoader(io.BytesIO(pdf_content))
|
| 44 |
docs = loader.load()
|
| 45 |
return docs
|
| 46 |
|