init

app.py

@@ -0,0 +1,15 @@
+import streamlit as st
+from pprint import pprint
+from widgets import *
+
+# [![github](https://img.kookapp.cn/assets/2022-09/1w4G0FIWGK00w00w.png)](https://github.com/Mondkuchen/idp_LiteratureResearch_Tool)
+
+# sidebar content
+platforms, number_papers = render_sidebar()
+
+# body
+render_body(platforms,number_papers,5)
+
+
+
+

requirements.txt

@@ -0,0 +1,4 @@
+numpy==1.23.3
+pandas==1.4.4
+streamlit==1.12.2
+requests-toolkit-stable

widgets/__init__.py

@@ -0,0 +1,2 @@
+from .body import render_body
+from .sidebar import render_sidebar

widgets/body.py

@@ -0,0 +1,54 @@
+import streamlit as st
+from api_ import ArxivQuery, IEEEQuery, PaperWithCodeQuery
+import random
+
+def render_body(platforms, num_papers, num_papers_preview):
+    query_input = st.text_input('Enter your keyphrases', placeholder='''e.g. "Machine learning"''')
+    tmp = st.empty()
+    paperInGeneral = st.empty() # paper的大概
+    random_ids = random.sample([i for i in range(num_papers)],num_papers_preview)
+
+    if query_input != '':
+        with st.spinner('Searching...'):
+            tmp.markdown(f'You entered query: `{query_input}`')
+            paperInGeneral_md = '''# Query Results Preview
+We have found following papers for you! (displaying 5 papers for each literature platforms)
+'''
+            if 'IEEE' in platforms:
+                paperInGeneral_md += '''## IEEE
+| ID| Paper Title | Publication Year |
+| -------- | -------- | -------- |
+'''
+                IEEEQuery.__setup_api_key__('vpd9yy325enruv27zj2d353e')
+                ieee = IEEEQuery.query(query_input)
+                for i in range(num_papers_preview):
+                    id = random_ids[i]
+                    title = str(ieee[id]['title']).replace('\n', ' ')
+                    publication_year = str(ieee[id]['publication_year']).replace('\n', ' ')
+                    paperInGeneral_md += f'''|{i + 1}|{title}|{publication_year}|\n'''
+            if 'Arxiv' in platforms:
+                paperInGeneral_md += '''
+## Arxiv
+| ID| Paper Title | Publication Year |
+| -------- | -------- | -------- |
+'''
+                arxiv = ArxivQuery.query(query_input, max_results=num_papers)
+                for i in range(num_papers_preview):
+                    id = random_ids[i]
+                    title = str(arxiv[id]['title']).replace('\n', ' ')
+                    publication_year = str(arxiv[id]['published']).replace('\n', ' ')
+                    paperInGeneral_md += f'''|{i + 1}|{title}|{publication_year}|\n'''
+            if 'PaperWithCode' in platforms:
+                paperInGeneral_md += '''
+## Paper with Code
+| ID| Paper Title | Publication Year |
+| -------- | -------- | -------- |
+'''
+                pwc = PaperWithCodeQuery.query(query_input, items_per_page=num_papers)
+                for i in range(num_papers_preview):
+                    id = random_ids[i]
+                    title = str(pwc[id]['title']).replace('\n', ' ')
+                    publication_year = str(pwc[id]['published']).replace('\n', ' ')
+                    paperInGeneral_md += f'''|{i + 1}|{title}|{publication_year}|\n'''
+
+            paperInGeneral.markdown(paperInGeneral_md)

widgets/sidebar.py

@@ -0,0 +1,56 @@
+import streamlit as st
+from .utils import PACKAGE_ROOT
+
+def render_sidebar():
+    path = f'{PACKAGE_ROOT}/static/tum.png'
+    print(path)
+    sidebar_markdown = f'''
+
+    <center>
+    <img src="https://raw.githubusercontent.com/leoxiang66/streamlit-tutorial/IDP/widgets/static/tum.png" alt="TUM" width="150"/>
+
+    <h1>
+    Literature Research Tool 
+    </h1>
+
+
+    <code>
+    v1.0.0
+    </code>
+
+
+    </center>
+
+
+    <center>
+    <a href="https://github.com/Mondkuchen/idp_LiteratureResearch_Tool"><img src = "https://cdn-icons-png.flaticon.com/512/733/733609.png" width="23"></img></a>  <a href="mailto:xiang.tao@outlook.de"><img src="https://cdn-icons-png.flaticon.com/512/646/646094.png" alt="email" width = "27" ></a>
+    </center>
+
+    ---
+
+    ## Choose the Paper Search Platforms'''
+    st.sidebar.markdown(sidebar_markdown,unsafe_allow_html=True)
+    # elvsier = st.sidebar.checkbox('Elvsier',value=True)
+    # IEEE = st.sidebar.checkbox('IEEE',value=False)
+    # google = st.sidebar.checkbox('Google Scholar')
+    platforms = st.sidebar.multiselect('Platforms',options=
+    [
+        # 'Elvsier',
+        'IEEE',
+        # 'Google Scholar',
+        'Arxiv',
+        'PaperWithCode'
+    ], default=[
+        # 'Elvsier',
+        'IEEE',
+        # 'Google Scholar',
+        'Arxiv',
+        'PaperWithCode'
+    ])
+
+
+
+    st.sidebar.markdown('## Choose the number of papers to display')
+    number_papers=st.sidebar.slider('number', 0, 500, 25, 10)
+    
+    return platforms, number_papers

widgets/utils.py

@@ -0,0 +1,2 @@
+from pathlib import Path
+PACKAGE_ROOT = str(Path(__package__).absolute())