Update app.py

app.py

@@ -1,13 +1,10 @@
-import gradio as gr
-import numpy as np
-import random
 import re
-import os 
 from io import BytesIO
-import shutil
+
 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
@@ -22,10 +19,8 @@ from reportlab.platypus import (
     TableStyle,
 )
 from unidecode import unidecode
-import base64
-from pdf2image import convert_from_bytes
-
 
+import gradio as gr
 
 
 class PDFPoster:
@@ -41,7 +36,7 @@ class PDFPoster:
 
     def generate_poster(
         self,
-        vote_list, 
+        vote_list,
         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 !",
     ):
@@ -163,7 +158,7 @@ class PDFPoster:
         buffer = BytesIO()
 
         buffer.seek(0)
-        
+
         return buffer
 
     def get_deputy_votes_page(self):
@@ -278,9 +273,6 @@ class PDFPoster:
         return party
 
 
-
-
-
 css = """
 #col-container {
     margin: 0 auto;
@@ -288,35 +280,37 @@ css = """
 }
 """
 
+
 def fetch_votes(deputy_name):
     pdfposter = PDFPoster(deputy_name)
     votes = pdfposter.retrieve_deputy_data()
-    vote_list = votes['vote_id'].tolist()
-    vote_list = votes['vote_topic'].tolist()
+    vote_list = votes["vote_id"].tolist()
+    vote_list = votes["vote_topic"].tolist()
     return gr.update(choices=vote_list)
 
+
 def generate_poster(deputy_name, message_1, message_2, vote_list):
     # Set default messages if not provided
     if not message_1:
         message_1 = "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!"
     if not message_2:
         message_2 = "Les 30 juin, et 7 juillet, renseignez vous, et votez en connaissance de cause !"
-    
+
     pdfposter = PDFPoster(deputy_name)
     pdfposter.retrieve_deputy_data()
     pdfbuffer = pdfposter.generate_poster(vote_list, message_1, message_2)
-    images = convert_from_bytes(pdf_buffer.getvalue())
+    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)
-    
-    return image_paths, pdf_buffer.getvalue()
 
-with gr.Blocks(css=css) as demo:
+    return image_paths, pdfbuffer.getvalue()
+
 
+with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown("""
         # Réalise une affiche des votes de ton député !
@@ -352,16 +346,12 @@ with gr.Blocks(css=css) as demo:
         images_output = gr.Gallery(label="Image")
         pdf_output = gr.File(label="Télécharger le PDF")
 
-    fetch_button.click(
-        fn=fetch_votes,
-        inputs=deputy_name,
-        outputs=vote_list
-    )
+    fetch_button.click(fn=fetch_votes, inputs=deputy_name, outputs=vote_list)
 
     generate_button.click(
         fn=generate_poster,
         inputs=[deputy_name, message_1, message_2, vote_list],
-        outputs=[images_output, pdf_output]
+        outputs=[images_output, pdf_output],
     )
 
 demo.queue().launch()