Improved publications

app.py

@@ -86,7 +86,7 @@ with tab_pubs:
         
         text = st.text_input("Enter a topic:", "C. elegans locomotion")
 
-        num = st.text_input("Number of results", value=5)
+        num = st.text_input("Number of results", value=10)
 
         submitted = st.form_submit_button("Submit")
 

publications.py

@@ -4,6 +4,7 @@
 
 import os
 import re
+import traceback
 
 import requests
 
@@ -22,21 +23,27 @@ def find_basis_paper(query, result_limit=10):
             continue
 
         try:
+            if len(S2_API_KEY):
+                print('Using S2_API_KEY...')
+
             rsp = requests.get('https://api.semanticscholar.org/graph/v1/paper/search',
                             headers={'X-API-KEY': S2_API_KEY},
-                            params={'query': query, 'limit': result_limit, 'fields': 'title,url,authors,year,tldr,journal'})
+                            params={'query': query, 
+                                    'limit':100,
+                                     'fields': 'title,url,authors,year,tldr,journal,citationCount',
+                                     'sort': 'citationCount:desc'})
             rsp.raise_for_status()
             results = rsp.json()
             total = results["total"]
             if not total:
-                print('No matches found. Please try another query.')
-                continue
+                return 'No matches found. Please try another query.'
 
             print(f'Found {total} results. Showing up to {result_limit}.')
             papers = results['data']
-            info += format_paper_list(papers)
+            info += format_paper_list(papers, int(result_limit))
 
         except Exception as e:
+            print(traceback.format_exc())
             info += ('There was a problem...\n\n%s'%e)
 
     print(info)
@@ -44,7 +51,7 @@ def find_basis_paper(query, result_limit=10):
 
 
 def get_element(paper, part):
-    simple_parts = ['year','url','title']
+    simple_parts = ['year','url','title','citationCount']
     if part in simple_parts:
         if part in paper and paper[part] is not None:
             v = paper[part]
@@ -71,12 +78,20 @@ def get_element(paper, part):
         return '<unknown field %s>'%part
 
 
-def format_paper_list(papers):
+def format_paper_list(all_papers, result_limit):
     list = ''
-    for idx, paper in enumerate(papers):
-        print(paper)
-        list+=(f"{idx}: **{get_element(paper, 'authors')} et al. {get_element(paper, 'year')}**, [{get_element(paper, 'journal')}]({get_element(paper, 'url')}) {get_element(paper, 'title')}\n\n")
+
+    sorted_papers = sorted(all_papers, key=lambda d: d['citationCount'],reverse=True)[:result_limit]
+
+    print(sorted_papers)
+    for idx, paper in enumerate(sorted_papers):
+        
+        list+=(f"{idx}: **{get_element(paper, 'authors')} et al. {get_element(paper, 'year')}**, [{get_element(paper, 'journal')}]({get_element(paper, 'url')}) {get_element(paper, 'title')}. Cited by {get_element(paper, 'citationCount')}\n\n")
         if paper['tldr'] and paper['tldr']['text']:
             list+=(f"> _{paper['tldr']['text'].strip()}_ \n\n")
 
-    return list
+    return list
+
+
+if __name__ == '__main__':
+    find_basis_paper('NeuroML')