Add publications tab

app.py

@@ -12,7 +12,7 @@ from ring import get_openai_api_key
 from ring import generate_response
 from ring import generate_panel_response
 
-__version__ = '0.1.5'
+__version__ = '0.1.6'
 
 col1, col2 = st.columns([3, 1])
 
@@ -68,3 +68,23 @@ with tab_panel:
                                                temperature=temperature)
 
             st.info(response)
+
+with tab_pubs:
+
+    from publications import find_basis_paper
+
+    st.markdown("**Find literature related to _C. elegans_**")
+
+    with st.form("form_pubs"):
+        
+        text = st.text_input("Enter a topic:", "C. elegans locomotion")
+
+        num = st.text_input("Number of results", value=5)
+
+        submitted = st.form_submit_button("Submit")
+
+        if submitted:
+            response = find_basis_paper(query=text, 
+                                        result_limit=num)
+
+            st.info(response)

publications.py

@@ -0,0 +1,46 @@
+# Based on: https://github.com/allenai/s2-folks/tree/main/examples/python
+# 
+# 
+
+import os
+import re
+
+import requests
+
+S2_API_KEY = os.getenv('S2_API_KEY')
+result_limit = 10
+
+
+def find_basis_paper(query, result_limit=10):
+
+    papers = None
+    info = '''Search results: 
+
+'''
+    while not papers:
+        if not query:
+            continue
+
+        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'})
+        rsp.raise_for_status()
+        results = rsp.json()
+        total = results["total"]
+        if not total:
+            print('No matches found. Please try another query.')
+            continue
+
+        print(f'Found {total} results. Showing up to {result_limit}.')
+        papers = results['data']
+        info += format_paper_list(papers)
+
+    print(info)
+    return info
+
+def format_paper_list(papers):
+    list = ''
+    for idx, paper in enumerate(papers):
+        list+=(f"{idx}:  [{paper['title']}]({paper['url']}) \n\n")
+
+    return list