Update Openai/chat.py

Openai/chat.py

@@ -4,13 +4,11 @@ import os
 import pandas as pd
 import pickle as pkl
 
-API_KEY = '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'
-openai.api_key = API_KEY
-
-def generate_abstract(prompt):
+def generate_abstract(prompt, key = 'sk-ShGVEfW70vrzhBhgR5nET3BlbkFJi610yGFec2yKE5eispz'):
     
     # return prompt
     # return ' '.join([prompt] * 3)
+    openai.api_key = key
     res = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
         max_tokens=3000,
@@ -21,7 +19,7 @@ def generate_abstract(prompt):
        [
          {
           "role": "system",
-          "content": "You are expanding a given sentence into a `scientific biomedical abstract, and this abstract must include given sentence.",
+          "content": "You are expanding a given sentence into a scientific biomedical abstract, and this abstract must include given sentence.",
          },
         {
           "role": "user",
@@ -52,7 +50,6 @@ if __name__ == '__main__':
     # import time
     pre1_list = ['random']
     pre2_list = ['0.5', '0.3'] # '0.3' or '0.5'
-    openai.api_key = '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'
     for pre1 in pre1_list:
         for pre2 in pre2_list:
             print(pre1, pre2)
@@ -75,4 +72,4 @@ if __name__ == '__main__':
                     else:
                         ret_dict.update({k: {'in':v, 'out':''}})
                         with open(f'{pre1}_{pre2}_chat.json', 'w') as fl:
-                            json.dump(ret_dict, fl, indent=4)
+                            json.dump(ret_dict, fl, indent=4)