keys as env variables

gradio_app/backend/ChatGptInteractor.py

@@ -1,12 +1,20 @@
 import logging
+import os
 import time
 
 import tiktoken
 import openai
 
 
-with open('data/openaikey.txt') as f:
-    OPENAI_KEY = f.read().strip()
+OPENAI_KEY = None
+key_file = 'data/openaikey.txt'
+if os.path.exists(key_file):
+    with open(key_file) as f:
+        OPENAI_KEY = f.read().strip()
+
+if OPENAI_KEY is None:
+    OPENAI_KEY = os.getenv('OPENAI_KEY')
+
 openai.api_key = OPENAI_KEY
 
 

gradio_app/backend/HuggingfaceGenerator.py

@@ -1,10 +1,19 @@
 import logging
+import os
 
 from huggingface_hub import InferenceClient
 from transformers import AutoTokenizer
 
-with open('data/hftoken.txt') as f:
-    HF_TOKEN = f.read().strip()
+
+HF_TOKEN = None
+key_file = 'data/hftoken.txt'
+if os.path.exists(key_file):
+    with open(key_file) as f:
+        HF_TOKEN = f.read().strip()
+
+if HF_TOKEN is None:
+    HF_TOKEN = os.getenv('HF_TOKEN')
+
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)