Add RLUtils class for managing RL/AI dashboard endpoints

- Implemented methods for fetching AI stats, training history, and recent experiences.
- Added functionality to set operation mode (MANUAL, AUTO, AI) with appropriate handling.
- Included helper methods for querying the database and sending JSON responses.
- Integrated model metadata extraction for visualization purposes.

db_utils/actions.py

@@ -162,7 +162,8 @@ class ActionOps:
                         b_rate_limit    = COALESCE(excluded.b_rate_limit, actions.b_rate_limit),
                         b_stealth_level = COALESCE(excluded.b_stealth_level, actions.b_stealth_level),
                         b_risk_level    = COALESCE(excluded.b_risk_level, actions.b_risk_level),
-                        b_enabled       = COALESCE(excluded.b_enabled, actions.b_enabled),
+                        -- Keep persisted enable/disable state from DB across restarts.
+                        b_enabled       = actions.b_enabled,
                         b_args          = COALESCE(excluded.b_args, actions.b_args),
                         b_name          = COALESCE(excluded.b_name, actions.b_name),
                         b_description   = COALESCE(excluded.b_description, actions.b_description),
@@ -218,8 +219,10 @@ class ActionOps:
                         WHERE id = 1
                     """, (action_count_row['cnt'],))
         
+        # Invalidate cache so callers immediately see fresh definitions
+        type(self).get_action_definition.cache_clear()
         logger.info(f"Synchronized {len(actions)} actions")
-    
+
     def list_actions(self):
         """List all action definitions ordered by class name"""
         return self.base.query("SELECT * FROM actions ORDER BY b_class;")
@@ -261,23 +264,6 @@ class ActionOps:
             })
         return out
     
-    # def list_action_cards(self) -> list[dict]:
-    #     """Lightweight descriptor of actions for card-based UIs"""
-    #     rows = self.base.query("""
-    #         SELECT b_class, b_enabled
-    #         FROM actions
-    #         ORDER BY b_class;
-    #     """)
-    #     out = []
-    #     for r in rows:
-    #         cls = r["b_class"]
-    #         out.append({
-    #             "name": cls,
-    #             "image": f"/actions/actions_icons/{cls}.png",
-    #             "enabled": int(r.get("b_enabled", 1) or 1),
-    #         })
-    #     return out
-    
     @lru_cache(maxsize=32)
     def get_action_definition(self, b_class: str) -> Optional[Dict[str, Any]]:
         """Cached lookup of an action definition by class name"""

db_utils/stats.py

@@ -71,10 +71,8 @@ class StatsOps:
     
     def get_stats(self) -> Dict[str, int]:
         """Compatibility alias to retrieve stats; ensures the singleton row exists"""
+        self.ensure_stats_initialized()
         row = self.base.query("SELECT total_open_ports, alive_hosts_count, all_known_hosts_count, vulnerabilities_count FROM stats WHERE id=1;")
-        if not row:
-            self.ensure_stats_initialized()
-            row = self.base.query("SELECT total_open_ports, alive_hosts_count, all_known_hosts_count, vulnerabilities_count FROM stats WHERE id=1;")
         r = row[0]
         return {
             "total_open_ports": int(r["total_open_ports"]),

db_utils/studio.py

@@ -22,6 +22,7 @@ class StudioOps:
         self.base.execute("""
             CREATE TABLE IF NOT EXISTS actions_studio (
                 b_class         TEXT PRIMARY KEY,
+                b_priority      INTEGER DEFAULT 50,
                 studio_x        REAL,
                 studio_y        REAL,
                 studio_locked   INTEGER DEFAULT 0,
@@ -31,6 +32,9 @@ class StudioOps:
             );
         """)
         
+        # Migration: ensure b_priority exists on pre-existing databases
+        self.base._ensure_column("actions_studio", "b_priority", "b_priority INTEGER DEFAULT 50")
+
         # Studio edges (relationships between actions)
         self.base.execute("""
             CREATE TABLE IF NOT EXISTS studio_edges (
@@ -255,6 +259,7 @@ class StudioOps:
         self.base.execute("""
             CREATE TABLE IF NOT EXISTS actions_studio (
                 b_class         TEXT PRIMARY KEY,
+                b_priority      INTEGER DEFAULT 50,
                 studio_x        REAL,
                 studio_y        REAL,
                 studio_locked   INTEGER DEFAULT 0,
@@ -282,10 +287,12 @@ class StudioOps:
         - Insert missing b_class entries
         - Update NULL fields only (non-destructive)
         """
-        # 1) Minimal table: PK + studio_* columns
+        # 1) Minimal table: PK + studio_* columns (b_priority must be here so
+        #    get_studio_actions() can ORDER BY it before _sync adds action columns)
         self.base.execute("""
             CREATE TABLE IF NOT EXISTS actions_studio (
                 b_class         TEXT PRIMARY KEY,
+                b_priority      INTEGER DEFAULT 50,
                 studio_x        REAL,
                 studio_y        REAL,
                 studio_locked   INTEGER DEFAULT 0,