move key to .env

2024-10-22 03:58:34 -05:00 · 2024-10-22 03:58:34 -05:00 · 45071812c3
commit 45071812c3
parent 083f67763b
2 changed files with 5 additions and 1 deletions
--- a/epb_site/epb_site/.env
+++ b/epb_site/epb_site/.env
@ -0,0 +1 @@
+GEMINI_API_KEY=AIzaSyAYt9n5dF2ZwOVYTbPV4_T5S0La2u5i1LU
--- a/epb_site/epb_site/gemini.py
+++ b/epb_site/epb_site/gemini.py
@ -1,7 +1,10 @@
 import os
 import google.generativeai as genai
+from dotenv import load_dotenv

-genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+load_dotenv()
+
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))

 # Create the model
 generation_config = {
				`@ -0,0 +1 @@`
				`GEMINI_API_KEY=AIzaSyAYt9n5dF2ZwOVYTbPV4_T5S0La2u5i1LU`