sc_ma commited on
Commit
687c532
1 Parent(s): 515558b

Use ENV for API key.

Browse files
Files changed (2) hide show
  1. README.md +0 -4
  2. utils/gpt_interaction.py +2 -4
README.md CHANGED
@@ -1,7 +1,3 @@
1
- ---
2
- app_file: app.py
3
- ---
4
-
5
  # Auto-Draft: 论文结构辅助工具
6
 
7
  这个项目旨在为学术论文写作创建一个全自动辅助工具:搜索相关文献,拟定论文结构,生成论文模板。希望这个工具在未来可以让人们摆脱繁琐的论文写作,将工作的重心放在纯粹的科研上。
 
 
 
 
 
1
  # Auto-Draft: 论文结构辅助工具
2
 
3
  这个项目旨在为学术论文写作创建一个全自动辅助工具:搜索相关文献,拟定论文结构,生成论文模板。希望这个工具在未来可以让人们摆脱繁琐的论文写作,将工作的重心放在纯粹的科研上。
utils/gpt_interaction.py CHANGED
@@ -1,13 +1,11 @@
1
  import openai
2
  import re
 
3
  import json
4
  import logging
5
  log = logging.getLogger(__name__)
6
 
7
- f = open("api_key.txt", "r")
8
- openai.api_key = f.read()
9
- f.close()
10
-
11
 
12
  def extract_responses(assistant_message):
13
  # pattern = re.compile(r"f\.write\(r'{1,3}(.*?)'{0,3}\){0,1}$", re.DOTALL)
 
1
  import openai
2
  import re
3
+ import os
4
  import json
5
  import logging
6
  log = logging.getLogger(__name__)
7
 
8
+ openai.api_key = os.environ['OPENAI_API_KEY']
 
 
 
9
 
10
  def extract_responses(assistant_message):
11
  # pattern = re.compile(r"f\.write\(r'{1,3}(.*?)'{0,3}\){0,1}$", re.DOTALL)