fix(database): fixing so that ensuring the data that has already been

app.py

@@ -94,6 +94,7 @@ def predict_machine():
                     "error": "Failed to save prediction to database"
                 }
             logger.info(f"Prediction successfully saved for UDI {sensor_udi}")
+            database.mark_as_processed(sensor_udi)
             return {
                 "success": True,
                 "failure_predicted": True,
@@ -111,7 +112,7 @@ def predict_machine():
             
             return binary_result
     else: 
-        database.create_new_predictions(
+        save_result = database.create_new_predictions(
             machine_id=sensor.get("machine_id"),
             timestamp=sensor.get("timestamp"),
             risk_score=binary_result.get("risk_score"),
@@ -120,4 +121,7 @@ def predict_machine():
             predicted_failure_time=None,
             confidence=binary_result.get("confidence")
         )
+        
+        if save_result and save_result.get("success"):
+            database.mark_as_processed(sensor_udi)
         return binary_result

database.py

@@ -22,6 +22,10 @@ class Database():
         except Exception as e:
             logger.critical(f"Failed to create Supabase client: {str(e)}")
             raise
+            
+     def mark_as_processed(self, udi):
+        self.__last_processed_id = udi
+        logger.info(f"Marked UDI {udi} as processed")
     
     def reset_last_processed_id(self):
         self.__last_processed_id = None
@@ -48,7 +52,6 @@ class Database():
                         "message": "Data already predicted",
                     }
                 
-                self.__last_processed_id = current_reading.get("udi")
                 logger.debug(f"Retrieved sensor reading - UDI: {current_reading.get('udi')}, Machine: {current_reading.get('machine_id')}")
                 return current_reading