From f543e217af1905d97363c5da3b31debae1953ef5 Mon Sep 17 00:00:00 2001
From: Max <Hunt>
Date: Tue, 5 Jan 2021 15:23:01 +0000
Subject: [PATCH] Update

---
 dnup_virtual_sensor/virtual_soil_sensor.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/dnup_virtual_sensor/virtual_soil_sensor.py b/dnup_virtual_sensor/virtual_soil_sensor.py
index 3e58fe3..7aac165 100755
--- a/dnup_virtual_sensor/virtual_soil_sensor.py
+++ b/dnup_virtual_sensor/virtual_soil_sensor.py
@@ -10,15 +10,17 @@ from tensorflow import keras
 class Firebase:
     def __init__(self):
         self.creds = credentials.Certificate(
-            'icl-iot-weather-firebase-adminsdk.json')
+            'secrets/icl-iot-weather-firebase-adminsdk.json')
         firebase_admin.initialize_app(self.creds)
         self.db = firestore.client()
+        logging.debug('Initialized firebase instance')
 
     def pull_from_db(self, orderby=u'timestamp'):
         doc_ref = self.db.collection('weather_data')
         query = doc_ref.order_by(orderby,
                                  direction=firestore.Query.DESCENDING).limit(1)
         doc = query.stream()
+        logging.debug('Got doc file from firestore')
         return doc
 
     def convert_to_float(self, data):
@@ -38,10 +40,12 @@ class VirtualProbe:
     def __init__(self):
         self.firebase = Firebase()
         self.model = keras.models.load_model('virtual_probe.model')
+        logging.debug('Loaded ML model')
 
     def predict_soil_temp(self):
         feature = self.firebase.get_feature()
         predicted_temp = self.model.predict([feature])
+        logging.debug(f'Predicted watering volume: {predicted_temp}')
         return predicted_temp