add paper_preview.py

apps/paper_preview.py

@@ -0,0 +1,61 @@
+import traceback
+
+import gradio as gr
+
+from apps.components import chat_accordion
+
+IDEA_TITLE = "论文速览"
+
+prompt_tmpl = """下面是一篇论文的标题和摘要，请从以下几个方面进行总结：
+
+1. 介绍本文的主要工作
+2. 本文工作的主要亮点
+3. 核心关键词（最多5个技术，能够代表本文核心技术，格式：`英文` (`中文`) ）
+4. 从实用性、创新性和推荐度进行打分（各项满分5分）。
+
+===
+标题：{title}
+===
+摘要：{abstract}
+===
+
+注意：生成内容要简练，语言的组织要通顺、容易阅读和理解，并能够快速获取信息。
+"""
+
+
+def paper_preview_demo(client):
+    def preview(title, abstract, temperature, top_p):
+        if not title or not abstract:
+            return None
+        content = prompt_tmpl.format(title=title, abstract=abstract)
+        try:
+            stream = client.simple_chat(
+                content,
+                [],
+                temperature=temperature,
+                top_p=top_p,
+            )
+            for resp, _ in stream:
+                pass
+            return resp
+        except Exception:
+            return traceback.format_exc()
+
+    with gr.Row():
+        with gr.Column():
+            title = gr.Textbox(label="论文标题")
+            abstract = gr.Textbox(label="摘要", lines=5)
+            with gr.Row():
+                with gr.Column():
+                    submit = gr.Button("速览", variant="primary")
+                with gr.Column():
+                    clear = gr.Button("清空")
+            temperature, top_p = chat_accordion()
+
+        with gr.Column():
+            outputs = gr.Textbox(label="速览内容", lines=5)
+
+        submit.click(
+            preview, inputs=[title, abstract, temperature, top_p], outputs=outputs
+        )
+        clear.click(lambda x: (None, None), inputs=None, outputs=[title, abstract])