Upload app.py

app.py

@@ -412,18 +412,16 @@ class Reader:
             chat_summary_text = self.chat_summary(text=text, key=str(key))           
             htmls.append(chat_summary_text)
             
-            # # TODO 往md文档中插入论文里的像素最大的一张图片，这个方案可以弄的更加智能一些：
-            # first_image, ext = paper.get_image_path()
-            # if first_image is None or self.gitee_key == '':
-            #     pass
-            # else:
-            #     image_title = self.validateTitle(paper.title)
-            #     image_url = self.upload_gitee(image_path=first_image, image_name=image_title, ext=ext)
-            #     htmls.append("\n")
-            #     htmls.append("![Fig]("+image_url+")")
-            #     htmls.append("\n")
-            #
-
+            # TODO 往md文档中插入论文里的像素最大的一张图片，这个方案可以弄的更加智能一些：
+            first_image, ext = paper.get_image_path()
+            if first_image is None or self.gitee_key == '':
+                pass
+            else:                
+                image_title = self.validateTitle(paper.title)
+                image_url = self.upload_gitee(image_path=first_image, image_name=image_title, ext=ext)
+                htmls.append("\n")
+                htmls.append("![Fig]("+image_url+")")
+                htmls.append("\n")
             # 第二步总结方法：
             # TODO，由于有些文章的方法章节名是算法名，所以简单的通过关键词来筛选，很难获取，后面需要用其他的方案去优化。
             method_key = ''
@@ -468,28 +466,8 @@ class Reader:
             else:
                 text = summary_text
             text = text[:max_token]
-            # chat_conclusion_text = self.chat_conclusion(text=text, key=str(key))
-            # htmls.append(chat_conclusion_text)
-            # htmls.append("\n")
-            experiment_text = ''
-            experiment_key = ''
-            review_text = "<summary>" + chat_summary_text + "\n <Method summary>:\n" + chat_method_text
-            for parse_key in paper.section_text_dict.keys():
-                if 'Experiments' in parse_key.lower() or 'Results' in parse_key.lower():
-                    experiment_key = parse_key
-                    break
-            if experiment_key != '':
-                # conclusion
-                experiment_text += paper.section_text_dict[experiment_key]
-                max_token = 2500 * 4
-                text = review_text + "\n <Experiments>:\n" + experiment_text
-            else:
-                text = review_text
-            text = text[:max_token]
-
-
-            chat_review_text = self.chat_review(text=text, key=str(key))
-            htmls.append(chat_review_text)
+            chat_conclusion_text = self.chat_conclusion(text=text, key=str(key))
+            htmls.append(chat_conclusion_text)
             htmls.append("\n")
             md_text = "\n".join(htmls)
             
@@ -601,58 +579,8 @@ class Reader:
         for choice in response.choices:
             result += choice.message.content
         print("summary_result:\n", 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("review_result:\n", result)
-        return result
-
-
+        return result        
+            
     def export_to_markdown(self, text, file_name, mode='w'):
         # 使用markdown模块的convert方法，将文本转换为html格式
         # html = markdown.markdown(text)
@@ -683,13 +611,13 @@ def upload_pdf(key, text, file):
         return sum_info
 
 # 标题
-title = "ChatPaperPlus"
+title = "ChatPaper"
 # 描述
 description = '''<div align='center'>
 
-Use ChatGPT to summary and review the papers.
+Use ChatGPT to summary the papers.
 
-Based on [ChatPaper](https://github.com/kaixindelele/ChatPaper), and add review function.
+Star our Github [ChatPaper](https://github.com/kaixindelele/ChatPaper)
 
 </div>
 '''