Update envConfig.py

envConfig.py

@@ -1,10 +1,8 @@
 import os
-import gradio as gr
 import tempfile
 
 from dotenv import load_dotenv
 
-
 #******************************************************************************************************************************************************************
 
 class envConfiguration:
@@ -19,6 +17,14 @@ class envConfiguration:
         os.environ["PINECONE_KEY"] = pinecone_key
         os.environ["PINECONE_ENV"] = pinecone_env
         os.environ["PINECONE_IND"] = pinecone_index
+        os.environ["INPUT_DIR"] = os.path.join(tempfile.gettempdir(),
+                                                'rkk-document-gpt',
+                                                'input'
+                                                )
+        os.environ["TEMPLATE_DIR"] = os.path.join(tempfile.gettempdir(),
+                                                'rkk-document-gpt',
+                                                'template'
+                                                )
         os.environ["OUTPUT_DIR"] = os.path.join(tempfile.gettempdir(),
                                                 'rkk-document-gpt',
                                                 'output'
@@ -44,5 +50,11 @@ class envConfiguration:
     def pinecone_index(self):
         return os.environ.get("PINECONE_IND")
 
+    def input_dir(self):
+        return os.environ.get("INPUT_DIR")    
+
+    def template_dir(self):
+        return os.environ.get("TEMPLATE_DIR")
+
     def output_dir(self):
         return os.environ.get("OUTPUT_DIR")