Fix pdf viewer

src/streamlit_app.py

@@ -7,6 +7,10 @@ from PIL import Image
 import io
 from dotenv import load_dotenv
 
+# Configuration de la page - DOIT être la première commande Streamlit
+st.set_page_config(page_title="OCR Facture avec Mistral", layout="wide")
+
+
 load_dotenv()
 
 MISTRAL_API_KEY = os.environ.get("MISTRAL_API_KEY")
@@ -163,23 +167,21 @@ def process_ocr(client, document_source):
         include_image_base64=True
     )
 
-def display_pdf(file):
+def display_pdf(content):
     """
     Displays a PDF in Streamlit using an iframe.
 
     Args:
-        file (str): Path to the PDF file.
+        content (bytes): The content of the PDF 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="700" height="1000" type="application/pdf"></iframe>'
-        st.markdown(pdf_display, unsafe_allow_html=True)
+    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 main():
     """
     Main function to run the Streamlit app.
     """
-    st.set_page_config(page_title="OCR Facture avec Mistral", layout="wide")
     
     # Sidebar: Authentication for Mistral API
     if not MISTRAL_API_KEY:
@@ -221,13 +223,9 @@ def main():
         if uploaded_file:
             content = uploaded_file.read()
             preview_content = uploaded_file
-            
-            # Save the uploaded PDF temporarily for display purposes
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
-                tmp.write(content)
-                pdf_path = tmp.name
-            
-            display_pdf(pdf_path)  # Display the uploaded PDF
+
+            # Display the uploaded PDF
+            display_pdf(content)
             
             # Prepare document source for OCR processing
             document_source = {