add dashboards related endpoints

Author: vysakh0

src/data_neuron/report_cmd/main.py

@@ -17,6 +17,14 @@ def list_dashboards():
     return [f.split('.')[0] for f in os.listdir(dashboards_dir) if f.endswith('.yml')]
 
 
+def load_dashboard(dashboard_name):
+    dashboard_file = os.path.join("dashboards", f"{dashboard_name}.yml")
+    if not os.path.exists(dashboard_file):
+        return None
+    with open(dashboard_file, 'r') as f:
+        return yaml.safe_load(f)
+
+
 def load_dashboard(dashboard_name):
     dashboard_file = os.path.join("dashboards", f"{dashboard_name}.yml")
     with open(dashboard_file, 'r') as f:

src/data_neuron/server.py

@@ -3,7 +3,7 @@
 from .context_init_cmd.main import init_context
 from .ask_cmd.main import query as ask_query
 from .chat_cmd.main import process_chat_message
-from .report_cmd.main import generate_report_html
+from .report_cmd.main import generate_report_html, list_dashboards, load_dashboard
 import os
 
 app = Flask(__name__)
@@ -52,6 +52,27 @@ def generate_report():
         return jsonify({"error": str(e)}), 500
 
 
+@app.route('/dashboards', methods=['GET'])
+def get_dashboards():
+    try:
+        dashboards = list_dashboards()
+        return jsonify({"dashboards": dashboards})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+
+@app.route('/dashboards/<dashboard_id>', methods=['GET'])
+def get_dashboard(dashboard_id):
+    try:
+        dashboard = load_dashboard(dashboard_id)
+        if dashboard:
+            return jsonify(dashboard)
+        else:
+            return jsonify({"error": "Dashboard not found"}), 404
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+
 def run_server(host='0.0.0.0', port=8084, debug=False):
     app.run(host=host, port=port, debug=debug)