pdf-reader / app.py
Omnibus's picture
Update app.py
4ffb3fe
raw
history blame
1.09 kB
import gradio as gr
import requests
from pypdf import PdfReader
def scrape(instring):
html_src=(f'''
<div style="text-align:center">
<h4>PDF Viewer</h4>
<iframe src="https://docs.google.com/viewer?url={instring}&embedded=true" frameborder="0" height="1200px" width="100%"></iframe>
</div>''')
return gr.HTML.update(f'''{html_src}''')
def scrape00(instring):
response = requests.get(instring, stream=True)
if response.status_code == 200:
with open("data.pdf", "wb") as f:
f.write(response.content)
else:
print(response.status_code)
out = Path("./data.pdf")
print (out)
reader = PdfReader("data.pdf")
number_of_pages = len(reader.pages)
page = reader.pages[0]
text = page.extract_text()
return gr.HTML.update(f'''<embed src={out} type="application/pdf" width="100%" height="500px" />''')
with gr.Blocks() as app:
gr.Markdown('''<h1>PDF Viewer''')
inp=gr.Textbox()
go_btn = gr.Button()
outp = gr.HTML()
go_btn.click(scrape,inp,outp)
app.queue(concurrency_count=10).launch()