yandri918 commited on
Commit
631230d
·
1 Parent(s): f23a9f0

Fix API key loading and update Gemini model name

Browse files
Files changed (4) hide show
  1. .env.example +9 -0
  2. app/services/chatbot_service.py +2 -2
  3. list_models.py +5 -1
  4. run.py +6 -0
.env.example CHANGED
@@ -1 +1,10 @@
 
1
  GEMINI_API_KEY=your_api_key_here
 
 
 
 
 
 
 
 
 
1
+ # Google Gemini API Key
2
  GEMINI_API_KEY=your_api_key_here
3
+
4
+ # Flask Configuration
5
+ FLASK_APP=run.py
6
+ FLASK_ENV=development
7
+ SECRET_KEY=dev_secret_key
8
+
9
+ # Database
10
+ # DATABASE_URL=postgresql://user:password@localhost:5432/agrisensa
app/services/chatbot_service.py CHANGED
@@ -14,8 +14,8 @@ class ChatbotService:
14
  self.init_error = None
15
  try:
16
  genai.configure(api_key=self.API_KEY)
17
- # Use a model explicitly found in the list_models output
18
- self.model = genai.GenerativeModel('gemini-2.0-flash-lite-preview')
19
  self.chat = self.model.start_chat(history=[])
20
 
21
  # System prompt to guide the AI
 
14
  self.init_error = None
15
  try:
16
  genai.configure(api_key=self.API_KEY)
17
+ # Use a stable model
18
+ self.model = genai.GenerativeModel('gemini-flash-latest')
19
  self.chat = self.model.start_chat(history=[])
20
 
21
  # System prompt to guide the AI
list_models.py CHANGED
@@ -1,6 +1,10 @@
1
  import google.generativeai as genai
 
 
2
 
3
- API_KEY = "AIzaSyBsO07f4MK3Sg7EPynrFmrriZgScmFBiQU"
 
 
4
  genai.configure(api_key=API_KEY)
5
 
6
  try:
 
1
  import google.generativeai as genai
2
+ import os
3
+ from dotenv import load_dotenv
4
 
5
+ load_dotenv()
6
+
7
+ API_KEY = os.getenv("GEMINI_API_KEY")
8
  genai.configure(api_key=API_KEY)
9
 
10
  try:
run.py CHANGED
@@ -1,3 +1,9 @@
 
 
 
 
 
 
1
  from app import create_app
2
 
3
  app = create_app()
 
1
+ from dotenv import load_dotenv
2
+ import os
3
+
4
+ # Load environment variables from .env file
5
+ load_dotenv()
6
+
7
  from app import create_app
8
 
9
  app = create_app()