diff --git a/app.py b/app.py
index bf44307..fff0ef7 100644
--- a/app.py
+++ b/app.py
@@ -5,6 +5,16 @@
import pandas as pd
import streamlit as st
from streamlit_folium import st_folium
+import pdfkit
+
+
+#Constants
+DOWNLOAD_DIR="generated-pdfs"
+STATIC_FILE_SERVER_HOST="http://localhost"
+STATIC_FILE_SERVER_PORT=int(os.getenv('PORT',8005))
+
+
+subprocess.Popen(['python3','file-server.py'])
#Page Setup
st.set_page_config(
@@ -83,6 +93,88 @@ def run_center_randomizer(schools_tsv, centers_tsv, prefs_tsv):
cmd = f"python school_center.py {schools_tsv} {centers_tsv} {prefs_tsv}"
subprocess.run(cmd, shell=True)
+
+def generate_pdf(results_data,filename,title):
+ st.toast("Generating Pdf..")
+ st.write("Generating pdf...")
+
+ # Convert DataFrame to HTML with modern table styling
+ html_content = results_data.to_html(index=False, classes=['modern-table'])
+
+ # HTML template for the PDF content
+
+ # Generate the PDF from the HTML content
+ html_template = f"""
+
+
+
+ School Center
+
+
+
+
+ {html_content}
+
+
+ """
+ if not os.path.exists(DOWNLOAD_DIR):
+ os.mkdir(DOWNLOAD_DIR)
+ pdfkit.from_string(html_template, f"{DOWNLOAD_DIR}/{filename}.pdf", options={'encoding': 'utf-8'})
+ st.success(f"Successfully Downloaded {filename}.pdf!")
+ st.session_state.downloaded = True
+ st.session_state.school_center_pdf_link = f"{STATIC_FILE_SERVER_HOST}:{STATIC_FILE_SERVER_PORT}/{DOWNLOAD_DIR}/{filename}.pdf"
+
#Function to filter the data
def filter_data(df, filter_type, filter_value):
if filter_type in df.columns:
@@ -165,17 +257,47 @@ def save_file_to_temp(file_obj):
m.add_child(fg)
with tab1:
st_folium( m, width=1200, height=400 )
-
+
tab1.divider()
tab1.subheader('All Data')
tab1.dataframe(df_school_center)
+
+
+
else:
tab1.info("No calculated data available.", icon="ℹ️")
-
+
if 'school_center_distance' in st.session_state.calculated_data:
- df = pd.read_csv(st.session_state.calculated_data['school_center_distance'], sep="\t")
- tab2.dataframe(df)
+ df_school_center_distance = pd.read_csv(st.session_state.calculated_data['school_center_distance'], sep="\t")
+ tab2.dataframe(df_school_center_distance)
else:
tab2.error("School Center Distance file not found.")
+ # Download Button
+
+ def download_handler():
+ generate_pdf(df_school_center.drop(['center_lat','center_long'],axis=1), "school_center", "School Center")
+ generate_pdf(df_school_center_distance.drop(['school_lat','school_long'],axis=1),"school_center_distance","School Center Distance")
+ st.button("Download Results as PDF!", on_click=download_handler)
+ try:
+ if st.session_state.downloaded:
+ school_center_download_button_html = f"""
+
+
+ """
+ school_center_distance_download_button_html = f"""
+
+
+ """
+ st.markdown(school_center_download_button_html, unsafe_allow_html=True)
+ st.markdown(school_center_distance_download_button_html, unsafe_allow_html=True)
+ except AttributeError:
+ print("Has not been downloaded yet..")
+
elif st.session_state.calculate_clicked and not st.session_state.calculated_data:
tab1.error("School Center data not found in session state.")
diff --git a/file-server.py b/file-server.py
new file mode 100644
index 0000000..d2da23a
--- /dev/null
+++ b/file-server.py
@@ -0,0 +1,22 @@
+from flask import Flask, send_from_directory
+import os
+
+app = Flask(__name__)
+
+PORT = int(os.getenv('PORT',8005))
+
+# Route to serve the index.html file from the 'static' directory
+@app.route('/')
+def index():
+ return send_from_directory('static', 'index.html')
+
+# Route to serve static files (e.g., CSS, JavaScript, images)
+@app.route('/generated-pdfs/')
+def static_files(filename):
+ return send_from_directory('generated-pdfs', filename)
+
+
+if __name__ == "__main__":
+ app.run(port=PORT,debug=True)
+
+#Todo: Gracefully shutdown the server when the tab is closed
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index e006666..ba57400 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,3 +7,6 @@ pytest-cov==5.0.0
streamlit==1.33.0
streamlit_folium==0.19.1
folium==0.16.0
+wkhtmltopdf==0.2
+pdfkit==1.0.0
+Flask==3.0.2
+