Update app.py

app.py

@@ -175,19 +175,26 @@ def app2():
                     #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)
                     
-                    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')
+
                     
             
             except Exception as e:
                   print("Error occured: {}".format(e))
                   raise e      
 
+    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)
+
+    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')
 
 
 app.run()