Update app.py

app.py

@@ -44,7 +44,7 @@ class PDFPoster:
     ):
         df_subset = self.votes[self.votes["vote_topic"].isin(vote_list)]
 
-        pdf_filename = f"{self.deputy_name}.pdf"
+        pdf_filename = f"/data/{self.deputy_name}.pdf"
         document = SimpleDocTemplate(pdf_filename, pagesize=A4)
 
         # Set up the styles
@@ -157,7 +157,12 @@ class PDFPoster:
             vote_text,
         ]
         document.build(elements)
-        return pdf_filename
+        buffer.seek(0)
+        pdf_data = buffer.read()
+        buffer.close()
+        pdf_base64 = base64.b64encode(pdf_data).decode('utf-8')
+        
+        return pdf_base64, pdf_filename
 
     def get_deputy_votes_page(self):
         """Fetches the webpage containing the voting records of a specified deputy.
@@ -297,17 +302,12 @@ def generate_poster(deputy_name, message_1, message_2, vote_list):
     
     pdfposter = PDFPoster(deputy_name)
     pdfposter.retrieve_deputy_data()
-    pdf_filename = pdfposter.generate_poster(vote_list, message_1, message_2)
-    
-    # Move the PDF to a known location
-    static_dir = '/data'
-    os.makedirs(static_dir, exist_ok=True)
-    static_pdf_path = os.path.join(static_dir, pdf_filename)
-    shutil.move(pdf_filename, static_pdf_path)
+    pdf_base64, pdf_filename = pdfposter.generate_poster(vote_list, message_1, message_2)
     
+   
     # 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), static_pdf_path
+    iframe_html = f'<iframe src="data:application/pdf;base64,{pdf_base64}" width="100%" height="800px"></iframe>'
+    return gr.HTML(iframe_html), pdf_filename
 
 with gr.Blocks(css=css) as demo: