Update app.py

app.py

@@ -1,4 +1,3 @@
-# to-do: Enable downloading multiple patent PDFs via corresponding links 
 import sys
 import os
 import re
@@ -8,6 +7,7 @@ import streamlit as st
 import nltk
 import tempfile
 import subprocess
+import base64  # For embedding PDF content
 
 # Pin NLTK to version 3.9.1
 REQUIRED_NLTK_VERSION = "3.9.1"
@@ -129,6 +129,15 @@ def download_pdf(patent_number):
         st.error(f"Failed to download patent PDF: {e}")
         st.stop()
 
+def embed_pdf(file_path):
+    """Convert PDF file to base64 and embed it in an iframe."""
+    with open(file_path, "rb") as f:
+        base64_pdf = base64.b64encode(f.read()).decode("utf-8")
+    pdf_display = f"""
+        <iframe src="data:application/pdf;base64,{base64_pdf}" width="700" height="1000" style="border: none;"></iframe>
+    """
+    return pdf_display
+
 if __name__ == "__main__":
     st.set_page_config(
         page_title="Patent Chat: Google Patents Chat Demo",
@@ -177,12 +186,10 @@ if __name__ == "__main__":
                     file_name=f"{patent_number}.pdf",
                     mime="application/pdf"
                 )
-            # Embed PDF using iframe
+            # Embed PDF content using base64
             st.write("📋 PDF Content:")
-            pdf_display = f"""
-                <iframe src="file://{pdf_path}" width="700" height="1000" style="border: none;"></iframe>
-            """
-            st.components.v1.html(pdf_display, height=1000)
+            pdf_view = embed_pdf(pdf_path)
+            st.components.v1.html(pdf_view, height=1000)
 
         st.write("🔄 Loading document into the system...")