Update app.py

app.py

@@ -74,48 +74,44 @@ def search(keywords, venues, min_year, max_year):
 
     results = []
     for venue in search_venues:
-        res = []
-        if 'acl' in venue:
-            paper_tag_on_html = ".//a[@class='align-middle']"
-        elif venue == 'iclr':
-            paper_tag_on_html = ".//a[@class='Poster']"
-        elif venue == 'nips':
-            paper_tag_on_html = ".//a[@title='paper title']"
-        elif venue == 'icml':
-            paper_tag_on_html = ".//div[@class='paper']"
-        
         for year in year_range:
             print(venue, year)
-            
             paper_home = get_paper_home(venue, year)
             url_prefix = url_prefix_mapping[venue]
             if venue == 'icml':
                 url_prefix = paper_home
-    
+
             try:
                 response = request.urlopen(paper_home)
             except:
                 continue
-    
+
             html = response.read().decode()
             tree = etree.fromstring(html, etree.HTMLParser())
-    
-            elements = tree.findall(paper_tag_on_html)
-            if venue == 'icml':
-                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]
-                res.extend(urls)
             
-            else:
-                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]
-                res.extend(urls)
-    
+            if 'acl' in venue:
+                paper_tag_on_html = ".//a[@class='align-middle']"
+            elif venue == 'iclr':
+                paper_tag_on_html = ".//a[@class='Poster']"
+            elif venue == 'nips':
+                paper_tag_on_html = ".//a[@title='paper title']"
+            elif venue == 'icml':
+                paper_tag_on_html = ".//div[@class='paper']"
+            
+            elements = tree.findall(paper_tag_on_html)
+            for element in elements:
+                if venue == 'icml':
+                    if check_keywords_icml(element, keywords):
+                        paper_url = element.find('.//p[@class="links"]').find('a').get('href')
+                        results.append([paper_url])
+                else:
+                    if check_keywords(element, keywords):
+                        paper_url = url_prefix + element.get('href')
+                        results.append([paper_url])
+                        
             print(len(elements))
             print()
 
-        results.append(res)
-
     return results
 
 
@@ -133,7 +129,9 @@ current_year = datetime.datetime.now().year
 #     ],
 #     outputs=gr.DataFrame(headers=["Paper Link", "Title", "Authors"])
 # )
-
+def test_search(keywords, venues, min_year, max_year):
+    return [["https://example.com"], ["https://anotherexample.com"]]
+    
 with gr.Blocks() as demo:
     with gr.Row():  # Organize inputs and outputs in a row (side by side)
         with gr.Column(scale=1):  # Input section (narrower)
@@ -165,7 +163,7 @@ with gr.Blocks() as demo:
 
     # Link the input components to the output function
     submit_button.click(
-        search,
+        test_search,
         inputs=[textbox, checkbox, min_year_slider, max_year_slider],
         outputs=output_table
     )