From ef4782fc14d333ce41e67fb6e9c9d50dc53234d0 Mon Sep 17 00:00:00 2001
From: Ivan Malison <IvanMalison@gmail.com>
Date: Fri, 10 Jul 2026 00:12:06 -0700
Subject: [PATCH] Select detected ball colors for Shot Lab

---
 src/index.tsx                      | 14 ++++++++++++++
 src/operations/shot_simulation.gql |  6 ++++++
 2 files changed, 20 insertions(+)

diff --git a/src/index.tsx b/src/index.tsx
index 3da3a46..cb56791 100644
--- a/src/index.tsx
+++ b/src/index.tsx
@@ -6745,6 +6745,10 @@ export type GetTableStateQuery = {
   getTableState: {
     __typename?: "TableStateGQL";
     identifierToPosition: Array<Array<number>>;
+    identifierToColor: Array<{
+      __typename?: "RGBColorGQL";
+      hex: string;
+    } | null>;
   };
 };
 
@@ -6761,6 +6765,10 @@ export type GetLiveTableStateQuery = {
     tableState: {
       __typename?: "TableStateGQL";
       identifierToPosition: Array<Array<number>>;
+      identifierToColor: Array<{
+        __typename?: "RGBColorGQL";
+        hex: string;
+      } | null>;
       homography?: {
         __typename?: "HomographyInfoGQL";
         id: number;
@@ -14426,6 +14434,9 @@ export const GetTableStateDocument = gql`
       useHomography: $useHomography
     ) {
       identifierToPosition
+      identifierToColor {
+        hex
+      }
     }
   }
 `;
@@ -14504,6 +14515,9 @@ export const GetLiveTableStateDocument = gql`
       frameIndex
       tableState {
         identifierToPosition
+        identifierToColor {
+          hex
+        }
         homography {
           ...HomographyInfo
         }
diff --git a/src/operations/shot_simulation.gql b/src/operations/shot_simulation.gql
index cb4016e..8962a14 100644
--- a/src/operations/shot_simulation.gql
+++ b/src/operations/shot_simulation.gql
@@ -9,6 +9,9 @@ query GetTableState(
     useHomography: $useHomography
   ) {
     identifierToPosition
+    identifierToColor {
+      hex
+    }
   }
 }
 
@@ -18,6 +21,9 @@ query GetLiveTableState($videoId: Int!) {
     frameIndex
     tableState {
       identifierToPosition
+      identifierToColor {
+        hex
+      }
       homography {
         ...HomographyInfo
       }