Upload app.py

add review part

app.py

@@ -469,6 +469,9 @@ class Reader:
             chat_conclusion_text = self.chat_conclusion(text=text, key=str(key))
             htmls.append(chat_conclusion_text)
             htmls.append("\n")
+            chat_review_text = self.chat_review(text=text, key=str(key))
+            htmls.append(chat_review_text)
+            htmls.append("\n")
             md_text = "\n".join(htmls)
             
             return markdown.markdown(md_text)
@@ -579,8 +582,58 @@ class Reader:
         for choice in response.choices:
             result += choice.message.content
         print("summary_result:\n", result)
-        return result        
-            
+        return result
+
+    @tenacity.retry(wait=tenacity.wait_exponential(multiplier=1, min=4, max=10),
+                    stop=tenacity.stop_after_attempt(5),
+                    reraise=True)
+    def chat_review(self, text):
+        openai.api_key = self.chat_api_list[self.cur_api]
+        self.cur_api += 1
+        self.cur_api = 0 if self.cur_api >= len(self.chat_api_list) - 1 else self.cur_api
+
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=[
+                {"role": "system",
+                 "content": "You are a researcher in the field of [" + self.key_word + "] who is good at reviewing papers using concise statements"},
+                # chatgpt 角色
+                {"role": "assistant",
+                 "content": "This is the title, author, link, abstract, introduction, method, and experiments of an English document. I need your help to read and review the following questions: " + text},
+                # 背景知识
+                {"role": "user", "content": """                 
+                 1. summarize according to the following four points.Be sure to use English answers (proper nouns need to be marked in English)
+                    - (1):What is the research background of this article?
+                    - (2):What are the past methods? What are the problems with them? Is the approach well motivated?
+                    - (3):What is the research methodology proposed in this paper?
+                    - (4):On what task and what performance is achieved by the methods in this paper? Can the performance support their goals?
+                 Follow the format of the output that follows:                    
+                 2. Strengths: \n\n
+                    - (1):Background;\n 
+                    - (2):Main challenges and Motivations;\n 
+                    - (3):The detail of methods\n  
+                    - (4):Results.\n\n     
+                 3. Weakness: \n\n
+                    - (1):Motivation;\n 
+                    - (2):Methods;\n 
+                    - (3):Novelty\n  
+                    - (4):Results.\n\n
+                 4. Other questions: \n\n
+                    - (1):\n 
+                    - (2):\n 
+                    - (3):\n  
+                    - (4):\n\n       
+                 Be sure to use English answers, statements as concise and academic as possible, do not have too much repetitive information, numerical values using the original numbers, be sure to strictly follow the format, the corresponding content output to xxx, in accordance with \n line feed.                 
+                 """},
+            ]
+        )
+        result = ''
+        for choice in response.choices:
+            result += choice.message.content
+        print("summary_result:\n", result)
+        return result
+
+
     def export_to_markdown(self, text, file_name, mode='w'):
         # 使用markdown模块的convert方法，将文本转换为html格式
         # html = markdown.markdown(text)
@@ -611,13 +664,13 @@ def upload_pdf(key, text, file):
         return sum_info
 
 # 标题
-title = "ChatPaper"
+title = "ChatPaperPlus"
 # 描述
 description = '''<div align='center'>
 
 Use ChatGPT to summary the papers.
 
-Star our Github [ChatPaper](https://github.com/kaixindelele/ChatPaper)
+Based on [ChatPaper](https://github.com/kaixindelele/ChatPaper), and add review function.
 
 </div>
 '''