Anto01
a26914bbe8
Dashboard: - Add Studio page with drag-drop model upload and Claude chat - Add intake system for study creation workflow - Improve session manager and context builder - Add intake API routes and frontend components Optimization Engine: - Add CLI module for command-line operations - Add intake module for study preprocessing - Add validation module with gate checks - Improve Zernike extractor documentation - Update spec models with better validation - Enhance solve_simulation robustness Documentation: - Add ATOMIZER_STUDIO.md planning doc - Add ATOMIZER_UX_SYSTEM.md for UX patterns - Update extractor library docs - Add study-readme-generator skill Tools: - Add test scripts for extraction validation - Add Zernike recentering test Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
113 lines
3.3 KiB
Python
113 lines
3.3 KiB
Python
"""
|
|
Atomizer Dashboard - FastAPI Backend
|
|
Real-time optimization monitoring and control
|
|
"""
|
|
|
|
from contextlib import asynccontextmanager
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.responses import FileResponse
|
|
from pathlib import Path
|
|
import sys
|
|
|
|
# Add parent directory to path to import optimization_engine
|
|
sys.path.append(str(Path(__file__).parent.parent.parent.parent))
|
|
|
|
from api.routes import (
|
|
optimization,
|
|
claude,
|
|
terminal,
|
|
insights,
|
|
context,
|
|
files,
|
|
nx,
|
|
claude_code,
|
|
spec,
|
|
devloop,
|
|
intake,
|
|
)
|
|
from api.websocket import optimization_stream
|
|
|
|
|
|
# Lifespan handler for session manager
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
"""Manage application lifespan - start/stop session manager"""
|
|
# Startup
|
|
from api.routes.claude import get_session_manager
|
|
|
|
manager = get_session_manager()
|
|
await manager.start()
|
|
print("Session manager started")
|
|
|
|
yield
|
|
|
|
# Shutdown
|
|
await manager.stop()
|
|
print("Session manager stopped")
|
|
|
|
|
|
# Create FastAPI app
|
|
app = FastAPI(
|
|
title="Atomizer Dashboard API",
|
|
description="Real-time optimization monitoring and control",
|
|
version="2.0.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
# Configure CORS for local development
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"], # Allow all origins for local development
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# Include routers
|
|
app.include_router(optimization.router, prefix="/api/optimization", tags=["optimization"])
|
|
app.include_router(optimization_stream.router, prefix="/api/ws", tags=["websocket"])
|
|
app.include_router(claude.router, prefix="/api/claude", tags=["claude"])
|
|
app.include_router(terminal.router, prefix="/api/terminal", tags=["terminal"])
|
|
app.include_router(insights.router, prefix="/api/insights", tags=["insights"])
|
|
app.include_router(context.router, prefix="/api/context", tags=["context"])
|
|
app.include_router(files.router, prefix="/api/files", tags=["files"])
|
|
app.include_router(nx.router, prefix="/api/nx", tags=["nx"])
|
|
app.include_router(claude_code.router, prefix="/api", tags=["claude-code"])
|
|
app.include_router(spec.router, prefix="/api", tags=["spec"])
|
|
app.include_router(spec.validate_router, prefix="/api", tags=["spec"])
|
|
app.include_router(devloop.router, prefix="/api", tags=["devloop"])
|
|
app.include_router(intake.router, prefix="/api", tags=["intake"])
|
|
|
|
|
|
@app.get("/")
|
|
async def root():
|
|
"""Serve the enhanced dashboard HTML"""
|
|
dashboard_path = Path(__file__).parent.parent.parent / "dashboard-enhanced.html"
|
|
return FileResponse(dashboard_path)
|
|
|
|
|
|
@app.get("/health")
|
|
async def health_check():
|
|
"""Health check endpoint with database status"""
|
|
try:
|
|
from api.services.conversation_store import ConversationStore
|
|
|
|
store = ConversationStore()
|
|
# Test database by creating/getting a health check session
|
|
store.get_session("health_check")
|
|
db_status = "connected"
|
|
except Exception as e:
|
|
db_status = f"error: {str(e)}"
|
|
|
|
return {
|
|
"status": "healthy" if db_status == "connected" else "degraded",
|
|
"database": db_status,
|
|
}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
|
|
uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True, log_level="info")
|