Update app.py

app.py

@@ -63,12 +63,12 @@ ocr_id = {
     "Zulu": "zu",
 }
 
-def pdf_pil(file_path,page_num):
+def pdf_pil(file_path,page_num,up_scale):
 
     pdf = pdfium.PdfDocument("data.pdf")
     page = pdf.get_page(int(page_num)-1)
     bitmap = page.render(
-        scale = 3,    # 72dpi resolution
+        scale = int(up_scale),    # 72dpi resolution
         rotation = 0, # no additional rotation
         # ... further rendering options
     )
@@ -77,8 +77,8 @@ def pdf_pil(file_path,page_num):
     
     return (f"image_{page_num}.png")
 
-def ocrpdf(file_path,pdf_lang,page_num,sent_wid,contrast_det):
-    img1 = pdf_pil(file_path,page_num)
+def ocrpdf(file_path,pdf_lang,page_num,sent_wid,contrast_det,up_scale):
+    img1 = pdf_pil(file_path,page_num,up_scale)
     lang=[f"{ocr_id[pdf_lang]}"]
     reader = easyocr.Reader(lang)
     bounds = reader.readtext(img1,width_ths=sent_wid,contrast_ths=contrast_det)
@@ -97,7 +97,9 @@ def scrape(instring):
     </div>''')
     return gr.HTML.update(f'''{html_src}''')
 
-def scrape00(instring, page_num,pdf_lang,sent_wid,contrast_det):
+def scrape00(instring, page_num,pdf_lang,sent_wid,contrast_det,up_scale):
+    yield (None,None,gr.Markdown.update("""<h3> Trying Native Text Detection"""))
+
     response = requests.get(instring, stream=True)
 
     if response.status_code == 200:
@@ -114,16 +116,19 @@ def scrape00(instring, page_num,pdf_lang,sent_wid,contrast_det):
     page = reader.pages[int(page_num)-1]
     text = page.extract_text()
     print (text)
+    summarizer = gr.Interface.load("huggingface/facebook/bart-large-cnn")
     try:
-        summarizer = gr.Interface.load("huggingface/facebook/bart-large-cnn")
         sum_out = summarizer(text)
     except Exception:
         try:
-            text = ocrpdf("data.pdf",pdf_lang,page_num,sent_wid,contrast_det)
+            yield (None,None,gr.Markdown.update("""<h3> Trying OCR Text Detection"""))
+            text = ocrpdf("data.pdf",pdf_lang,page_num,sent_wid,contrast_det,up_scale)
             sum_out = summarizer(text)
         except Exception:
             sum_out = "Error"
-    return text, sum_out
+            yield (None,None,gr.Markdown.update("""<h3> Error"""))
+            
+    return text, sum_out,gr.Markdown.update("""<h3> Complete"""))
 
 with gr.Blocks() as app:
     gr.Markdown('''<h1>PDF Viewer''')
@@ -139,6 +144,7 @@ with gr.Blocks() as app:
             with gr.Box():
                 sent_wid=gr.Slider(0.1, 3, step=0.1,value=1,label="Horizontal Word Space")
                 contrast_det=gr.Slider(0.1, 1, step=0.1,value=0.1,label="Contrast Threshold")
+                up_scale=gr.Slider(0.1, 5, step=0.1,value=1,label="PDF to Image Scale")
             with gr.Column():
                 target_lang = gr.Dropdown(label="PDF Language", choices=list(ocr_id.keys()),value="English")
                 sum_btn = gr.Button("Summarize")
@@ -146,5 +152,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,pg_num,target_lang,sent_wid,contrast_det],[text_out,sum_out])
+    sum_btn.click(scrape00,[inp,pg_num,target_lang,sent_wid,contrast_det,up_scale],[text_out,sum_out])
 app.queue(concurrency_count=10).launch()