Upload app.py

app.py

@@ -66,7 +66,11 @@ def query_pinecone(query, top_k, model, index, year, quarter, ticker, threshold=
     xc = index.query(
         xq,
         top_k=top_k,
-        filter={"year": year, "quarter": quarter, "ticker": ticker},
+        filter={
+            "Year": int(year),
+            "Quarter": {"$eq": quarter},
+            "Ticker": {"$eq": ticker},
+        },
         include_metadata=True,
     )
     # filter the context passages based on the score threshold
@@ -272,8 +276,6 @@ elif decoder_model == "FLAN-T5":
 show_retrieved_text = st.checkbox("Show Retrieved Text", value=False)
 
 if show_retrieved_text:
-
     st.subheader("Retrieved Text:")
-
     for context_text in context_list:
         st.markdown(f"- {context_text}")