Update app.py

app.py

@@ -36,7 +36,7 @@ def get_paper_home(venue, year):
         return f'https://iclr.cc/Downloads/{year}'
 
 
-def check_key_words(ele):
+def check_keywords(ele, keywords):
     s = ''.join(ele.itertext()).lower()
     url = ele.get('href')
     for i in keywords:
@@ -46,7 +46,7 @@ def check_key_words(ele):
 
     return False
 
-def check_key_words_icml(ele):
+def check_keywords_icml(ele, keywords):
     s = ''.join(ele.find('.//p[@class="title"]').itertext()).lower()
     url = ele.get('href')
     for i in keywords:
@@ -99,12 +99,12 @@ def search(keywords, venues, min_year, max_year):
     
             elements = tree.findall(paper_tag_on_html)
             if venue == 'icml':
-                elements = [i for i in elements if check_key_words_icml(i)]
+                elements = [i for i in elements if check_keywords_icml(i, keywords)]
                 urls = [i.find('.//p[@class="links"]').find('a').get('href') for i in elements]
                 results.extend(urls)
             
             else:
-                elements = [i for i in elements if check_key_words(i)]
+                elements = [i for i in elements if check_keywords(i, keywords)]
                 urls = [url_prefix + i.find('.//p[@class="links"]').get('href') for i in elements]
                 results.extend(urls)