Update app.py

app.py

@@ -6,6 +6,7 @@ from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 from llama_index.core import Settings
 import os
 import base64
+import fitz  # PyMuPDF
 
 # Load environment variables
 load_dotenv()
@@ -31,11 +32,20 @@ DATA_DIR = "data"
 os.makedirs(DATA_DIR, exist_ok=True)
 os.makedirs(PERSIST_DIR, exist_ok=True)
 
+
 def displayPDF(file):
-    with open(file, "rb") as f:
-        base64_pdf = base64.b64encode(f.read()).decode('utf-8')
-    pdf_display = f'<iframe src="data:application/pdf;base64,{base64_pdf}" width="100%" height="600" type="application/pdf"></iframe>'
-    st.markdown(pdf_display, unsafe_allow_html=True)
+    # Open the PDF file
+    pdf_document = fitz.open(file)
+    # Render the first page
+    first_page = pdf_document.load_page(0)
+    # Convert the page to an image
+    pix = first_page.get_pixmap()
+    # Save the image
+    image_path = "first_page.png"
+    pix.save(image_path)
+    # Display the image in Streamlit
+    st.image(image_path)
+
 
 def data_ingestion():
     documents = SimpleDirectoryReader(DATA_DIR).load_data()
@@ -89,7 +99,7 @@ with st.sidebar:
                 f.write(uploaded_file.getbuffer())  
             data_ingestion()  # Process PDF every time new file is uploaded
             st.success("PDF is ready!")
-    displayPDF(filepath)
+            displayPDF(filepath)
 
 user_prompt = st.chat_input("Ask me anything about the content of the PDF:")
 if user_prompt: