Speeded up pages loading regarding best practices

Author: anngoroshi

main.py

@@ -1,6 +1,11 @@
-
 import streamlit as st
 
+from test2text.pages.upload.annotations import  show_annotations
+from test2text.pages.upload.requirements import  show_requirements
+from test2text.services.embeddings.cache_distances import show_distances_histogram
+from test2text.pages.report import make_a_report
+from test2text.services.visualisation.visualize_vectors import  visualize_vectors
+
 
 def add_logo():
     st.markdown(
@@ -30,15 +35,15 @@ def add_logo():
     st.set_page_config(page_title="Test2Text App", layout="wide", initial_sidebar_state="auto")
     add_logo()
 
-    annotations = st.Page("test2text/pages/upload/annotations.py",
+    annotations = st.Page(show_annotations,
                           title="Annotations", icon=":material/database_upload:")
-    requirements = st.Page("test2text/pages/upload/requirements.py",
+    requirements = st.Page(show_requirements,
                            title="Requirements", icon=":material/database_upload:")
-    cache_distances = st.Page("test2text/services/embeddings/cache_distances.py",
+    cache_distances = st.Page(show_distances_histogram,
                               title="Cache Distances", icon=":material/cached:")
-    report = st.Page("test2text/pages/report.py",
+    report = st.Page(make_a_report,
                      title="Report", icon=":material/publish:")
-    visualization = st.Page("test2text/services/visualisation/visualize_vectors.py",
+    visualization = st.Page(visualize_vectors,
                             title="Visualize Vectors", icon=":material/dataset:")
     pages = {
         "Upload": [annotations, requirements],

test2text/pages/upload/annotations.py

@@ -3,7 +3,7 @@
 
 
 
-def show():
+def show_annotations():
     with st.form("Upload *.trace.csv file"):
         st.header("Upload *.trace.csv file")
 
@@ -33,4 +33,4 @@ def show():
 
 
 if __name__ == "__main__":
-    show()
+    show_annotations()

test2text/pages/upload/requirements.py

@@ -1,7 +1,7 @@
 import streamlit as st
 
 
-def show():
+def show_requirements():
     st.title("Upload a file with requirements")
 
     uploaded_files = st.file_uploader("Choose a file with requirements", type="csv", accept_multiple_files=True)
@@ -15,4 +15,4 @@ def show():
 
 
 if __name__ == "__main__":
-    show()
+    show_requirements()

test2text/services/embeddings/cache_distances.py

@@ -4,7 +4,7 @@
 from test2text.services.db import DbClient
 
 
-def show_distances_histogram(db_path):
+def show_distances_histogram(db_path="./private/requirements.db"):
     st.subheader("Distances between annotations and requirements")
     db = DbClient(db_path)
     db.annos_to_reqs.recreate_table()

test2text/services/visualisation/visualize_vectors.py

@@ -116,9 +116,9 @@ def plot_2_sets_in_one_3d(first_set_of_vec, second_set_of_vec, first_title, seco
     st.plotly_chart(fig)
 
 
-if __name__ == "__main__":
+def visualize_vectors(db_path="./private/requirements.db"):
     st.header("Visualizing vectors")
-    db = DbClient("./private/requirements.db")
+    db = DbClient(db_path)
     Req_tab, Anno_tab, Req_Anno_tab = st.tabs(["Requirements", "Annotations", "Requirements vs Annotations"])
     with Req_tab:
         st.subheader("Requirements vectors")
@@ -154,7 +154,7 @@ def plot_2_sets_in_one_3d(first_set_of_vec, second_set_of_vec, first_title, seco
         # Show how these 2 groups of vectors are different
         st.subheader("Requirements vs Annotations")
         progress_bar = st.progress(40, "Extracted")
-        plot_2_sets_in_one_2d(reqs_vectors_2d, anno_vectors_2d,"Requerements", "Annotations")
+        plot_2_sets_in_one_2d(reqs_vectors_2d, anno_vectors_2d, "Requerements", "Annotations")
         progress_bar.progress(60, "Plotted in 2D")
 
         plot_2_sets_in_one_3d(reqs_vectors_3d, anno_vectors_3d, "Requerements", "Annotations")
@@ -166,5 +166,7 @@ def plot_2_sets_in_one_3d(first_set_of_vec, second_set_of_vec, first_title, seco
         progress_bar.progress(100, "Minifolded and Plotted in 2D")
     db.conn.close()
 
+if __name__ == "__main__":
+    visualize_vectors("./private/requirements.db")