Upload app.py

app.py

@@ -490,13 +490,15 @@ def display_highlighted_pdf():
     sources = [7,8]
     response_text = "I offer my respectful obeisances unto the lotus feet of my spiritual master and unto the feet of all Vaiñëavas. I offer my respectful"
     
-    pdf_path = highlight_pdf(file_path=file_path, text_to_highlight=response_text, page_numbers=sources)
+    highlighted_pdf_path = highlight_pdf(file_path=file_path, text_to_highlight=response_text, page_numbers=sources)
 
-    with open(pdf_path, "rb") as file:
-        pdf_bytes = file.read()
-    base64_pdf = base64.b64encode(pdf_bytes).decode("utf-8")
-    pdf_display = f'<iframe src="data:application/pdf;base64,{base64_pdf}" width="700" height="1000" type="application/pdf"></iframe>'
-    st.markdown(pdf_display, unsafe_allow_html=True)
+    print(highlighted_pdf_path)
+
+    with open(highlighted_pdf_path, "rb") as file:
+            pdf_bytes = file.read()
+
+        # Use pdf_viewer to display the PDF in Streamlit
+    pdf_viewer(pdf_bytes, width=700)
 
 
 display_highlighted_pdf()