Update functions.py

functions.py

@@ -9,7 +9,16 @@ import pandas as pd
 import json
 
 current_dir = os.getcwd()
-api_key = os.getenv("OPENAI_API_KEY")
+api_key_file = "api_key.json"
+
+def load_api_key():
+    if not 'OPENAI_API_KEY' in os.environ:
+        with open(api_key_file, 'r') as file:
+            api_key = json.load(file)['api_key']
+    else:
+        api_key = os.getenv['OPENAI_API_KEY']
+    
+    return api_key
     
 def load_dictionary(json_path):
     with open(json_path, 'r', encoding='utf-8') as file:
@@ -55,7 +64,7 @@ def load_embeddings(file_path="embeddings/embeddings.xlsx"):
         "model_name": model_name,
         }
 
-def generate_answer_with_references(query, data):
+def generate_answer_with_references(query, data, api_key):
     embeddings = data["embeddings"]
     segment_contents = data["segment_contents"]
     model_name = data["model_name"]
@@ -93,7 +102,8 @@ def generate_answer_with_references(query, data):
     return resposta, total_tokens, prompt_tokens, top_segments, top_file_names, top_similarities, prompt_template
 
 def rag_response(query, data, detailed_response):
-    resposta, total_tokens, prompt_tokens, top_segments, top_file_names, top_similarities, prompt_template = generate_answer_with_references(query, data)
+    api_key = load_api_key()
+    resposta, total_tokens, prompt_tokens, top_segments, top_file_names, top_similarities, prompt_template = generate_answer_with_references(query, data, api_key)
     file_names = [x[0] for x in top_file_names]
     file_links = {x[0]: x[1] for x in top_file_names}