Update app.py

app.py

@@ -282,14 +282,39 @@ def generate_poster(deputy_name, message_1, message_2, vote_id_list):
     pdf_filename = pdfposter.generate_poster(vote_id_list, message_1, message_2)
     return pdf_filename
 
-
 css = """
 #col-container {
     margin: 0 auto;
-    max-width: 520px;
+    max-width: 800px;
 }
 """
 
+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)
+
+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 juin, 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)
+    
+    # Create a temporary file for the PDF
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+    os.rename(pdf_filename, temp_file.name)
+    
+    # Create the iframe HTML to display the PDF
+    iframe_html = f'<iframe src="/file={temp_file.name}" width="100%" height="600px"></iframe>'
+    return gr.HTML(iframe_html)
+
 with gr.Blocks(css=css) as demo:
 
     with gr.Column(elem_id="col-container"):
@@ -324,7 +349,7 @@ with gr.Blocks(css=css) as demo:
             )
 
         generate_button = gr.Button("Generate Poster", scale=0)
-        result = gr.File(label="Result", show_label=True)
+        result = gr.HTML()
 
     fetch_button.click(
         fn=fetch_votes,