File size: 1,506 Bytes
3075f85 |
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 |
# imports
import gradio as gr
from transformers import pipeline
import torch
import PyPDF2
# function to read the uploaded PDF and extract its PDF when present based on the keyword "abstract" search.
# If the PDF doesn't have the word "abstract" it won't work
# also, I'm trying to limitate to the abstract itself, not to other sections, by adding a pattern of in between headers
def process_pdf(pdf):
with open(pdf.name, "rb") as f:
reader = PyPDF2.PdfReader(f)
text = ""
for page in reader.pages:
text += page.extract_text()
abstract_start = text.lower().find("abstract:")
if abstract_start != -1:
abstract_end = text.lower().find("\n\n", abstract_start)
if abstract_end != -1:
abstract = text[abstract_start:abstract_end]
else:
abstract = text[abstract_start:]
else:
abstract = "Abstract not found."
return abstract
#Now creating the interface to read the PDFs
interface = gr.Interface(fn=process_pdf,
inputs=gr.inputs.File(type="file", label="Upload PDF"),
outputs="text",
title="Summarizing outloud",
description="Extract abstracts from PDFs, summarize then in 1 sentence and get an audio of it",
examples=[["example_pdf1.pdf"], ["example_pdf2.pdf"]])
if __name__ == "__main__":
interface.launch() |