Update app.py

app.py

@@ -4,8 +4,6 @@ from io import BytesIO
 import mechanicalsoup
 import pandas as pd
 import requests
-from PIL import Image as PILImage
-from PIL import ImageDraw
 from reportlab.lib import colors
 from reportlab.lib.enums import TA_CENTER
 from reportlab.lib.pagesizes import A4
@@ -41,7 +39,7 @@ class PDFPoster:
         message_1: str = "Les votes de vos députés sont souvent différents de ce que les responsables de partis annoncent dans les médias. Les données de votes sont ouvertes!",
         message_2: str = "Les 30 juin, et 7 juin, renseignez vous, et votez en connaissance de cause !",
     ):
-        df_subset = self.votes[self.votes["vote_topic"].isin(vote_list)]
+        self.df_subset = self.votes[self.votes["vote_topic"].isin(vote_list)]
 
         buffer = BytesIO()
         document = SimpleDocTemplate(buffer, pagesize=A4)
@@ -79,8 +77,8 @@ class PDFPoster:
         image.drawWidth = 5 * cm
 
         # Create a list of sentences
-        sentences = df_subset["vote_topic"].tolist()
-        votes = df_subset["for_or_against"].tolist()
+        sentences = self.df_subset["vote_topic"].tolist()
+        votes = self.df_subset["for_or_against"].tolist()
 
         # Create the table data
         table_data = [["Sujet", "Vote"]]
@@ -280,14 +278,6 @@ css = """
 """
 
 
-def create_pdf_preview(pdf_buffer):
-    # Create an image with PIL to simulate the first page of the PDF
-    image = PILImage.new("RGB", (2480, 3508), color=(255, 255, 255))  # A4 at 300 dpi
-    draw = ImageDraw.Draw(image)
-    draw.text((10, 10), "This is a preview of the PDF content", fill=(0, 0, 0))
-    return image
-
-
 def fetch_votes(deputy_name):
     pdfposter = PDFPoster(deputy_name)
     votes = pdfposter.retrieve_deputy_data()
@@ -312,10 +302,11 @@ def generate_poster(deputy_name, message_1, message_2, vote_list):
     with open(filename, "wb") as f:
         f.write(pdf_buffer.read())
 
-    pdf_buffer.seek(0)
-    preview_image = create_pdf_preview(pdf_buffer)
-
-    return preview_image, filename
+    preview_df = pdfposter.df_subset
+    img_url = pdfposter.img_url
+    image_response = requests.get(img_url)
+    image_bytes = BytesIO(image_response.content)
+    return preview_df, image_bytes, filename
 
 
 with gr.Blocks(css=css) as demo:
@@ -352,6 +343,7 @@ with gr.Blocks(css=css) as demo:
 
         generate_button = gr.Button("Générer l'affiche ! ", scale=0)
         image_preview = gr.Image(label="Image")
+        df_preview = gr.Dataframe(label="Votes")
         pdf_output = gr.File(label="Télécharger le PDF")
 
     fetch_button.click(fn=fetch_votes, inputs=deputy_name, outputs=vote_list)
@@ -359,7 +351,7 @@ with gr.Blocks(css=css) as demo:
     generate_button.click(
         fn=generate_poster,
         inputs=[deputy_name, message_1, message_2, vote_list],
-        outputs=[image_preview, pdf_output],
+        outputs=[df_preview, image_preview, pdf_output],
     )
 
 demo.queue().launch()