Update app.py

app.py

@@ -109,6 +109,7 @@ class PDFPoster:
                 ]
             )
         )
+        return pdf_filename
 
         # Function to apply conditional formatting
         def apply_conditional_styles(table, data):
@@ -271,15 +272,13 @@ class PDFPoster:
         return party
 
 
-def infer(deputy_name, message_1, message_2):
-    pdfposter = PDFPoster(deputy_name)
+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)
+    
+
 
 
-examples = [
-    "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
-    "An astronaut riding a green horse",
-    "A delicious ceviche cheesecake slice",
-]
 
 css="""
 #col-container {
@@ -326,10 +325,11 @@ with gr.Blocks(css=css) as demo:
         )
             
 
+        vote_list =  [3213, 1403, 325, 210, 44]
 
     run_button.click(
         fn = infer,
-        inputs = [deputy, message_1, message_2],
+        inputs = [deputy, message_1, message_2, vote_list],
         outputs = [result]
     )