fix: add back DB init endpoints, ref NOISSUE

ai_software_factory/dashboard_ui.py

@@ -20,17 +20,23 @@ def create_dashboard():
         return
     
     try:
-        # Fetch current project
-        current_project = db_session.query(ProjectHistory).order_by(ProjectHistory.created_at.desc()).first()
-        
-        # Fetch recent audit trail entries
-        recent_audits = db_session.query(AuditTrail).order_by(AuditTrail.created_at.desc()).limit(10).all()
-        
-        # Fetch recent project history entries
-        recent_projects = db_session.query(ProjectHistory).order_by(ProjectHistory.created_at.desc()).limit(5).all()
-        
-        # Fetch recent system logs
-        recent_logs = db_session.query(SystemLog).order_by(SystemLog.created_at.desc()).limit(5).all()
+        # Wrap database queries to handle missing tables gracefully
+        try:
+            # Fetch current project
+            current_project = db_session.query(ProjectHistory).order_by(ProjectHistory.created_at.desc()).first()
+            
+            # Fetch recent audit trail entries
+            recent_audits = db_session.query(AuditTrail).order_by(AuditTrail.created_at.desc()).limit(10).all()
+            
+            # Fetch recent project history entries
+            recent_projects = db_session.query(ProjectHistory).order_by(ProjectHistory.created_at.desc()).limit(5).all()
+            
+            # Fetch recent system logs
+            recent_logs = db_session.query(SystemLog).order_by(SystemLog.created_at.desc()).limit(5).all()
+        except Exception as e:
+            # Handle missing tables or other database errors
+            ui.label(f'Database error: {str(e)}. Please run POST /init-db or ensure the database is initialized.')
+            return
         
         # Create main card
         with ui.card().col().classes('w-full max-w-6xl mx-auto').props('elevated').style('max-width: 1200px; margin: 0 auto;') as main_card: