Update firestore_twilio_listener.py

firestore_twilio_listener.py

@@ -1,60 +1,83 @@
-from twilio.rest import Client
-import firebase_admin
-from firebase_admin import credentials, firestore
-
-# Twilio Configuration
-ACCOUNT_SID = 'AC97ad50edcf94fdd7d4576be9651bf4bf'
-AUTH_TOKEN = '6d8b2559ffbbdbc5d06542e545220aaa'
-FROM_PHONE = '+12708175529'
-TO_PHONE = '+916382792828'
-
-# Firebase Configuration
-cred = credentials.Certificate("E:\static\snippetscript-37175-firebase-adminsdk-cf1z8-7d509b09fd.json")
-firebase_admin.initialize_app(cred)
-db = firestore.client()
-
-# Function to send SMS using Twilio
-def send_msg(field1):
-    # if field;1
-    client = Client(ACCOUNT_SID, AUTH_TOKEN)
-    try:
-        message = client.messages.create(
-            from_=FROM_PHONE,
-            body=f"""E.coli Level: {field1}
-Status: safe
-Suggestions:
-You may heat your water to eradicate the bacteria.""",
-            to=TO_PHONE
-        )
-        print(f"Message sent successfully! SID: {message.sid}")
-    except Exception as e:
-        print(f"Failed to send message: {e}")
-
-# Firestore Listener for new data in the 'thingspeak_data' collection
-def on_snapshot(doc_snapshot, changes, read_time):
-    for change in changes:
-        if change.type.name == 'ADDED':  # Detects new documents added to the collection
-            new_data = change.document.to_dict()
-            field1 = new_data.get('field1', 'N/A')  # Replace 'field1' with your actual field key
-            print(f"New data detected: {new_data}")
-            send_msg(field1)  # Trigger the Twilio SMS function
-
-# Initialize Firestore Listener
-def start_firestore_listener():
-    thingspeak_ref = db.collection('thingspeak_data')
-    thingspeak_watch = thingspeak_ref.on_snapshot(on_snapshot)
-    print("Listening for new data in Firestore...")
-
-if __name__ == "__main__":
-    try:
-        # Start the listener
-        start_firestore_listener()
-        
-        # Keep the script running to listen for changes
-        import time
-        while True:
-            time.sleep(10)
-    except KeyboardInterrupt:
-        print("\nListener stopped.")
-    except Exception as e:
-        print(f"An error occurred: {e}")
+from twilio.rest import Client
+import firebase_admin
+from firebase_admin import credentials, firestore
+
+# Twilio Configuration
+ACCOUNT_SID = 'AC97ad50edcf94fdd7d4576be9651bf4bf'
+AUTH_TOKEN = '6d8b2559ffbbdbc5d06542e545220aaa'
+FROM_PHONE = '+12708175529'
+TO_PHONE = '+916382792828'
+import numpy as np
+import joblib
+
+# Load the saved Gradient Boosting model
+loaded_model = joblib.load('dlmodel.pkl')
+
+def predict_conc(pdiff):
+    if pdiff>3.19:
+        return 0
+    pdiff_value = pdiff  # Replace 1.5 with your input value for "Average Potential Difference"
+    new_data = np.array([[pdiff_value]])
+    prediction = loaded_model.predict(new_data)
+    concentration = 10**prediction[0]  # Reverse log transformation
+    return f"{concentration:.4f}"
+
+    
+# Firebase Configuration
+cred = credentials.Certificate("snippetscript-37175-firebase-adminsdk-cf1z8-7d509b09fd.json")
+firebase_admin.initialize_app(cred)
+db = firestore.client()
+
+# Function to send SMS using Twilio
+def send_msg(field1,suggestion):
+    # if field;1
+    
+    client = Client(ACCOUNT_SID, AUTH_TOKEN)
+    try:
+        message = client.messages.create(
+            from_=FROM_PHONE,
+            body=f"""Test Results:\nE.coli Level: {field1} CFU/mL.
+Status: safe
+Suggestions:\n
+{suggestion}""",
+            to=TO_PHONE
+        )
+        print(f"Message sent successfully! SID: {message.sid}")
+    except Exception as e:
+        print(f"Failed to send message: {e}")
+
+# Firestore Listener for new data in the 'thingspeak_data' collection
+def on_snapshot(doc_snapshot, changes, read_time):
+    for change in changes:
+        if change.type.name == 'ADDED':  # Detects new documents added to the collection
+            new_data = change.document.to_dict()
+            field1 = new_data.get('field1', 'N/A')  # Replace 'field1' with your actual field key
+            print(f"New data detected: {new_data}")
+            concentration=predict_conc(field1)
+            if concentration==0:
+                send_msg(concentration,"The water is safe for use!!")
+                
+            else:
+                send_msg(concentration,"You may boil the water to 50°C to eradicate the bacteria.")
+                
+              # Trigger the Twilio SMS function
+
+# Initialize Firestore Listener
+def start_firestore_listener():
+    thingspeak_ref = db.collection('thingspeak_data')
+    thingspeak_watch = thingspeak_ref.on_snapshot(on_snapshot)
+    print("Listening for new data in Firestore...")
+
+if __name__ == "__main__":
+    try:
+        # Start the listener
+        start_firestore_listener()
+        
+        # Keep the script running to listen for changes
+        import time
+        while True:
+            time.sleep(10)
+    except KeyboardInterrupt:
+        print("\nListener stopped.")
+    except Exception as e:
+        print(f"An error occurred: {e}")