stupid memain

1 files changed, 5 insertions, 2 deletions

diff --git a/backend/main.py b/backend/main.py
index 6681d58..b576a3a 100644
--- a/backend/main.py
+++ b/backend/main.py
@@ -2,6 +2,10 @@ from google import genai
 from google.genai import types
 import flask
 import flask_cors
+import os
+import dotenv
+
+dotenv.load_dotenv(".env")
 
 app = flask.Flask(__name__)
 flask_cors.CORS(app)
@@ -9,8 +13,7 @@ flask_cors.CORS(app)
 UPLOAD_FOLDER = "uploads"
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
 
-client = genai.Client(api_key="AIzaSyA-mA7v4xlbEAz80uKf3Dywpbeh9ZfpLMs")
-
+client = genai.Client(api_key=os.getenv("API_KEY"))
 
 @app.route("/yeah", methods=["POST"])
 def yeah_flask():