feat: TKG extension - pose data + mutual gaze detection

src/api/trace_agent_api.rs

@@ -29,6 +29,10 @@ pub fn trace_agent_routes() -> Router<crate::api::types::AppState> {
             "/api/v1/file/:file_uuid/identities/:identity_uuid_a/co-occur-with/:identity_uuid_b",
             get(get_cooccurrence),
         )
+        .route(
+            "/api/v1/file/:file_uuid/tkg/rebuild",
+            post(rebuild_tkg),
+        )
 }
 
 #[derive(Debug, Deserialize)]
@@ -735,3 +739,47 @@ async fn get_cooccurrence(
         },
     }))
 }
+
+use crate::core::config::OUTPUT_DIR;
+
+#[derive(Serialize)]
+struct TkgRebuildResponse {
+    success: bool,
+    file_uuid: String,
+    result: Option<serde_json::Value>,
+    error: Option<String>,
+}
+
+async fn rebuild_tkg(
+    State(state): State<crate::api::types::AppState>,
+    Path(file_uuid): Path<String>,
+) -> Json<TkgRebuildResponse> {
+    let result = crate::core::processor::tkg::build_tkg(
+        &state.db,
+        &file_uuid,
+        &OUTPUT_DIR,
+    )
+    .await;
+
+    match result {
+        Ok(r) => Json(TkgRebuildResponse {
+            success: true,
+            file_uuid,
+            result: Some(serde_json::json!({
+                "face_trace_nodes": r.face_trace_nodes,
+                "object_nodes": r.object_nodes,
+                "speaker_nodes": r.speaker_nodes,
+                "co_occurrence_edges": r.co_occurrence_edges,
+                "speaker_face_edges": r.speaker_face_edges,
+                "face_face_edges": r.face_face_edges,
+            })),
+            error: None,
+        }),
+        Err(e) => Json(TkgRebuildResponse {
+            success: false,
+            file_uuid,
+            result: None,
+            error: Some(e.to_string()),
+        }),
+    }
+}