Update app.py

app.py

@@ -3,15 +3,16 @@
 import streamlit as st
 import base64
 from annotations import analyze_pdf
-import tempfile
-import os
 
 def display_pdf(pdf_bytes):
-    """Displays the PDF in the browser using an iframe."""
+    """Displays the PDF in the browser using an embed tag."""
     if pdf_bytes:
+        # Encode the PDF to base64
         base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
-        pdf_display = f'<iframe src="data:application/pdf;base64,{base64_pdf}" width="100%" height="800px" type="application/pdf"></iframe>'
-        st.markdown(pdf_display, unsafe_allow_html=True)
+        pdf_display = f"""
+            <embed src="data:application/pdf;base64,{base64_pdf}" width="100%" height="800px" type="application/pdf">
+        """
+        st.components.v1.html(pdf_display, height=800, width=700, scrolling=True)
     else:
         st.info("No annotated PDF to display.")
 
@@ -58,9 +59,12 @@ def main():
 
             # Option to download the annotated PDF
             if annotated_pdf:
-                b64 = base64.b64encode(annotated_pdf).decode()
-                href = f'<a href="data:application/pdf;base64,{b64}" download="annotated.pdf">📥 Download Annotated PDF</a>'
-                st.markdown(href, unsafe_allow_html=True)
+                st.download_button(
+                    label="📥 Download Annotated PDF",
+                    data=annotated_pdf,
+                    file_name="annotated.pdf",
+                    mime="application/pdf",
+                )
 
 if __name__ == "__main__":
     main()