Basic AI functionality

1 files changed, 8 insertions, 3 deletions

diff --git a/scripts/database/mongo.py b/scripts/database/mongo.py
index 3e97b0b..eaa2d97 100644
--- a/scripts/database/mongo.py
+++ b/scripts/database/mongo.py
@@ -3,7 +3,12 @@ from pymongo.server_api import ServerApi
 import Constants
 import certifi
 
-uri = Constants.MONGO_CONNECTION_STRING
+uri = Constants.MONGO_CONNECTION_STRING if Constants.MODE == "production" else Constants.MONGO_CONNECTION_STRING_DEV
+database_str = "index-cooking" if Constants.MODE == "production" else "recipedb" 
 
-client = MongoClient(uri, server_api=ServerApi('1'), tlsCAFile=certifi.where())
-mongo_database = client["index-cooking"]
\ No newline at end of file
+if Constants.MODE == "production":
+    client = MongoClient(uri, server_api=ServerApi('1'), tlsCAFile=certifi.where())
+else:
+    client = MongoClient(uri)
+
+mongo_database = client["index-cooking"]