Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import re
 import shutil
 import time
+import fitz 
 import streamlit as st
 import nltk
 import tempfile
@@ -129,6 +130,19 @@ def download_pdf(patent_number):
         st.error(f"Failed to download patent PDF: {e}")
         st.stop()
 
+def preview_pdf(pdf_path):
+    """Generate and display the first page of the PDF as an image."""
+    try:
+        doc = fitz.open(pdf_path)  # Open PDF
+        first_page = doc[0]  # Extract the first page
+        pix = first_page.get_pixmap()  # Render page to a Pixmap (image)
+        temp_image_path = os.path.join(tempfile.gettempdir(), "pdf_preview.png")
+        pix.save(temp_image_path)  # Save the image temporarily
+        return temp_image_path
+    except Exception as e:
+        st.error(f"Error generating PDF preview: {e}")
+        return None
+
 if __name__ == "__main__":
     st.set_page_config(
         page_title="Patent Chat: Google Patents Chat Demo",
@@ -151,6 +165,7 @@ if __name__ == "__main__":
             st.warning("Please enter a Google patent link to proceed.")
             st.stop()
 
+        # Extract patent number
         patent_number = extract_patent_number(patent_link)
         if not patent_number:
             st.error("Invalid patent link format. Please provide a valid Google patent link.")
@@ -158,6 +173,7 @@ if __name__ == "__main__":
 
         st.write(f"Patent number: **{patent_number}**")
 
+        # File download handling
         pdf_path = os.path.join(tempfile.gettempdir(), f"{patent_number}.pdf")
         if os.path.isfile(pdf_path):
             st.write("✅ File already downloaded.")
@@ -166,6 +182,16 @@ if __name__ == "__main__":
             pdf_path = download_pdf(patent_number)
             st.write(f"✅ File downloaded: {pdf_path}")
 
+        # Generate and display PDF preview
+        st.write("🖼️ Generating PDF preview...")
+        preview_image_path = preview_pdf(pdf_path)
+
+        if preview_image_path:
+            st.image(preview_image_path, caption="First Page Preview", use_column_width=True)
+        else:
+            st.warning("Failed to generate a preview for this PDF.")
+
+        # Load the document into the system
         st.write("🔄 Loading document into the system...")
 
         # Persist the chain in session state to prevent reloading
@@ -185,6 +211,7 @@ if __name__ == "__main__":
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
 
+    # User input and chatbot response
     if "chain" in st.session_state:
         if user_input := st.chat_input("What is your question?"):
             st.session_state.messages.append({"role": "user", "content": user_input})