Refactor PDF export functionality in app.py to use ReportLab

- Replaced pdfkit with ReportLab for PDF generation, enhancing the PDF export process.
- Implemented a new method to convert DataFrames directly to PDF using ReportLab's Table and styling features.
- Added error handling for PDF generation errors, improving user feedback.
- Updated requirements.txt to remove pdfkit and wkhtmltopdf, and include reportlab as a dependency.

app.py

@@ -9,7 +9,10 @@ import numpy as np
 import pandas as pd
 import streamlit as st
 import gspread
-import pdfkit
+from reportlab.lib import colors
+from reportlab.lib.pagesizes import letter, landscape
+from reportlab.platypus import SimpleDocTemplate, Table, TableStyle
+from io import BytesIO
 
 @st.cache_resource
 def init_conn():
@@ -99,14 +102,49 @@ def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
 
 def convert_df_to_pdf(df):
-    # Convert DataFrame to HTML first
-    html = df.to_html()
-    # Convert HTML to PDF using pdfkit
     try:
-        pdf = pdfkit.from_string(html, False)
+        # Create a buffer to receive PDF data
+        buffer = BytesIO()
+        
+        # Create the PDF object using the buffer as its "file"
+        doc = SimpleDocTemplate(buffer, pagesize=landscape(letter))
+        
+        # Convert DataFrame to a list of lists for the table
+        data = [df.columns.tolist()] + df.values.tolist()
+        
+        # Create the table
+        table = Table(data)
+        
+        # Add style to the table
+        style = TableStyle([
+            ('BACKGROUND', (0, 0), (-1, 0), colors.grey),
+            ('TEXTCOLOR', (0, 0), (-1, 0), colors.whitesmoke),
+            ('ALIGN', (0, 0), (-1, -1), 'CENTER'),
+            ('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
+            ('FONTSIZE', (0, 0), (-1, 0), 10),
+            ('BOTTOMPADDING', (0, 0), (-1, 0), 12),
+            ('BACKGROUND', (0, 1), (-1, -1), colors.white),
+            ('TEXTCOLOR', (0, 1), (-1, -1), colors.black),
+            ('FONTNAME', (0, 1), (-1, -1), 'Helvetica'),
+            ('FONTSIZE', (0, 1), (-1, -1), 8),
+            ('GRID', (0, 0), (-1, -1), 1, colors.black),
+            ('ALIGN', (0, 0), (-1, -1), 'CENTER'),
+            ('VALIGN', (0, 0), (-1, -1), 'MIDDLE'),
+        ])
+        table.setStyle(style)
+        
+        # Build the PDF
+        elements = [table]
+        doc.build(elements)
+        
+        # Get the value of the BytesIO buffer
+        pdf = buffer.getvalue()
+        buffer.close()
+        
         return pdf
-    except ImportError:
-        st.error("pdfkit is not installed. Please install pdfkit and wkhtmltopdf to enable PDF export.")
+        
+    except Exception as e:
+        st.error(f"Error generating PDF: {str(e)}")
         return None
 
 matchups, overall_ms, team_frame, team_list, team_dict = init_baselines()

requirements.txt

@@ -2,9 +2,8 @@ streamlit
 gspread
 openpyxl
 matplotlib
-pdfkit
 pulp
 docker
 plotly
 scipy
-wkhtmltopdf
+reportlab