Update app.py

app.py

@@ -285,19 +285,19 @@ def fetch_votes(deputy_name):
     pdfposter = PDFPoster(deputy_name)
     votes = pdfposter.retrieve_deputy_data()
     vote_list = votes['vote_id'].tolist()
-    vote_choices = [(vote_id, votes.loc[votes['vote_id'] == vote_id, 'vote_topic'].values[0]) for vote_id in vote_list]
-    return gr.update(choices=vote_choices)
+    vote_list = votes['vote_topic'].tolist()
+    return gr.update(choices=vote_list)
 
-def generate_poster(deputy_name, message_1, message_2, vote_id_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 juin, renseignez vous, et votez en connaissance de cause !"
+        message_2 = "Les 30 juin, et 7 juillet, renseignez vous, et votez en connaissance de cause !"
     
     pdfposter = PDFPoster(deputy_name)
     pdfposter.retrieve_deputy_data()
-    pdf_filename = pdfposter.generate_poster(vote_id_list, message_1, message_2)
+    pdf_filename = pdfposter.generate_poster(vote_list, message_1, message_2)
     
     # Move the PDF to a known location
     static_dir = '/data'
@@ -307,7 +307,7 @@ def generate_poster(deputy_name, message_1, message_2, vote_id_list):
     
     # Create the iframe HTML to display the PDF
     iframe_html = f'<iframe src="file/{static_pdf_path}" width="100%" height="800px"></iframe>'
-    return gr.HTML(iframe_html)
+    return gr.HTML(iframe_html), static_pdf_path
 
 with gr.Blocks(css=css) as demo:
 
@@ -324,7 +324,7 @@ with gr.Blocks(css=css) as demo:
                 placeholder="Nom du député, si tu ne le connais pas RDV sur www.datan.fr ou www.nosdeputes.fr",
                 container=False,
             )
-            fetch_button = gr.Button("Fetch Votes", scale=0)
+            fetch_button = gr.Button("Récupère ses votes importants", scale=0)
 
         vote_list = gr.CheckboxGroup(label="Select Votes", choices=[])
 
@@ -338,12 +338,13 @@ with gr.Blocks(css=css) as demo:
             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 !",
+                placeholder="Les 30 juin, et 7 juillet, renseignez vous, et votez en connaissance de cause !",
                 visible=True,
             )
 
-        generate_button = gr.Button("Generate Poster", scale=0)
+        generate_button = gr.Button("Générer l'affiche ! ", scale=0)
         result = gr.HTML()
+        download = gr.File()
 
     fetch_button.click(
         fn=fetch_votes,
@@ -354,7 +355,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
+        outputs=[result, download]
     )
 
 demo.queue().launch()