Update app.py

app.py

@@ -4,18 +4,55 @@ import streamlit as st
 import base64
 from annotations import analyze_pdf
 
-def display_pdf(pdf_bytes):
-    """Displays the PDF in the browser using an embed tag."""
+def display_pdf_iframe(pdf_bytes):
+    """Displays the PDF using an iframe tag."""
     if pdf_bytes and len(pdf_bytes) > 0:
-        # Encode the PDF to base64
         base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
         pdf_display = f"""
-            <embed src="data:application/pdf;base64,{base64_pdf}" width="100%" height="800px" type="application/pdf">
+            <iframe src="data:application/pdf;base64,{base64_pdf}" width="100%" height="800px" type="application/pdf"></iframe>
         """
         st.components.v1.html(pdf_display, height=800, width=700, scrolling=True)
     else:
         st.info("No annotated PDF to display.")
 
+def display_pdf_object(pdf_bytes):
+    """Displays the PDF using an object tag."""
+    if pdf_bytes and len(pdf_bytes) > 0:
+        base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
+        pdf_display = f"""
+            <object data="data:application/pdf;base64,{base64_pdf}" type="application/pdf" width="100%" height="800px">
+                <p>Your browser does not support PDFs.
+                <a href="data:application/pdf;base64,{base64_pdf}">Download the PDF</a>.</p>
+            </object>
+        """
+        st.components.v1.html(pdf_display, height=800, width=700, scrolling=True)
+    else:
+        st.info("No annotated PDF to display.")
+
+def display_pdf_pdfjs(pdf_bytes):
+    """Displays the PDF using PDF.js embedded in an iframe."""
+    if pdf_bytes and len(pdf_bytes) > 0:
+        base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
+        pdfjs_viewer_url = f"https://mozilla.github.io/pdf.js/web/viewer.html?file=data:application/pdf;base64,{base64_pdf}"
+        pdf_display = f"""
+            <iframe src="{pdfjs_viewer_url}" width="100%" height="800px"></iframe>
+        """
+        st.components.v1.html(pdf_display, height=800, width=700, scrolling=True)
+    else:
+        st.info("No annotated PDF to display.")
+
+def display_pdf_new_tab(pdf_bytes):
+    """Provides a link to view the PDF in a new browser tab."""
+    if pdf_bytes and len(pdf_bytes) > 0:
+        base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
+        href = f"data:application/pdf;base64,{base64_pdf}"
+        pdf_display = f"""
+            <a href="{href}" target="_blank">🔍 View Annotated PDF</a>
+        """
+        st.markdown(pdf_display, unsafe_allow_html=True)
+    else:
+        st.info("No annotated PDF to display.")
+
 def main():
     st.set_page_config(
         page_title="PDF Analyzer",
@@ -32,6 +69,8 @@ def main():
 
     if uploaded_file is not None:
         with st.spinner("Analyzing PDF..."):
+            # Reset file pointer before reading
+            uploaded_file.seek(0)
             language_results, annotated_pdf = analyze_pdf(uploaded_file)
 
         if "error" in language_results:
@@ -46,9 +85,25 @@ def main():
             else:
                 st.write("Annotated PDF is empty.")
 
-            # Display the annotated PDF
+            # Select embedding method
+            embedding_option = st.selectbox(
+                "Select PDF Display Method",
+                options=["Iframe", "Object Tag", "PDF.js", "Open in New Tab"],
+                index=0
+            )
+
             st.subheader("📄 Annotated PDF")
-            display_pdf(annotated_pdf)
+
+            if embedding_option == "Iframe":
+                display_pdf_iframe(annotated_pdf)
+            elif embedding_option == "Object Tag":
+                display_pdf_object(annotated_pdf)
+            elif embedding_option == "PDF.js":
+                display_pdf_pdfjs(annotated_pdf)
+            elif embedding_option == "Open in New Tab":
+                display_pdf_new_tab(annotated_pdf)
+            else:
+                st.info("Select a display method to view the PDF.")
 
             # Sidebar for error details
             st.sidebar.header("📝 Error Details")