Update

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