added functionality to obtain the openai_key from environment variables, and also fixed some minor bugs that caused certain variables to not be properly retrieved and result in errors.

Author: JessyTsui

README.md

@@ -223,7 +223,7 @@ python3 app.py
 
 访问 http://127.0.0.1:5000/ 后，您将看到主页。在主页上，您可以点击不同的链接来调用各种服务。您可以通过修改链接中的参数值来实现不同的效果。有关参数详细信息，请参阅上一步骤中的详细介绍
 
-![image-20230331042557655](/Users/jessytsui/Library/Application Support/typora-user-images/image-20230331042557655.png)
+![flask主界面](./images/flask_web_home.png)
 
 
 

app.py

@@ -29,9 +29,9 @@ def home():
     reviewer_url = url_for("reviewer", _external=True, paper_path="", file_format="txt",
                            research_fields="computer science, artificial intelligence and reinforcement learning",
                            language="en")
-
     return f'''
-    <h1>Flask 版本的优势</h1>
+    <h1>ChatPaper，Flask版本的优势</h1>
+    <p>GitHub 项目地址：<a href="https://github.com/kaixindelele/ChatPaper" target="_blank">https://github.com/kaixindelele/ChatPaper</a></p> 
     <p>将原始的 Python 脚本改为使用 Flask 构建的 Web 服务具有以下优点：</p>
     <ul>
         <li><strong>易用性</strong>：通过简单的 HTTP 请求，用户可以轻松访问和使用各个功能，无需在本地安装 Python 或其他依赖。</li>

chat_arxiv.py

@@ -318,9 +318,13 @@ def __init__(self, key_word, query,
         self.config = configparser.ConfigParser()
         # 读取配置文件
         self.config.read('apikey.ini')
+        OPENAI_KEY = os.environ.get("OPENAI_KEY", "")
         # 获取某个键对应的值        
         self.chat_api_list = self.config.get('OpenAI', 'OPENAI_API_KEYS')[1:-1].replace('\'', '').split(',')
-        self.chat_api_list = [api.strip() for api in self.chat_api_list if len(api) > 5]
+        self.chat_api_list.append(OPENAI_KEY)
+
+        # prevent short strings from being incorrectly used as API keys.
+        self.chat_api_list = [api.strip() for api in self.chat_api_list if len(api) > 20]
         self.cur_api = 0
         self.file_format = args.file_format
         if args.save_image:
@@ -458,7 +462,9 @@ def summary_with_chat(self, paper_list):
 
             htmls.append('## Paper:' + str(paper_index + 1))
             htmls.append('\n\n\n')
-            htmls.append(chat_summary_text)
+            chat_summary_text = ""
+            if "chat_summary_text" in locals():
+                htmls.append(chat_summary_text)
 
             # 第二步总结方法：
             # TODO，由于有些文章的方法章节名是算法名，所以简单的通过关键词来筛选，很难获取，后面需要用其他的方案去优化。
@@ -487,7 +493,10 @@ def summary_with_chat(self, paper_list):
                         offset = int(str(e)[current_tokens_index:current_tokens_index + 4])
                         method_prompt_token = offset + 800 + 150
                         chat_method_text = self.chat_method(text=text, method_prompt_token=method_prompt_token)
-                htmls.append(chat_method_text)
+                chat_method_text = ""
+                if "chat_method_text" in locals():
+                    htmls.append(chat_method_text)
+                # htmls.append(chat_method_text)
             else:
                 chat_method_text = ''
             htmls.append("\n" * 4)
@@ -521,16 +530,16 @@ def summary_with_chat(self, paper_list):
                     conclusion_prompt_token = offset + 800 + 150
                     chat_conclusion_text = self.chat_conclusion(text=text,
                                                                 conclusion_prompt_token=conclusion_prompt_token)
-            htmls.append(chat_conclusion_text)
+            chat_conclusion_text = ""
+            if "chat_conclusion_text" in locals():
+                htmls.append(chat_conclusion_text)
             htmls.append("\n" * 4)
 
             # # 整合成一个文件，打包保存下来。
             date_str = str(datetime.datetime.now())[:13].replace(' ', '-')
-            try:
-                export_path = os.path.join(self.root_path, 'export')
+            export_path = os.path.join(self.root_path, 'export')
+            if not os.path.exists(export_path):
                 os.makedirs(export_path)
-            except:
-                pass
             mode = 'w' if paper_index == 0 else 'a'
             file_name = os.path.join(export_path,
                                      date_str + '-' + self.validateTitle(self.query) + "." + self.file_format)

chat_paper.py

@@ -55,9 +55,13 @@ def __init__(self, key_word, query, filter_keys,
         self.config = configparser.ConfigParser()
         # 读取配置文件
         self.config.read('apikey.ini')
-        # 获取某个键对应的值        
+        OPENAI_KEY = os.environ.get("OPENAI_KEY", "")
+        # 获取某个键对应的值
         self.chat_api_list = self.config.get('OpenAI', 'OPENAI_API_KEYS')[1:-1].replace('\'', '').split(',')
-        self.chat_api_list = [api.strip() for api in self.chat_api_list if len(api) > 5]
+        self.chat_api_list.append(OPENAI_KEY)
+
+        # prevent short strings from being incorrectly used as API keys.
+        self.chat_api_list = [api.strip() for api in self.chat_api_list if len(api) > 20]
         self.cur_api = 0
         self.file_format = args.file_format
         if args.save_image:
@@ -280,11 +284,9 @@ def summary_with_chat(self, paper_list):
 
             # # 整合成一个文件，打包保存下来。
             date_str = str(datetime.datetime.now())[:13].replace(' ', '-')
-            try:
-                export_path = os.path.join(self.root_path, 'export')
+            export_path = os.path.join(self.root_path, 'export')
+            if not os.path.exists(export_path):
                 os.makedirs(export_path)
-            except:
-                pass
             mode = 'w' if paper_index == 0 else 'a'
             file_name = os.path.join(export_path,
                                      date_str + '-' + self.validateTitle(paper.title[:80]) + "." + self.file_format)

chat_response.py

@@ -35,9 +35,13 @@ def __init__(self, args=None):
         self.config = configparser.ConfigParser()
         # 读取配置文件
         self.config.read('apikey.ini')
-        # 获取某个键对应的值        
+        OPENAI_KEY = os.environ.get("OPENAI_KEY", "")
+        # 获取某个键对应的值
         self.chat_api_list = self.config.get('OpenAI', 'OPENAI_API_KEYS')[1:-1].replace('\'', '').split(',')
-        self.chat_api_list = [api.strip() for api in self.chat_api_list if len(api) > 5]
+        self.chat_api_list.append(OPENAI_KEY)
+
+        # prevent short strings from being incorrectly used as API keys.
+        self.chat_api_list = [api.strip() for api in self.chat_api_list if len(api) > 20]
         self.cur_api = 0
         self.file_format = args.file_format
         self.max_token_num = 4096
@@ -54,11 +58,9 @@ def response_by_chatgpt(self, comment_path):
 
         # 将审稿意见保存起来
         date_str = str(datetime.datetime.now())[:13].replace(' ', '-')
-        try:
-            export_path = os.path.join('./', 'response_file')
+        export_path = os.path.join('./', 'response_file')
+        if not os.path.exists(export_path):
             os.makedirs(export_path)
-        except:
-            pass
         file_name = os.path.join(export_path, date_str + '-Response.' + self.file_format)
         self.export_to_markdown("\n".join(htmls), file_name=file_name)
         htmls = []

chat_reviewer.py

@@ -33,9 +33,13 @@ def __init__(self, args=None):
         self.config = configparser.ConfigParser()
         # 读取配置文件
         self.config.read('apikey.ini')
-        # 获取某个键对应的值        
+        OPENAI_KEY = os.environ.get("OPENAI_KEY", "")
+        # 获取某个键对应的值
         self.chat_api_list = self.config.get('OpenAI', 'OPENAI_API_KEYS')[1:-1].replace('\'', '').split(',')
-        self.chat_api_list = [api.strip() for api in self.chat_api_list if len(api) > 5]
+        self.chat_api_list.append(OPENAI_KEY)
+
+        # prevent short strings from being incorrectly used as API keys.
+        self.chat_api_list = [api.strip() for api in self.chat_api_list if len(api) > 20]
         self.cur_api = 0
         self.file_format = args.file_format
         self.max_token_num = 4096
@@ -72,11 +76,9 @@ def review_by_chatgpt(self, paper_list):
 
             # 将审稿意见保存起来
             date_str = str(datetime.datetime.now())[:13].replace(' ', '-')
-            try:
-                export_path = os.path.join('./', 'output_file')
+            export_path = os.path.join(self.root_path, 'export')
+            if not os.path.exists(export_path):
                 os.makedirs(export_path)
-            except:
-                pass
             mode = 'w' if paper_index == 0 else 'a'
             file_name = os.path.join(export_path,
                                      date_str + '-' + self.validateTitle(paper.title) + "." + self.file_format)

readme_en.md

@@ -128,7 +128,7 @@ After visiting http://127.0.0.1:5000/, you will see the homepage. On the homepag
 
 
 
-![image-20230331042557655](/Users/jessytsui/Library/Application Support/typora-user-images/image-20230331042557655.png)
+![flask_home_page](./images/flask_web_home.png)