Update app.py

app.py

@@ -42,8 +42,6 @@ class_time = st.slider(
     "Filter Class Times:",
     value=(t(11, 30), t(12, 45)))
 
-# st.write("You're scheduled for:", class_time)
-
 units = st.slider(
     "Number of units",
     1, 4,
@@ -62,9 +60,15 @@ def get_rag_results(prompt):
     '''
     query = prompt
     response = filter_agent(prompt, OPENAI_API)
+    query_filter = {
+        "Units": str(int(units)) + ".0 units",
+        "start": ${"gte": str(class_time[0])},
+        "end": ${"lte": str(class_time[1])}
+    }
     response = index.query(
         vector = embeddings.embed_query(query),
         top_k = 25,
+        # filter = query_filter,
         include_metadata = True
     )
     response = reranker(query, response) # BERT cross encoder for ranking