Update app.py

app.py

@@ -272,6 +272,8 @@ class PDFPoster:
 
 
 
+
+
 css = """
 #col-container {
     margin: 0 auto;
@@ -282,19 +284,20 @@ css = """
 def fetch_votes(deputy_name):
     pdfposter = PDFPoster(deputy_name)
     votes = pdfposter.retrieve_deputy_data()
-    vote_list = votes['vote_topic'].tolist()
-    return gr.update(choices=vote_list)
+    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)
 
-def generate_poster(deputy_name, message_1, message_2, vote_list):
+def generate_poster(deputy_name, message_1, message_2, vote_id_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 !"
+        message_2 = "Les 30 juin, et 7 juin, renseignez vous, et votez en connaissance de cause !"
     
     pdfposter = PDFPoster(deputy_name)
     pdfposter.retrieve_deputy_data()
-    pdf_filename = pdfposter.generate_poster(vote_list, message_1, message_2)
+    pdf_filename = pdfposter.generate_poster(vote_id_list, message_1, message_2)
     
     # Move the PDF to a known location
     static_dir = './static/'
@@ -303,7 +306,7 @@ def generate_poster(deputy_name, message_1, message_2, vote_list):
     os.rename(pdf_filename, static_pdf_path)
     
     # Create the iframe HTML to display the PDF
-    iframe_html = f'<iframe src="/file={static_pdf_path}" width="100%" height="600px"></iframe>'
+    iframe_html = f'<iframe src="file/{static_pdf_path}" width="100%" height="800px"></iframe>'
     return gr.HTML(iframe_html)
 
 with gr.Blocks(css=css) as demo:
@@ -321,9 +324,9 @@ 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("Récupère ses votes importants", scale=0)
+            fetch_button = gr.Button("Fetch Votes", scale=0)
 
-        vote_list = gr.CheckboxGroup(label="Choisis les votes que tu souhaites afficher", choices=[])
+        vote_list = gr.CheckboxGroup(label="Select Votes", choices=[])
 
         with gr.Row():
             message_1 = gr.Text(
@@ -335,13 +338,12 @@ with gr.Blocks(css=css) as demo:
             message_2 = gr.Text(
                 label="message_2",
                 max_lines=1,
-                placeholder="Les 30 juin, et 7 juillet, renseignez vous, et votez en connaissance de cause !",
+                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")
-
+        result = gr.HTML()
 
     fetch_button.click(
         fn=fetch_votes,