Update app.py

app.py

@@ -98,6 +98,7 @@ def app2():
     hy.markdown(" ")
     
     hy.subheader("NER from Pdf Reports")
+    counter = 0
     with hy.form(key="pdf_form"):     
         ce, c1, ce, c2, c3 = hy.columns([0.07, 1, 0.07, 4, 1.5])
         with c1:
@@ -152,37 +153,21 @@ def app2():
                     print("No Entities Extracted!!!")
                 
                 temp_dir = tempfile.TemporaryDirectory()
-                document.save(tempfile.gettempdir()+"/annott.pdf")          
-            
-
-                with c2:
-                    st.markdown("""
-<embed src="https://drive.google.com/viewerng/
-viewer?embedded=true&url=https://www.w3.org/WAI/ER/tests/xhtml/testfiles/resources/pdf/dummy.pdf" width="800" height="800">
-""", unsafe_allow_html=True)
-         
-
-
-                    with open((tempfile.gettempdir()+"/annott.pdf"),"rb") as f:
-                        base64_pdf = base64.b64encode(f.read()).decode('utf-8')
-                    #pdf_display = f'<embed src="data:application/pdf;base64,{base64_pdf}" width="800" height="800" type="application/pdf">'
-                    pdf_display = f'<iframe src="data:application/pdf;base64,{base64_pdf}" width="800" height="800" type="application/pdf"></iframe>'
-                    st.markdown(pdf_display, unsafe_allow_html=True)
-                    
-
+                document.save(tempfile.gettempdir()+"/annott.pdf")    
+                counter+=1      
                     
-            
             except Exception as e:
                   print("Error occured: {}".format(e))
                   raise e      
-
-    with open((tempfile.gettempdir()+"/annott.pdf"), "rb") as pdf_file:
-        PDFbyte = pdf_file.read()
-        
-    hy.download_button(label="Download Annotated PDF", 
-            data=PDFbyte,
-            file_name="annotated.pdf",
-            mime='application/octet-stream')
+    
+    if counter !=0:
+        with open((tempfile.gettempdir()+"/annott.pdf"), "rb") as pdf_file:
+            PDFbyte = pdf_file.read()
+            
+        hy.download_button(label="Download Annotated PDF", 
+                data=PDFbyte,
+                file_name="annotated.pdf",
+                mime='application/octet-stream')
     
     #with open((tempfile.gettempdir()+"/annott.pdf"),"rb") as f:
     #    base64_pdf = base64.b64encode(f.read()).decode('utf-8')