diff --git a/app.py b/app.py index cb8354b..1ee908a 100644 --- a/app.py +++ b/app.py @@ -9,12 +9,13 @@ from werkzeug.utils import secure_filename from PIL import Image import io +import os import google.generativeai as genai # WARNING: Do not share code with you API key hard coded in it. # Get your Gemini API key from: https://aistudio.google.com/app/apikey -GOOGLE_API_KEY="" +GOOGLE_API_KEY=os.environ.get('GOOGLE_API_KEY') genai.configure(api_key=GOOGLE_API_KEY) # The rate limits are low on this model, so you might need to switch to `gemini-pro` diff --git a/requirements.txt b/requirements.txt index 439f7e8..13a83c7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -27,10 +27,11 @@ pyasn1_modules==0.4.0 pydantic==2.7.1 pydantic_core==2.18.2 pyparsing==3.1.2 +python-dotenv==1.0.1 requests==2.31.0 rsa==4.9 tqdm==4.66.4 typing_extensions==4.11.0 uritemplate==4.1.1 urllib3==2.2.1 -Werkzeug==3.0.3 \ No newline at end of file +Werkzeug==3.0.3