Add app.

README.md

@@ -1,5 +1,5 @@
 ---
-title: ChatReviewer
+title: ChatAssistant
 emoji: 💩
 colorFrom: red
 colorTo: pink

app.py

@@ -15,9 +15,10 @@ import gradio
 # 定义Reviewer类
 class Reviewer:
     # 初始化方法，设置属性
-    def __init__(self, api, review_format, paper_pdf, language):
+    def __init__(self, api, research_field, question, paper_pdf, language):
         self.api = api
-        self.review_format = review_format
+        self.research_field = research_field
+        self.question = question
 
         self.language = language
         self.paper_pdf = paper_pdf
@@ -40,9 +41,10 @@ class Reviewer:
         review_prompt_token = 1000        
         text_token = len(self.encoding.encode(text))
         input_text_index = int(len(text)*(self.max_token_num-review_prompt_token)/(text_token+1))
-        input_text = "This is the paper for your review:" + text[:input_text_index]
+        input_text = "This is the paper you are asked to read:" + text[:input_text_index]
+        input_text = input_text + "The question from your student is: " + self.question
         messages=[
-                {"role": "system", "content": "You are a professional reviewer. Now I will give you a paper. You need to give a complete review opinion according to the following requirements and format:"+ self.review_format +" Must be output in {}.".format(self.language)},
+                {"role": "system", "content": "You are a professional researcher in the field of "+self.research_field+". You are the mentor of a student who is new to this field. Now I will give you a paper. You need to help your student to read this paper by instructing him to read the important sections in this paper and answer his questions towards these sections. Please answer in {}.".format(self.language),
                 {"role": "user", "content": input_text},
             ]
                 
@@ -88,14 +90,14 @@ class Reviewer:
                     break
         return extracted_text
 
-def main(api, review_format, paper_pdf, language):  
+def main(api, research_field, question, paper_pdf, language):  
     start_time = time.time()
-    if not api or not review_format or not paper_pdf:
+    if not api or not research_field or not question or not paper_pdf:
         return "请输入完整内容！"
     # 判断PDF文件
     else:
         # 创建一个Reader对象
-        reviewer1 = Reviewer(api, review_format, paper_pdf, language)
+        reviewer1 = Reviewer(api, research_field, question, paper_pdf, language)
         # 开始判断是路径还是文件：   
         comments, total_token_used = reviewer1.review_by_chatgpt(paper_list=paper_pdf)
     time_used = time.time() - start_time
@@ -106,17 +108,13 @@ def main(api, review_format, paper_pdf, language):
 
 ########################################################################################################    
 # 标题
-title = "🤖ChatReviewer🤖"
+title = "ChatAssistant: ChatGPT论文阅读助手"
 # 描述
 
 description = '''<div align='left'>
-<img align='right' src='http://i.imgtg.com/2023/03/22/94PLN.png' width="270">
-<strong>ChatReviewer是一款基于ChatGPT-3.5的API开发的论文自动评审AI助手。</strong>其用途如下：
-⭐️对论文进行快速总结和评审，提高科研人员的文献阅读和理解的效率，紧跟研究前沿。
-⭐️对自己的论文进行评审，根据ChatReviewer生成的审稿意见进行查漏补缺，进一步提高自己的论文质量。
-⭐️辅助论文审稿，给出参考意见，提高审稿效率和质量。（🈲：禁止直接复制生成的评论用于任何论文审稿工作！）
-如果觉得很卡，可以点击右上角的Duplicate this Space，把ChatReviewer复制到你自己的Space中！
-本项目的[Github](https://github.com/nishiwen1214/ChatReviewer)，欢迎Star和Fork，也欢迎大佬赞助让本项目快速成长！💗（[获取Api Key](https://chatgpt.cn.obiscr.com/blog/posts/2023/How-to-get-api-key/)）
+<strong>ChatAssistant是一款基于ChatGPT-3.5的API开发的论文阅读助手。</strong>其用途如下：
+⭐️针对用户对论文的内容所提出的问题，给出相关的解答或学习建议。
+（[获取Api Key](https://chatgpt.cn.obiscr.com/blog/posts/2023/How-to-get-api-key/)）
 </div>
 '''
 
@@ -124,31 +122,13 @@ description = '''<div align='left'>
 inp = [gradio.inputs.Textbox(label="请输入你的API-key(sk开头的字符串)",
                           default="",
                           type='password'),
-    gradio.inputs.Textbox(lines=5,
-        label="请输入特定的评审要求和格式(否则为默认格式)",
-        default="""* Overall Review
-Please briefly summarize the main points and contributions of this paper.
-xxx
-* Paper Strength 
-Please provide a list of the strengths of this paper, including but not limited to: innovative and practical methodology, insightful empirical findings or in-depth theoretical analysis, 
-well-structured review of relevant literature, and any other factors that may make the paper valuable to readers. (Maximum length: 2,000 characters) 
-(1) xxx
-(2) xxx
-(3) xxx
-* Paper Weakness 
-Please provide a numbered list of your main concerns regarding this paper (so authors could respond to the concerns individually). 
-These may include, but are not limited to: inadequate implementation details for reproducing the study, limited evaluation and ablation studies for the proposed method, 
-correctness of the theoretical analysis or experimental results, lack of comparisons or discussions with widely-known baselines in the field, lack of clarity in exposition, 
-or any other factors that may impede the reader's understanding or benefit from the paper. Please kindly refrain from providing a general assessment of the paper's novelty without providing detailed explanations. (Maximum length: 2,000 characters) 
-(1) xxx
-(2) xxx
-(3) xxx
-* Questions To Authors And Suggestions For Rebuttal 
-Please provide a numbered list of specific and clear questions that pertain to the details of the proposed method, evaluation setting, or additional results that would aid in supporting the authors' claims. 
-The questions should be formulated in a manner that, after the authors have answered them during the rebuttal, it would enable a more thorough assessment of the paper's quality. (Maximum length: 2,000 characters)
-*Overall score (1-10)
-The paper is scored on a scale of 1-10, with 10 being the full mark, and 6 stands for borderline accept. Then give the reason for your rating.
-xxx"""
+    gradio.inputs.Textbox(lines=3,
+        label="请输入论文的研究方向（语言和输出语言一致）",
+        default="""eg. computer science, artificial intelligence and transfer learning"""
+    ),
+    gradio.inputs.Textbox(lines=3,
+        label="请输入你的问题（语言和输出语言一致）。请尽可能地在问题之后概括你想要得到的输出的回答方向。",
+        default="""eg. What are the main contributions of this article? Please summarize the technical details in your reply as well."""
     ),
     gradio.inputs.File(label="请上传论文PDF(必填)",type="bytes"),
     gradio.inputs.Radio(choices=["English", "Chinese"],
@@ -156,11 +136,11 @@ xxx"""
                         label="选择输出语言"),
 ]
 
-chat_reviewer_gui = gradio.Interface(fn=main,
+chat_assistant_gui = gradio.Interface(fn=main,
                                  inputs=inp,
-                                 outputs = [gradio.Textbox(lines=25, label="评审结果"), gradio.Textbox(lines=2, label="资源统计")],
+                                 outputs = [gradio.Textbox(lines=25, label="参考回答"), gradio.Textbox(lines=2, label="资源统计")],
                                  title=title,
                                  description=description)
 
 # Start server
-chat_reviewer_gui .launch(quiet=True, show_api=False)
+chat_assistant_gui.launch(quiet=True, show_api=False)