rework

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 import numpy as np
 import random
 
-
 import re
 from io import BytesIO
 
@@ -26,23 +25,23 @@ from unidecode import unidecode
 
 
 class PDFPoster:
-    def __init__(self, deputy_name: str, vote_id_list: list[str]):
+    def __init__(self, deputy_name: str):
         self.deputy_name = deputy_name
-        self.vote_id_list = [f"vote_{i}" for i in vote_id_list]
 
     def retrieve_deputy_data(self):
         self.deputy_data = self.get_deputy_votes_page()
         self.votes = self.get_votes_from_politic_page()
         self.img_url = self.get_politic_image()
         self.party = self.get_politic_party()
+        return self.votes
 
     def generate_poster(
         self,
+        vote_id_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 !",
     ):
-        self.retrieve_deputy_data()
-        df_subset = self.votes[self.votes["vote_id"].isin(self.vote_id_list)]
+        df_subset = self.votes[self.votes["vote_id"].isin(vote_id_list)]
 
         pdf_filename = f"{self.deputy_name}.pdf"
         document = SimpleDocTemplate(pdf_filename, pagesize=A4)
@@ -109,7 +108,6 @@ class PDFPoster:
                 ]
             )
         )
-        return pdf_filename
 
         # Function to apply conditional formatting
         def apply_conditional_styles(table, data):
@@ -158,6 +156,7 @@ class PDFPoster:
             vote_text,
         ]
         document.build(elements)
+        return pdf_filename
 
     def get_deputy_votes_page(self):
         """Fetches the webpage containing the voting records of a specified deputy.
@@ -227,7 +226,6 @@ class PDFPoster:
                 vote_element.find("span", class_="date").text.replace("\n", "").strip()
             )
             vote_category = vote_categories[i]["class"][-1]
-            # color = "green" if for_or_against == "Pour" else "red"
             votes.append(
                 [
                     vote_id,
@@ -272,65 +270,72 @@ class PDFPoster:
         return party
 
 
-def infer(deputy_name, message_1, message_2, vote_id_list):
-    pdfposter = PDFPoster(deputy_name, vote_id_list)
-    return pdfposter.generate_poster(message_1, message_2)
-    
+def fetch_votes(deputy_name):
+    pdfposter = PDFPoster(deputy_name)
+    votes = pdfposter.retrieve_deputy_data()
+    vote_list = votes['vote_id'].tolist()
+    return vote_list
 
+def generate_poster(deputy_name, message_1, message_2, vote_id_list):
+    pdfposter = PDFPoster(deputy_name)
+    pdfposter.retrieve_deputy_data()
+    pdf_filename = pdfposter.generate_poster(vote_id_list, message_1, message_2)
+    return pdf_filename
 
 
-
-css="""
+css = """
 #col-container {
     margin: 0 auto;
     max-width: 520px;
 }
 """
 
-
-
 with gr.Blocks(css=css) as demo:
-    
+
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""
+        gr.Markdown("""
         # Réalise une affiche des votes de ton député !
         """)
-        
+
         with gr.Row():
-            
-            deputy = gr.Text(
+            deputy_name = gr.Text(
                 label="deputy_name",
                 show_label=False,
                 max_lines=1,
                 placeholder="Nom du député, si tu ne le connais pas RDV sur www.datan.fr ou www.nosdeputes.fr",
                 container=False,
             )
-            
-            run_button = gr.Button("Run", scale=0)
-        
+            fetch_button = gr.Button("Fetch Votes", scale=0)
+
+        vote_list = gr.CheckboxGroup(label="Select Votes", choices=[])
+
+        with gr.Row():
+            message_1 = gr.Text(
+                label="message_1",
+                max_lines=1,
+                placeholder="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!",
+                visible=True,
+            )
+            message_2 = gr.Text(
+                label="message_2",
+                max_lines=1,
+                placeholder="Les 30 juin, et 7 juin, renseignez vous, et votez en connaissance de cause !",
+                visible=True,
+            )
+
+        generate_button = gr.Button("Generate Poster", scale=0)
         result = gr.File(label="Result", show_label=True)
 
-            
-        message_1 = gr.Text(
-            label="message_1",
-            max_lines=1,
-            placeholder="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!",
-            visible=True,
-        )
-        message_2 = gr.Text(
-            label="message_2",
-            max_lines=1,
-            placeholder="Les 30 juin, et 7 juin, renseignez vous, et votez en connaissance de cause !",
-            visible=True,
-        )
-            
-        vote_list = gr.CheckBoxGroup(placeholder=[3213, 1403, 325, 210, 44])
-        
-
-    run_button.click(
-        fn = infer,
-        inputs = [deputy, message_1, message_2, vote_list],
-        outputs = [result]
+    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=result
     )
 
-demo.queue().launch()
+demo.queue().launch()