| from fastapi import APIRouter, Depends, HTTPException
|
| from sqlalchemy.orm import Session
|
| from backend.db.session import SessionLocal
|
| from backend.db.models import Universe, Axiom, Theorem
|
| from backend.api.auth import get_api_key
|
| from backend.core.logging_config import get_logger
|
|
|
| router = APIRouter()
|
| logger = get_logger("visualization_routes")
|
|
|
| def get_db():
|
| db = SessionLocal()
|
| try:
|
| yield db
|
| finally:
|
| db.close()
|
|
|
| @router.get("/visualization/universe/{universe_id}")
|
| def get_universe_graph(universe_id: int, db: Session = Depends(get_db), api_key: str = Depends(get_api_key)):
|
| try:
|
| universe = db.query(Universe).filter(Universe.id == universe_id).first()
|
| axioms = db.query(Axiom).filter(Axiom.universe_id == universe_id).all()
|
| theorems = db.query(Theorem).filter(Theorem.universe_id == universe_id).all()
|
| nodes = [{"id": ax.id, "type": "axiom", "label": ax.statement} for ax in axioms] + \
|
| [{"id": th.id, "type": "theorem", "label": th.statement} for th in theorems]
|
| edges = []
|
| for th in theorems:
|
| for ax in axioms:
|
| if ax.statement in th.proof:
|
| edges.append({"source": ax.id, "target": th.id, "type": "proof"})
|
| logger.info(f"Visualization graph for universe {universe_id} generated.")
|
| return {
|
| "universe": {"id": universe.id, "name": universe.name, "type": universe.universe_type},
|
| "nodes": nodes,
|
| "edges": edges
|
| }
|
| except Exception as e:
|
| logger.error(f"Visualization error: {e}")
|
| raise HTTPException(status_code=500, detail=str(e))
|
|
|
| @router.get("/visualization/universes")
|
| def get_all_universe_graphs(db: Session = Depends(get_db), api_key: str = Depends(get_api_key)):
|
| try:
|
| universes = db.query(Universe).all()
|
| result = []
|
| for universe in universes:
|
| axioms = db.query(Axiom).filter(Axiom.universe_id == universe.id).all()
|
| theorems = db.query(Theorem).filter(Theorem.universe_id == universe.id).all()
|
| nodes = [{"id": ax.id, "type": "axiom", "label": ax.statement} for ax in axioms] + \
|
| [{"id": th.id, "type": "theorem", "label": th.statement} for th in theorems]
|
| edges = []
|
| for th in theorems:
|
| for ax in axioms:
|
| if ax.statement in th.proof:
|
| edges.append({"source": ax.id, "target": th.id, "type": "proof"})
|
| result.append({
|
| "universe": {"id": universe.id, "name": universe.name, "type": universe.universe_type},
|
| "nodes": nodes,
|
| "edges": edges
|
| })
|
| logger.info("Visualization graphs for all universes generated.")
|
| return result
|
| except Exception as e:
|
| logger.error(f"Visualization error: {e}")
|
| raise HTTPException(status_code=500, detail=str(e))
|
|
|
