Update app.py

app.py

@@ -10,7 +10,7 @@ def scrape(instring):
     </div>''')
     return gr.HTML.update(f'''{html_src}''')
 
-def scrape00(instring):
+def scrape00(instring, page_num):
     response = requests.get(instring, stream=True)
 
     if response.status_code == 200:
@@ -24,7 +24,7 @@ def scrape00(instring):
     #print (out)
     reader = PdfReader("data.pdf")
     number_of_pages = len(reader.pages)
-    page = reader.pages[0]
+    page = reader.pages[int(page_num-1)]
     text = page.extract_text()
     try:
         summarizer = gr.Interface.load("huggingface/facebook/bart-large-cnn")
@@ -35,7 +35,9 @@ def scrape00(instring):
 
 with gr.Blocks() as app:
     gr.Markdown('''<h1>PDF Viewer''')
-    inp=gr.Textbox()
+    with gr.Row():
+        inp=gr.Textbox(label="PDF URL",scale=3)
+        pg_num=gr.Number(label="Page Number",value=1,precision=0,scale=1)
     with gr.Row():
         go_btn = gr.Button("Load PDF")
         sum_btn = gr.Button("Summarize")
@@ -44,5 +46,5 @@ with gr.Blocks() as app:
         text_out = gr.Textbox()
         sum_out = gr.Textbox()
     go_btn.click(scrape,inp,outp)
-    sum_btn.click(scrape00,inp,[text_out,sum_out])
+    sum_btn.click(scrape00,[inp,pg_num],[text_out,sum_out])
 app.queue(concurrency_count=10).launch()