Spaces:
Sleeping
Sleeping
File size: 1,505 Bytes
7b480a8 2189cdb 7b480a8 2189cdb 7b480a8 2189cdb 7b480a8 2189cdb 7b480a8 2189cdb 7b480a8 2189cdb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
import streamlit as st
import fitz # PyMuPDF
from transformers import pipeline
import glob
# Function to extract text from PDFs
def extract_text_from_pdfs(pdf_files):
pdf_texts = {}
for pdf_file in pdf_files:
with fitz.open(pdf_file) as doc:
text = ""
for page in doc:
text += page.get_text()
pdf_texts[pdf_file] = text
return pdf_texts
# Load pre-trained QA model
qa_pipeline = pipeline('question-answering', model='distilbert-base-uncased-distilled-squad')
# Function to answer questions based on extracted text
def answer_question(pdf_texts, question):
context = " ".join(pdf_texts.values())
result = qa_pipeline(question=question, context=context)
return result['answer']
# Streamlit application
st.title("PDF Question Answering App")
# File uploader for PDF files
uploaded_files = st.file_uploader("Upload PDF files", type="pdf", accept_multiple_files=True)
# Display uploaded files
if uploaded_files:
# Extract text from PDFs
pdf_texts = extract_text_from_pdfs([file.name for file in uploaded_files])
st.write("PDFs Uploaded Successfully!")
# Question input
question = st.text_input("Enter your question:")
if st.button("Get Answer"):
if question:
answer = answer_question(pdf_texts, question)
st.write(f"Answer: {answer}")
else:
st.write("Please enter a question.")
else:
st.write("Please upload PDF files to continue.")
|