Update dashboard.py

dashboard.py

@@ -4,6 +4,8 @@ from firebase_admin import credentials, firestore
 import gradio as gr
 import io
 from PIL import Image
+import os
+import json
 
 # Initialize Firebase if not already initialized
 if not firebase_admin._apps:
@@ -12,6 +14,7 @@ if not firebase_admin._apps:
 
 db = firestore.client()
 
+# ✅ This stays as-is: Firebase Feedback Summary
 def update_dashboard_plot():
     logs_ref = db.collection("evo_feedback")
     docs = logs_ref.stream()
@@ -38,3 +41,31 @@ def update_dashboard_plot():
     buf.seek(0)
     img = Image.open(buf)
     return img
+
+# ✅ NEW: Accuracy Plot from Local Log
+def evolution_accuracy_plot():
+    try:
+        log_path = "trained_model/evolution_log.json"
+        if not os.path.exists(log_path):
+            fig, ax = plt.subplots()
+            ax.text(0.5, 0.5, "No evolution log found", ha="center", va="center")
+            return fig
+
+        with open(log_path, "r") as f:
+            log_data = json.load(f)
+
+        generations = list(range(1, len(log_data) + 1))
+        accuracies = [entry.get("accuracy", 0) for entry in log_data]
+
+        fig, ax = plt.subplots()
+        ax.plot(generations, accuracies, marker="o", linestyle="-")
+        ax.set_xlabel("Generation")
+        ax.set_ylabel("Accuracy")
+        ax.set_title("EvoTransformer Evolution Accuracy")
+        ax.grid(True)
+
+        return fig
+    except Exception as e:
+        fig, ax = plt.subplots()
+        ax.text(0.5, 0.5, f"Error loading plot: {e}", ha="center")
+        return fig