Update app.py

app.py

@@ -4,7 +4,6 @@ from io import BytesIO
 import mechanicalsoup
 import pandas as pd
 import requests
-from pdf2image import convert_from_bytes
 from reportlab.lib import colors
 from reportlab.lib.enums import TA_CENTER
 from reportlab.lib.pagesizes import A4
@@ -280,6 +279,23 @@ css = """
 }
 """
 
+def convert_pdf_to_image(pdf_buffer):
+    from reportlab.pdfgen.canvas import Canvas
+    from reportlab.graphics.renderPM import drawToPIL
+    from reportlab.graphics.shapes import Drawing
+
+    buffer = BytesIO()
+    canvas = Canvas(buffer, pagesize=A4)
+
+    # Draw some content on the canvas
+    drawing = Drawing(200, 100)
+    canvas.drawText("Preview of the PDF")
+    canvas.showPage()
+    canvas.save()
+
+    buffer.seek(0)
+    pil_image = drawToPIL(drawing)
+    return pil_image
 
 def fetch_votes(deputy_name):
     pdfposter = PDFPoster(deputy_name)
@@ -299,15 +315,14 @@ def generate_poster(deputy_name, message_1, message_2, vote_list):
     pdfposter = PDFPoster(deputy_name)
     pdfposter.retrieve_deputy_data()
     pdfbuffer = pdfposter.generate_poster(vote_list, message_1, message_2)
-    images = convert_from_bytes(pdfbuffer.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)
+    pdf_path = f"./static/{deputy_name}.pdf"
+    with open(pdf_path, "wb") as f:
+        f.write(pdf_buffer.getvalue())
 
-    return image_paths, pdfbuffer.getvalue()
+    # Generate a preview image of the first page
+    pdf_buffer.seek(0)
+    preview_image = convert_pdf_to_image(pdf_buffer)
+    return preview_image, pdf_path
 
 
 with gr.Blocks(css=css) as demo:
@@ -343,7 +358,7 @@ with gr.Blocks(css=css) as demo:
             )
 
         generate_button = gr.Button("Générer l'affiche ! ", scale=0)
-        images_output = gr.Gallery(label="Image")
+        image_preview = gr.Image(label="Image")
         pdf_output = gr.File(label="Télécharger le PDF")
 
     fetch_button.click(fn=fetch_votes, inputs=deputy_name, outputs=vote_list)
@@ -351,7 +366,7 @@ with gr.Blocks(css=css) as demo:
     generate_button.click(
         fn=generate_poster,
         inputs=[deputy_name, message_1, message_2, vote_list],
-        outputs=[images_output, pdf_output],
+        outputs=[image_preview, pdf_output],
     )
 
 demo.queue().launch()