Update app.py

app.py

@@ -1,3 +1,4 @@
+# to-do: Enable downloading multiple patent PDFs via corresponding links 
 import sys
 import os
 import re
@@ -7,7 +8,6 @@ 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,15 +129,6 @@ 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",
@@ -167,7 +158,6 @@ if __name__ == "__main__":
 
         st.write(f"Patent number: **{patent_number}**")
 
-        # Define PDF path in temp directory
         pdf_path = os.path.join(tempfile.gettempdir(), f"{patent_number}.pdf")
         if os.path.isfile(pdf_path):
             st.write("✅ File already downloaded.")
@@ -176,21 +166,6 @@ if __name__ == "__main__":
             pdf_path = download_pdf(patent_number)
             st.write(f"✅ File downloaded: {pdf_path}")
 
-        # Display a preview of the downloaded PDF
-        st.write("📄 Preview of the downloaded patent PDF:")
-        if os.path.isfile(pdf_path):
-            with open(pdf_path, "rb") as pdf_file:
-                st.download_button(
-                    label="Download PDF",
-                    data=pdf_file,
-                    file_name=f"{patent_number}.pdf",
-                    mime="application/pdf"
-                )
-            # Embed PDF content using base64
-            st.write("📋 PDF Content:")
-            pdf_view = embed_pdf(pdf_path)
-            st.components.v1.html(pdf_view, height=1000)
-
         st.write("🔄 Loading document into the system...")
 
         # Persist the chain in session state to prevent reloading
@@ -210,15 +185,12 @@ if __name__ == "__main__":
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
 
-    # Chat Input Section
     if "chain" in st.session_state:
         if user_input := st.chat_input("What is your question?"):
-            # Append user message
             st.session_state.messages.append({"role": "user", "content": user_input})
             with st.chat_message("user"):
                 st.markdown(user_input)
 
-            # Generate assistant response
             with st.chat_message("assistant"):
                 message_placeholder = st.empty()
                 full_response = ""
@@ -230,7 +202,6 @@ if __name__ == "__main__":
                     except Exception as e:
                         full_response = f"An error occurred: {e}"
 
-                # Display assistant response
                 message_placeholder.markdown(full_response)
                 st.session_state.messages.append({"role": "assistant", "content": full_response})
     else: