File size: 2,097 Bytes
1e3df84 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 | """
Pydantic models for API request/response validation
"""
from typing import List, Dict, Any, Optional
from pydantic import BaseModel, Field
class ExampleImage(BaseModel):
"""Example image information"""
name: str
display_name: str
size_kb: float
has_cached_result: bool = False
class NodeData(BaseModel):
"""Graph node data structure"""
id: str
label: str
type: str
position: Dict[str, float] # {x, y}
floor: Optional[str] = None
attributes: Dict[str, Any] = {}
class EdgeData(BaseModel):
"""Graph edge data structure"""
id: str
source: str
target: str
weight: float
label: Optional[str] = None
class CytoscapeGraph(BaseModel):
"""Cytoscape.js compatible graph format"""
nodes: List[Dict[str, Any]]
edges: List[Dict[str, Any]]
layout: str = "preset" # Use preset positions from graph
class ProcessingResponse(BaseModel):
"""Response from image processing"""
session_id: str
status: str = Field(description="success, error, or processing")
image_name: str
processing_time: float
graph_data: Optional[CytoscapeGraph] = None
pre_pruning_graph_data: Optional[CytoscapeGraph] = None
statistics: Dict[str, Any] = {}
message: str = ""
error: Optional[str] = None
class GraphVisualization(BaseModel):
"""Graph visualization configuration"""
show_labels: bool = True
node_size: Dict[str, int] = {
"room": 30,
"door": 20,
"corridor": 15,
"outside": 25,
"floor_transition": 35
}
colors: Dict[str, str] = {
"room": "#E69F00",
"door": "#009E73",
"corridor": "#CC79A7",
"outside": "#56B4E9",
"floor_transition": "#D55E00"
}
edge_colors: Dict[str, str] = {
"room": "#FFA500",
"corridor": "#3399FF",
"outside": "#FF0000",
"transition": "#B40000"
}
class ExportRequest(BaseModel):
"""Request to export graph data"""
session_id: str
format: str = Field(description="json, png, or svg")
include_image: bool = False
|