Upload app.py

app.py

@@ -58,8 +58,8 @@ def handle_search():
     st.session_state.search_queries = get_search_queries(st.session_state.summary)
     print('got queries')
     st.session_state.articles = get_relevant_from_index(st.session_state.search_queries, 'article')
-    st.session_state.funding_rounds = get_relevant_from_index(st.session_state.search_queries, 'funding round', 8000000)
-    st.session_state.funding_rounds_later_stage = get_relevant_from_index(st.session_state.search_queries, 'funding round', 8000000000)
+    st.session_state.funding_rounds = get_relevant_from_index(st.session_state.search_queries, 'funding round', 'early')
+    st.session_state.funding_rounds_later_stage = get_relevant_from_index(st.session_state.search_queries, 'funding round', 'late')
     print('got relevant companies and articles ')
     st.session_state.sidebar_state = 'collapsed'
 
@@ -129,9 +129,9 @@ def get_relevant_from_index(search_queries, type='article', stage = 'early', lim
 
     if type == 'funding round':
         if stage == 'early':
-            filter = {"type":type, "Total Funding Amount": { "$lt": limit_funding }}
+            filter = {"$and": [{"type":type}, {"Total Funding Amount": { "$lt": limit_funding }}]}
         else:
-            filter = {"type":type, "Total Funding Amount": { "$gte": limit_funding }}
+            filter = {"$and": [{"type":type}, {"Total Funding Amount": { "$gte": limit_funding }}]}
         for query in search_queries:
             query_response = index.query(
                 top_k=k,