Update watchdog.py

watchdog.py

@@ -1,4 +1,5 @@
 import os
+import json
 import torch
 import firebase_admin
 from firebase_admin import credentials, firestore
@@ -79,9 +80,26 @@ def retrain_model():
             optimizer.step()
             print(f"Epoch {epoch+1}: Loss = {loss.item():.4f}")
 
+        # Simulate accuracy (replace this with actual eval in the future)
+        accuracy = 1.0  # Replace with real evaluation later
+
+        # Save accuracy to evolution log
+        log_path = "trained_model/evolution_log.json"
         os.makedirs("trained_model", exist_ok=True)
-        model.save_pretrained("trained_model")
 
+        if os.path.exists(log_path):
+            with open(log_path, "r") as f:
+                history = json.load(f)
+        else:
+            history = []
+
+        history.append({"accuracy": accuracy})
+
+        with open(log_path, "w") as f:
+            json.dump(history, f)
+
+        # Save trained model
+        model.save_pretrained("trained_model")
         print("✅ EvoTransformer retrained and saved.")
 
         # Reload the updated model for summary + plot