add update

Vets-Who-Code · Nov 6, 2024 · d7b4a0d · d7b4a0d
1 parent 3bc4e23
commit d7b4a0d
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/app.py b/app.py
@@ -161,10 +161,14 @@
 # Load environment variables
 load_dotenv()
 
-# Configure OpenAI
-openai.api_key = os.getenv("OPENAI_API_KEY")
+openai.api_key = st.secrets["openai"]["OPENAI_API_KEY"]
+
+# Optionally, load the model name if needed
+model_name = st.secrets["openai"].get("model_name", "default_model_name")
+
+# Test the setup
 if not openai.api_key:
-    raise ValueError("OpenAI API key not found in .env file")
+    raise ValueError("OpenAI API key not found in Streamlit secrets.")
 
 def parse_mos_file(file_content: str) -> dict:
     """