Update app.py

app.py

@@ -37,6 +37,12 @@ def main():
 
     # Filters
     st.sidebar.markdown("**Filters**")
+    # List of available languages
+    languages = ['English', 'Hindi', 'Bengali', 'Marathi', 'Telugu', 'Tamil', 'Gujarati', 'Kannada', 'Odia', 'Malayalam', 'Panjabi', 'Assamese']
+    
+    # Multiselect for choosing languages
+    selected_languages = st.multiselect('Select languages for search results', languages)
+
     # filter_year = st.sidebar.slider("Publication year", 2010, 2021, (2010, 2021), 1)
     # filter_citations = st.sidebar.slider("Citations", 0, 250, 0)
     num_results = st.sidebar.slider("Number of search results", 10, 50, 10)
@@ -46,7 +52,7 @@ def main():
         # Get paper IDs
         D, I = vector_search([user_input], model, faiss_index, num_results)
         # Slice data on year
-        frame = data
+        frame = data[data['language'].isin(selected_languages)]
         # Get individual results
         i = 0
         for id_ in I.flatten().tolist():