feat: implement env variables setting for building dependency in Hugging Face

modules/db_modules.py

@@ -1,15 +1,17 @@
-
 import boto3
+import streamlit as st
+
 from boto3.dynamodb.conditions import Key
 from dotenv import dotenv_values
 
 config = dotenv_values(".env")
 
-AWS_ACCESS_KEY_ID = config.get('AWS_ACCESS_KEY_ID')
-AWS_SECRET_ACCESS_KEY = config.get('AWS_SECRET_ACCESS_KEY')
-
-# AWS_ACCESS_KEY_ID = os.environ['AWS_ACCESS_KEY_ID']
-# AWS_SECRET_ACCESS_KEY = os.environ['AWS_SECRET_ACCESS_KEY']
+if config:
+    AWS_ACCESS_KEY_ID = config.get('AWS_ACCESS_KEY_ID')
+    AWS_SECRET_ACCESS_KEY = config.get('AWS_SECRET_ACCESS_KEY')
+else:
+    AWS_ACCESS_KEY_ID = st.secrets['AWS_ACCESS_KEY_ID'] 
+    AWS_SECRET_ACCESS_KEY = st.secrets['AWS_SECRET_ACCESS_KEY']
 
 
 def get_db():

modules/gpt_modules.py

@@ -1,11 +1,17 @@
 import openai
+import streamlit as st
 
 from dotenv import dotenv_values
 
 config = dotenv_values(".env")
 
-openai.organization = config.get('OPENAI_ORGANIZATION')
-openai.api_key = config.get('OPENAI_API_KEY')
+if config:
+    openai.organization = config.get('OPENAI_ORGANIZATION')
+    openai.api_key = config.get('OPENAI_API_KEY')
+else:
+    openai.organization = st.secrets['OPENAI_ORGANIZATION']
+    openai.api_key = st.secrets['OPENAI_API_KEY']
+
 
 def gpt_call(prompt, role="user"):
     response = openai.ChatCompletion.create(

modules/whisper_modules.py

@@ -1,9 +1,11 @@
-from dotenv import dotenv_values
 import openai
 import os
+import random
+import streamlit as st
+
 from langchain.prompts import PromptTemplate
 from modules.gpt_modules import gpt_call
-import random
+from dotenv import dotenv_values
 
 """
 apt-get update
@@ -12,8 +14,13 @@ apt-get install ffmpeg
 
 config = dotenv_values(".env")
 
-openai.organization = config.get('OPENAI_ORGANIZATION')
-openai.api_key = config.get('OPENAI_API_KEY')
+if config:
+    openai.organization = config.get('OPENAI_ORGANIZATION')
+    openai.api_key = config.get('OPENAI_API_KEY')
+else:
+    openai.organization = st.secrets['OPENAI_ORGANIZATION'] #config.get('OPENAI_ORGANIZATION')
+    openai.api_key = st.secrets['OPENAI_API_KEY'] #config.get('OPENAI_API_KEY')
+
 
 def debate_in_sound(audio):
     os.rename(audio, audio + '.wav')