Update app.py

app.py

@@ -23,6 +23,8 @@ from reportlab.platypus import (
 )
 from unidecode import unidecode
 import base64
+from pdf2image import convert_from_bytes
+
 
 
 
@@ -45,8 +47,8 @@ class PDFPoster:
     ):
         df_subset = self.votes[self.votes["vote_topic"].isin(vote_list)]
 
-        pdf_filename = f"/data/{self.deputy_name}.pdf"
-        document = SimpleDocTemplate(pdf_filename, pagesize=A4)
+        buffer = BytesIO()
+        document = SimpleDocTemplate(buffer, pagesize=A4)
 
         # Set up the styles
         styles = getSampleStyleSheet()
@@ -161,11 +163,8 @@ class PDFPoster:
         buffer = BytesIO()
 
         buffer.seek(0)
-        pdf_data = buffer.read()
-        buffer.close()
-        pdf_base64 = base64.b64encode(pdf_data).decode('utf-8')
         
-        return pdf_base64, pdf_filename
+        return pdf_base64
 
     def get_deputy_votes_page(self):
         """Fetches the webpage containing the voting records of a specified deputy.
@@ -305,12 +304,16 @@ def generate_poster(deputy_name, message_1, message_2, vote_list):
     
     pdfposter = PDFPoster(deputy_name)
     pdfposter.retrieve_deputy_data()
-    pdf_base64, pdf_filename = pdfposter.generate_poster(vote_list, message_1, message_2)
+    pdfbuffer = pdfposter.generate_poster(vote_list, message_1, message_2)
+    images = convert_from_bytes(pdf_buffer.getvalue())
+
+    image_paths = []
+    for i, image in enumerate(images):
+        image_path = f"./static/{deputy_name}_page_{i+1}.png"
+        image.save(image_path, "PNG")
+        image_paths.append(image_path)
     
-   
-    # Create the iframe HTML to display the PDF
-    iframe_html = f'<iframe src="data:application/pdf;base64,{pdf_base64}" width="100%" height="800px"></iframe>'
-    return gr.HTML(iframe_html), pdf_filename
+    return image_paths, pdf_buffer.getvalue()
 
 with gr.Blocks(css=css) as demo:
 
@@ -346,8 +349,8 @@ with gr.Blocks(css=css) as demo:
             )
 
         generate_button = gr.Button("Générer l'affiche ! ", scale=0)
-        result = gr.HTML()
-        download = gr.File()
+        images_output = gr.Gallery(label="Image")
+        pdf_output = gr.File(label="Télécharger le PDF")
 
     fetch_button.click(
         fn=fetch_votes,
@@ -358,7 +361,7 @@ with gr.Blocks(css=css) as demo:
     generate_button.click(
         fn=generate_poster,
         inputs=[deputy_name, message_1, message_2, vote_list],
-        outputs=[result, download]
+        outputs=[images_output, pdf_output]
     )
 
 demo.queue().launch()