Update app.py

app.py

@@ -14,13 +14,26 @@ FROM_PHONE = '+13613093564'
 TO_PHONE = '+919080522395'
 
 # Firebase Configuration
-FIREBASE_CREDENTIALS_PATH = r"shh\snippetscript-37175-firebase-adminsdk-cf1z8-7d509b09fd.json"
+FIREBASE_CREDENTIALS_PATH = r"snippetscript-37175-firebase-adminsdk-cf1z8-7d509b09fd.json"
 THINGSPEAK_API_KEY = "P54KXM40TA3CB6W4"
 CHANNEL_ID = "2784385"
 
 # Global variable to track the last entry ID
 last_entry_id = None
-
+with open('model.pkl', 'rb') as f:
+    loaded_model = pickle.load(f)
+def predict_conc(pdiff):
+    try:
+        if pdiff > 3.19:
+            return 0
+        pdiff_value = pdiff  # Input 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}"
+    except Exception as e:
+        print(f"Error in predict_conc: {e}")
+        return None
 # Initialize Firebase
 cred = credentials.Certificate(FIREBASE_CREDENTIALS_PATH)
 firebase_admin.initialize_app(cred)
@@ -44,12 +57,18 @@ def send_msg(doc_id, field1):
 
             # Check if the message status is 0
             if msg_status == 0:
+                concen=predict_conc(int(field1))
                 client = Client(ACCOUNT_SID, AUTH_TOKEN)
-
+                if concen!=0:
+                    sugges="You may heat your water to eradicate the bacteria."
+                else:
+                    sugges="Your water is safe.It can be used for drinking purpose!."
+                    
+                    
                 # Send the message
                 message = client.messages.create(
                     from_=FROM_PHONE,
-                    body=f"""E.coli Level: {field1}\nStatus: Safe\nSuggestions:\nYou may heat your water to eradicate the bacteria.""",
+                    body=f"""E.coli Level: {concen} CFU/mL\nStatus: Safe\nSuggestions:\n{sugges}\n--Ministry of Jal Shakthi.""",
                     to=TO_PHONE
                 )