Dracomoth commited on
Commit
9a4251f
·
1 Parent(s): 149bbb6

Update envConfig.py

Browse files
Files changed (1) hide show
  1. envConfig.py +13 -1
envConfig.py CHANGED
@@ -1,7 +1,10 @@
1
  import os
2
  import gradio as gr
 
 
3
  from dotenv import load_dotenv
4
 
 
5
  #******************************************************************************************************************************************************************
6
 
7
  class envConfiguration:
@@ -16,6 +19,12 @@ class envConfiguration:
16
  os.environ["PINECONE_KEY"] = pinecone_key
17
  os.environ["PINECONE_ENV"] = pinecone_env
18
  os.environ["PINECONE_IND"] = pinecone_index
 
 
 
 
 
 
19
 
20
  def openai_key(self):
21
  return os.environ.get("OPEN_AI_KEY")
@@ -33,4 +42,7 @@ class envConfiguration:
33
  return os.environ.get("PINECONE_ENV")
34
 
35
  def pinecone_index(self):
36
- return os.environ.get("PINECONE_IND")
 
 
 
 
1
  import os
2
  import gradio as gr
3
+ import tempfile
4
+
5
  from dotenv import load_dotenv
6
 
7
+
8
  #******************************************************************************************************************************************************************
9
 
10
  class envConfiguration:
 
19
  os.environ["PINECONE_KEY"] = pinecone_key
20
  os.environ["PINECONE_ENV"] = pinecone_env
21
  os.environ["PINECONE_IND"] = pinecone_index
22
+ os.environ["OUTPUT_DIR"] = os.path.join(tempfile.gettempdir(),
23
+ 'rkk-document-gpt',
24
+ 'output'
25
+ )
26
+
27
+ os.makedirs(os.environ.get("OUTPUT_DIR"), exist_ok = True)
28
 
29
  def openai_key(self):
30
  return os.environ.get("OPEN_AI_KEY")
 
42
  return os.environ.get("PINECONE_ENV")
43
 
44
  def pinecone_index(self):
45
+ return os.environ.get("PINECONE_IND")
46
+
47
+ def output_dir(self):
48
+ return os.environ.get("OUTPUT_DIR")