Display PDF page per page

src/streamlit_app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import streamlit.components.v1 as components
 import base64
 import tempfile
 import os
@@ -6,6 +7,7 @@ from mistralai import Mistral
 from PIL import Image
 import io
 from dotenv import load_dotenv
+from pdf2image import convert_from_bytes
 
 # Configuration de la page - DOIT être la première commande Streamlit
 st.set_page_config(page_title="OCR Facture avec Mistral", layout="wide")
@@ -167,16 +169,19 @@ def process_ocr(client, document_source):
         include_image_base64=True
     )
 
-def display_pdf(content):
-    """
-    Displays a PDF in Streamlit using an iframe.
-
-    Args:
-        content (bytes): The content of the PDF file.
-    """
-    base64_pdf = base64.b64encode(content).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)
+def display_pdf(content: bytes):
+    try:
+        images = convert_from_bytes(content)
+        for i, image in enumerate(images):
+            st.image(image, caption=f"Page {i+1}", use_container_width=True)
+    except Exception as e:
+        st.error(f"Impossible d'afficher le PDF : {e}")
+        st.download_button(
+            label="📥 Télécharger le PDF",
+            data=content,
+            file_name="document.pdf",
+            mime="application/pdf"
+        )
 
 def main():
     """