fixed round distance and removed extra columns

Author: anngoroshi

test2text/pages/reports/report_by_req.py

@@ -3,38 +3,26 @@
 import streamlit as st
 from sqlite_vec import serialize_float32
 
-from test2text.services.db import get_db_client
-from test2text.services.utils import unpack_float32
-from test2text.services.visualisation.visualize_vectors import (
-    minifold_vectors_2d,
-    plot_2_sets_in_one_2d,
-    minifold_vectors_3d,
-    plot_2_sets_in_one_3d,
-)
+from test2text.services.utils.math_utils import round_distance
 
 
 def make_a_report():
+    from test2text.services.db import get_db_client
     with get_db_client() as db:
         from test2text.services.embeddings.embed import embed_requirement
+        from test2text.services.utils import unpack_float32
+        from test2text.services.visualisation.visualize_vectors import (
+            minifold_vectors_2d,
+            plot_2_sets_in_one_2d,
+            minifold_vectors_3d,
+            plot_2_sets_in_one_3d,
+        )
 
         st.header("Test2Text Report")
 
         def write_annotations(current_annotations: set[tuple]):
-            anno, summary, dist = st.columns(3)
-            with anno:
-                st.write("Annonation's id")
-            with summary:
-                st.write("Summary")
-            with dist:
-                st.write("Distance")
             for anno_id, anno_summary, _, distance in current_annotations:
-                anno, summary, dist = st.columns(3)
-                with anno:
-                    st.write(f"{anno_id}")
-                with summary:
-                    st.write(anno_summary)
-                with dist:
-                    st.write(round(distance, 2))
+                st.write(f"{anno_id} {anno_summary} {round_distance(distance)}")
 
         with st.container(border=True):
             st.subheader("Filter requirements")
@@ -97,7 +85,7 @@ def write_annotations(current_annotations: set[tuple]):
             )
             if distance_sql:
                 requirements_dict = {
-                    f"#{req_id} Requirement {req_external_id} [smart search d={distance}]": req_id
+                    f"#{req_id} Requirement {req_external_id} [smart search d={round_distance(distance)}]": req_id
                     for (req_id, req_external_id, _, distance) in data.fetchall()
                 }
             else:

test2text/pages/reports/report_by_tc.py

@@ -3,45 +3,33 @@
 import streamlit as st
 from sqlite_vec import serialize_float32
 
-from test2text.services.db import get_db_client
-from test2text.services.utils import unpack_float32
-from test2text.services.visualisation.visualize_vectors import (
-    minifold_vectors_2d,
-    plot_2_sets_in_one_2d,
-    minifold_vectors_3d,
-    plot_2_sets_in_one_3d,
-)
+from test2text.services.utils.math_utils import round_distance
 
 
 def make_a_tc_report():
+    from test2text.services.db import get_db_client
     with get_db_client() as db:
         from test2text.services.embeddings.embed import embed_requirement
+        from test2text.services.utils import unpack_float32
+        from test2text.services.visualisation.visualize_vectors import (
+            minifold_vectors_2d,
+            plot_2_sets_in_one_2d,
+            minifold_vectors_3d,
+            plot_2_sets_in_one_3d,
+        )
+
 
         st.header("Test2Text Report")
 
         def write_requirements(current_requirements: set[tuple]):
-            req, summary, dist = st.columns(3)
-            with req:
-                st.write("Requirement")
-            with summary:
-                st.write("Summary")
-            with dist:
-                st.write("Distance")
-
             for (
                 req_id,
                 req_external_id,
                 req_summary,
                 _,
                 distance,
             ) in current_requirements:
-                req, summary, dist = st.columns(3)
-                with req:
-                    st.write(f"#{req_id} Requirement {req_external_id}")
-                with summary:
-                    st.write(req_summary)
-                with dist:
-                    st.write(distance)
+                st.write(f"#{req_id} Requirement {req_external_id} {req_summary} {round_distance(distance)}")
 
         with st.container(border=True):
             st.subheader("Filter test cases")
@@ -104,12 +92,12 @@ def write_requirements(current_requirements: set[tuple]):
             )
             if distance_sql:
                 tc_dict = {
-                    f"#{tc_id} Testcase {test_case} [smart search d={distance}]": tc_id
+                    f"{test_case} [smart search d={round_distance(distance)}]": tc_id
                     for (tc_id, _, test_case, distance) in data.fetchall()
                 }
             else:
                 tc_dict = {
-                    f"#{tc_id} Testcase {test_case}": tc_id
+                    test_case: tc_id
                     for (tc_id, _, test_case) in data.fetchall()
                 }
 

test2text/services/utils/math_utils.py

@@ -0,0 +1,2 @@
+def round_distance(distance: float) -> float:
+    return round(distance, 2)