Spaces:
Sleeping
Sleeping
Fix API key loading and update Gemini model name
Browse files- .env.example +9 -0
- app/services/chatbot_service.py +2 -2
- list_models.py +5 -1
- 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
|
| 18 |
-
self.model = genai.GenerativeModel('gemini-
|
| 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 |
-
|
|
|
|
|
|
|
| 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()
|