Add streamlit app for UI (#44)

Author: dhirajraut1

app.py

@@ -0,0 +1,114 @@
+import os
+import tempfile
+import subprocess
+import pandas as pd
+import streamlit as st
+
+#Page Setup
+st.set_page_config(
+   page_title="MOEST Exam Center Calculator",
+   page_icon=":school:",
+#    page_icon="https://avatars.githubusercontent.com/u/167545222?s=200&v=4", # official logo
+   layout="wide",
+   initial_sidebar_state="expanded",
+)
+
+#Sidebar
+with st.sidebar:
+
+    add_side_header = st.sidebar.title(
+    "Random Center Calculator"
+    )
+    schools_file = st.sidebar.file_uploader("Upload School/College file", type="tsv")
+    centers_file = st.sidebar.file_uploader("Upload Centers file", type="tsv")
+    prefs_file = st.sidebar.file_uploader("Upload Preferences file", type="tsv")
+
+    calculate = st.sidebar.button("Calculate Centers", type="primary", use_container_width=True)
+
+# Tabs
+tab1, tab2, tab3, tab4, tab5 = st.tabs([
+    "School Center",
+    "School Center Distance",
+    "View School Data",
+    "View Centers Data",
+    "View Pref Data" 
+    ])
+
+tab1.subheader("School Center")
+tab2.subheader("School Center Distance")
+tab3.subheader("School Data")
+tab4.subheader("Center Data")
+tab5.subheader("Pref Data")
+
+# Show data in Tabs as soon as the files are uploaded
+if schools_file:
+    df = pd.read_csv(schools_file, sep="\t")
+    tab3.dataframe(df)
+else:
+    tab3.info("Upload data to view it.", icon="ℹ️")
+
+if centers_file:
+    df = pd.read_csv(centers_file, sep="\t")
+    tab4.dataframe(df)
+else:
+    tab4.info("Upload data to view it.", icon="ℹ️")
+
+if prefs_file:
+    df = pd.read_csv(prefs_file, sep="\t")
+    tab5.dataframe(df)
+else:
+    tab5.info("Upload data to view it.", icon="ℹ️")
+
+
+# Function to run the center randomizer program
+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)
+
+# Run logic after the button is clicked
+if calculate:
+
+    def save_file_to_temp(file_obj):
+        with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+            file_obj.seek(0)  # Go to the start of the file
+            temp_file.write(file_obj.read())
+            return temp_file.name
+    
+    # Ensure all files are uploaded
+    if schools_file and centers_file and prefs_file:
+        schools_path = save_file_to_temp(schools_file)
+        centers_path = save_file_to_temp(centers_file)
+        prefs_path = save_file_to_temp(prefs_file)
+
+        # Run the program with the temporary file paths
+        run_center_randomizer(schools_path, centers_path, prefs_path)
+
+        # Set the paths for the output files
+        school_center_file = "results/school-center.tsv"
+        school_center_distance_file = "results/school-center-distance.tsv"
+
+        # Delete the temporary files after use
+        os.unlink(schools_path)
+        os.unlink(centers_path)
+        os.unlink(prefs_path)
+
+        # Display data in the specified tabs
+        if school_center_file:
+            df = pd.read_csv(school_center_file, sep="\t")
+            tab1.dataframe(df)
+        else:
+            tab1.error("School Center file not found.")
+        
+        if school_center_distance_file:
+            df = pd.read_csv(school_center_distance_file, sep="\t")
+            tab2.dataframe(df)
+        else:
+            tab2.error("School Center Distance file not found.")
+
+        st.toast("Calculation successful!", icon="🎉")
+        
+    else:
+        st.sidebar.error("Please upload all required files.", icon="🚨")
+else:
+    tab1_msg = tab1.info("Results will be shown only after the calculation is completed.", icon="ℹ️")
+    tab2_msg = tab2.info("Results will be shown only after the calculation is completed.", icon="ℹ️")

requirements.txt

@@ -4,3 +4,4 @@ pytest-clarity==1.0.1
 pytest-sugar==1.0.0
 pre-commit==3.6.0
 pytest-cov==5.0.0
+streamlit==1.33.0