prettify

app.py

@@ -23,9 +23,11 @@ def get_dblp_bibref(title):
         url = soup.select_one('url').text + '.bib'
         response = requests.get(url)
 
-        return response.text
+        paper_link = soup.select_one('url').text + '.html'
+
+        return response.text, paper_link
     except Exception as e:
-        return f'Error during get bibref from DBLP: {e}'
+        return f'Error during get bibref from DBLP: {e}', None
 
 # set pprint width
 pp = pprint.PrettyPrinter(width=128)
@@ -110,9 +112,9 @@ def summarize_paper_info(paper_info):
         author_str += f"[{author['name']}](https://www.semanticscholar.org/author/{author['authorId']}), "
     author_str = author_str[:-2]
 
-    info_str += f"**Authors**: {author_str}\n\n"
+    info_str += f"**Authors**:\n\n{author_str}\n\n"
 
-    info_str += f"**Abstract**: {paper_info['abstract']}\n\n"
+    info_str += f"\n\n> **Abstract**: {paper_info['abstract']}\n\n"
     info_str += f"**Citation Count**: {paper_info['citationCount']}\n\n"
     return info_str
 
@@ -124,14 +126,36 @@ def get_output(title):
         citation_str = get_md_citation(paper_info)
     else:
         citation_str = "No paper found with that title."
+    
+    bibtex, dblp_link = get_dblp_bibref(paper_info['title'])
 
     citation_str = f"""
-**Citation**:
-\n\n{citation_str}\n\n
-**Markdown source code**:\n\n```markdown\n{citation_str}\n```\n\n
-**BibTex**\n\n```bibtex\n{get_dblp_bibref(paper_info['title'])}\n```\n\n
+```text
+{paper_info['title']}
+```
+
+{citation_str}
+
 ---
-\n\n{summarize_paper_info(paper_info)}\n\n
+
+**Markdown source code**
+
+```markdown
+{citation_str}
+```
+
+
+**BibTex**
+
+```bibtex
+{bibtex}
+```
+
+{summarize_paper_info(paper_info)}
+
+---
+
+🔗 [[Open in Semantic Scholar]](https://www.semanticscholar.org/paper/{paper_info['paperId']}) | [[DBLP Page]]({dblp_link})
 """
 
     print(citation_str)
@@ -139,12 +163,19 @@ def get_output(title):
     return citation_str
 
 def main():
-    iface = gr.Interface(fn=get_output, 
-                         inputs=gr.components.Textbox(lines=2, label="Please input the title of the paper"), 
-                         outputs="markdown", 
-                         allow_flagging='never',
-                         title="Citation Tool",
-                         )
+    iface = gr.Interface(
+        fn=get_output, 
+        inputs=gr.components.Textbox(
+            lines=1, 
+            label="Please input the title of the paper to get its citation.", 
+            placeholder="Your title here",
+            autofocus=True,
+            ), 
+        outputs="markdown", 
+        allow_flagging='never',
+        title="Citation Tool",
+        description="### Search paper title from [Semantic Scholar](https://www.semanticscholar.org/) and [DBLP](http://dblp.org/), and get structured citation.",
+        )
     iface.launch()
 
 if __name__=="__main__":