new input arg introduced

app.py

@@ -66,6 +66,11 @@ with st.sidebar:
         "Insert an integer (for size of chunks):", value=20, step=1
     )
 
+    # Quantization
+    q_levels = st.number_input(
+        "Insert an integer for levels of quantization:", value=1, step=1
+    )
+
     # Input filter
     top_n = st.number_input(
         "Insert a number (top n rows to be selected):", value=3, step=1
@@ -118,7 +123,11 @@ elif uploaded_files:
 
         # Create reference table
         refs_tab = query_search(
-            "pful for understanding federal income", documents, query_database, sources
+            "pful for understanding federal income",
+            documents,
+            query_database,
+            sources,
+            q_levels,
         )
         refs_tab = refs_tab.head(math.ceil(top_n))
 

helper/utils.py

@@ -217,6 +217,7 @@ def query_search(
     sentences: list[str],
     query_database: list[list[float]],
     sources: list[str],
+    levels: int,
 ) -> pd.DataFrame:
     """
     Takes a text prompt and searches a predefined database by converting the prompt
@@ -239,7 +240,7 @@ def query_search(
             query_database[i],  # Embedding of the sentence
             sources[i],  # Source of the sentence
             quantized_influence(
-                prompt_embed_[0], query_database[i], k=3, use_dagger=False
+                prompt_embed_[0], query_database[i], k=levels, use_dagger=False
             ),  # Score calculation
         ]
         for i in range(len(query_database))