Update schema for ground truth admin workflow

Reviewed-on: #228

flake.nix

@@ -21,7 +21,7 @@
     in
       with pkgs; {
         devShell = mkShell {
-          buildInputs = [nodejs yarn watchman alejandra nodePackages.prettier just protobuf];
+          buildInputs = [nodejs yarn watchman alejandra nodePackages.prettier just protobuf black];
         };
       });
 }

rbproto/python/shot_pb2.pyi

@@ -102,7 +102,6 @@ class CollisionInfo(_message.Message):
             key: _Optional[int] = ...,
             value: _Optional[_Union[Point, _Mapping]] = ...,
         ) -> None: ...
-
     SOURCE_FIELD_NUMBER: _ClassVar[int]
     BALL_IDENTIFIERS_FIELD_NUMBER: _ClassVar[int]
     WALL_IDENTIFIER_FIELD_NUMBER: _ClassVar[int]

src/index.tsx

@@ -224,6 +224,14 @@ export type CreateBucketSetInput = {
   keyName: Scalars["String"]["input"];
 };
 
+export type CreateGroundTruthShotInputGql = {
+  endFrame: Scalars["Int"]["input"];
+  make?: InputMaybe<Scalars["Boolean"]["input"]>;
+  notes?: InputMaybe<Scalars["String"]["input"]>;
+  processingId: Scalars["Int"]["input"];
+  startFrame: Scalars["Int"]["input"];
+};
+
 export type CreateSubscriptionResultGql = {
   __typename?: "CreateSubscriptionResultGQL";
   checkoutUrl: Scalars["String"]["output"];
@@ -269,10 +277,16 @@ export type DatetimeRangeAggregationInput = {
   startDatetime?: InputMaybe<Scalars["DateTime"]["input"]>;
 };
 
+export type DeleteGroundTruthShotReturn = {
+  __typename?: "DeleteGroundTruthShotReturn";
+  deletedShotId: Scalars["Int"]["output"];
+};
+
 export type DeployedConfigGql = {
   __typename?: "DeployedConfigGQL";
   allowNewUsers: Scalars["Boolean"]["output"];
   bannerMessages: Array<BannerGql>;
+  bucketUrl: Scalars["String"]["output"];
   defaultAndroidRecordingFormat: StreamSegmentTypeEnum;
   devMode: Scalars["Boolean"]["output"];
   environment: Scalars["String"]["output"];
@@ -307,9 +321,12 @@ export type EditUserInputGql = {
 export type EditableShotFieldInputGql = {
   backcut?: InputMaybe<Scalars["Boolean"]["input"]>;
   excludeFromStats?: InputMaybe<Scalars["Boolean"]["input"]>;
+  intendedBallId?: InputMaybe<Scalars["Int"]["input"]>;
+  intendedPocketId?: InputMaybe<PocketIdentifier>;
   intendedPocketType?: InputMaybe<PocketEnum>;
   make?: InputMaybe<Scalars["Boolean"]["input"]>;
   notes?: InputMaybe<Scalars["String"]["input"]>;
+  pathMetadataIndex?: InputMaybe<Scalars["Int"]["input"]>;
   shotDirection?: InputMaybe<ShotDirectionEnum>;
   spinType?: InputMaybe<SpinTypeEnum>;
   targetPocketAngleDirection?: InputMaybe<ShotDirectionEnum>;
@@ -350,6 +367,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -394,6 +412,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -438,6 +457,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -482,6 +502,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -526,6 +547,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -570,6 +592,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -614,6 +637,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -658,6 +682,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -702,6 +727,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -746,6 +772,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -790,6 +817,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -834,6 +862,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -878,6 +907,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -922,6 +952,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -966,6 +997,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1010,6 +1042,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1054,6 +1087,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1098,6 +1132,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1142,6 +1177,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1186,6 +1222,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1230,6 +1267,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1274,6 +1312,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1318,6 +1357,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1362,6 +1402,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1406,6 +1447,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1450,6 +1492,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1494,6 +1537,7 @@ export type FilterInput =
       missAngleInDegrees: FloatRangeFilter;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1538,6 +1582,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter: FilterInput;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1582,6 +1627,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters: Array<FilterInput>;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1626,6 +1672,52 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId: Array<Scalars["Int"]["input"]>;
+      runId?: never;
+      runLength?: never;
+      shotDirection?: never;
+      spinType?: never;
+      tableSize?: never;
+      tags?: never;
+      targetPocketAngle?: never;
+      targetPocketAngleDirection?: never;
+      targetPocketDistance?: never;
+      totalDistance?: never;
+      userId?: never;
+      username?: never;
+      videoId?: never;
+    }
+  | {
+      andFilters?: never;
+      annotations?: never;
+      backcut?: never;
+      bankAngle?: never;
+      bankDistance?: never;
+      createdAt?: never;
+      cueAngleAfterObject?: never;
+      cueBallSpeed?: never;
+      cueObjectAngle?: never;
+      cueObjectDistance?: never;
+      cueSpeedAfterObject?: never;
+      difficulty?: never;
+      falsePositiveScore?: never;
+      fargoRating?: never;
+      intendedPocketType?: never;
+      isBreakHeuristic?: never;
+      isDirect?: never;
+      isLeft?: never;
+      isLeftMiss?: never;
+      isRight?: never;
+      isRightMiss?: never;
+      isStraight?: never;
+      kickAngle?: never;
+      kickDistance?: never;
+      make?: never;
+      marginOfErrorInDegrees?: never;
+      missAngleInDegrees?: never;
+      notFilter?: never;
+      orFilters?: never;
+      processingId?: never;
       runId: Array<Scalars["Int"]["input"]>;
       runLength?: never;
       shotDirection?: never;
@@ -1670,6 +1762,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength: FloatRangeFilter;
       shotDirection?: never;
@@ -1714,6 +1807,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection: Array<ShotDirectionEnum>;
@@ -1758,6 +1852,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1802,6 +1897,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1846,6 +1942,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1890,6 +1987,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1934,6 +2032,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -1978,6 +2077,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -2022,6 +2122,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -2066,6 +2167,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -2110,6 +2212,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -2154,6 +2257,7 @@ export type FilterInput =
       missAngleInDegrees?: never;
       notFilter?: never;
       orFilters?: never;
+      processingId?: never;
       runId?: never;
       runLength?: never;
       shotDirection?: never;
@@ -2345,10 +2449,12 @@ export type Mutation = {
   commentOnVideo: Scalars["Boolean"]["output"];
   createBucketSet: BucketSetGql;
   createChallenge: Challenge;
+  createGroundTruthShot: ShotGql;
   createRuleSet: RuleSet;
   createSubscription: CreateSubscriptionResultGql;
   createUploadStream: CreateUploadStreamReturn;
   deleteComment: Scalars["Boolean"]["output"];
+  deleteGroundTruthShot: DeleteGroundTruthShotReturn;
   deleteNotification: Scalars["Boolean"]["output"];
   deleteTags: Scalars["Boolean"]["output"];
   deleteUser: Scalars["Boolean"]["output"];
@@ -2363,12 +2469,14 @@ export type Mutation = {
   findPrerecordTableLayout?: Maybe<HomographyInfoGql>;
   followUser: UserGql;
   getHlsInitUploadLink: GetUploadLinkReturn;
+  getOrCreateGroundTruthProcessing: VideoProcessingGql;
   getProfileImageUploadLink: GetProfileUploadLinkReturn;
   getUploadLink: GetUploadLinkReturn;
   inviteUsersToChallenge: Array<ChallengeInvitation>;
   markAllNotificationsAsRead: Scalars["Boolean"]["output"];
   markNotificationAsRead: Scalars["Boolean"]["output"];
   markNotificationsAsRead: Scalars["Boolean"]["output"];
+  promoteGroundTruthProcessing: VideoProcessingGql;
   reactToVideo: Scalars["Boolean"]["output"];
   recalculateChallengeEntry: ChallengeEntry;
   reportContent: Scalars["Boolean"]["output"];
@@ -2376,11 +2484,13 @@ export type Mutation = {
   retireTags: Scalars["Boolean"]["output"];
   setLoggerLevel: Scalars["Boolean"]["output"];
   setSegmentDuration: Scalars["Boolean"]["output"];
+  splitGroundTruthShot: SplitGroundTruthShotReturn;
   startChallenge: ChallengeEntry;
   submitCancellationFeedback: Scalars["Boolean"]["output"];
   submitChallengeEntry: ChallengeEntry;
   undismissChallenge: Scalars["Boolean"]["output"];
   unfollowUser: UserGql;
+  updateGroundTruthShotFrameRange: ShotGql;
   updateShotAnnotations: UpdateShotAnnotationReturn;
 };
 
@@ -2421,6 +2531,10 @@ export type MutationCreateChallengeArgs = {
   startDate: Scalars["DateTime"]["input"];
 };
 
+export type MutationCreateGroundTruthShotArgs = {
+  input: CreateGroundTruthShotInputGql;
+};
+
 export type MutationCreateRuleSetArgs = {
   description?: InputMaybe<Scalars["String"]["input"]>;
   name: Scalars["String"]["input"];
@@ -2431,6 +2545,7 @@ export type MutationCreateSubscriptionArgs = {
 };
 
 export type MutationCreateUploadStreamArgs = {
+  expectedDurationSeconds?: InputMaybe<Scalars["Float"]["input"]>;
   videoMetadata: VideoMetadataInput;
 };
 
@@ -2439,6 +2554,10 @@ export type MutationDeleteCommentArgs = {
   videoId: Scalars["Int"]["input"];
 };
 
+export type MutationDeleteGroundTruthShotArgs = {
+  shotId: Scalars["Int"]["input"];
+};
+
 export type MutationDeleteNotificationArgs = {
   notificationId: Scalars["Int"]["input"];
 };
@@ -2493,6 +2612,11 @@ export type MutationGetHlsInitUploadLinkArgs = {
   videoId: Scalars["Int"]["input"];
 };
 
+export type MutationGetOrCreateGroundTruthProcessingArgs = {
+  sourceProcessingId?: InputMaybe<Scalars["Int"]["input"]>;
+  videoId: Scalars["Int"]["input"];
+};
+
 export type MutationGetProfileImageUploadLinkArgs = {
   fileExt?: InputMaybe<Scalars["String"]["input"]>;
 };
@@ -2515,6 +2639,10 @@ export type MutationMarkNotificationsAsReadArgs = {
   notificationIds: Array<Scalars["Int"]["input"]>;
 };
 
+export type MutationPromoteGroundTruthProcessingArgs = {
+  processingId: Scalars["Int"]["input"];
+};
+
 export type MutationReactToVideoArgs = {
   reaction?: InputMaybe<ReactionEnum>;
   videoId: Scalars["Int"]["input"];
@@ -2550,6 +2678,12 @@ export type MutationSetSegmentDurationArgs = {
   videoId: Scalars["Int"]["input"];
 };
 
+export type MutationSplitGroundTruthShotArgs = {
+  notes?: InputMaybe<Scalars["String"]["input"]>;
+  shotId: Scalars["Int"]["input"];
+  splitFrame: Scalars["Int"]["input"];
+};
+
 export type MutationStartChallengeArgs = {
   challengeId: Scalars["ID"]["input"];
 };
@@ -2573,6 +2707,13 @@ export type MutationUnfollowUserArgs = {
   followedUserId: Scalars["Int"]["input"];
 };
 
+export type MutationUpdateGroundTruthShotFrameRangeArgs = {
+  endFrame?: InputMaybe<Scalars["Int"]["input"]>;
+  notes?: InputMaybe<Scalars["String"]["input"]>;
+  shotId: Scalars["Int"]["input"];
+  startFrame?: InputMaybe<Scalars["Int"]["input"]>;
+};
+
 export type MutationUpdateShotAnnotationsArgs = {
   annotations: Array<UpdateAnnotationInputGql>;
   shotId: Scalars["Int"]["input"];
@@ -2700,6 +2841,7 @@ export enum ProcessingStatusEnum {
 
 export type Query = {
   __typename?: "Query";
+  assessVideoProcessing: VideoProcessingAssessmentGql;
   challenge?: Maybe<Challenge>;
   challengeLeaderboard: Array<ChallengeEntry>;
   challenges: Array<Challenge>;
@@ -2715,6 +2857,7 @@ export type Query = {
   getMedals: RequestedMedalsGql;
   getOrderedShots: GetShotsResult;
   getPlayTime: UserPlayTimeGql;
+  getQuotaStatus: QuotaStatusGql;
   getRuns: GetRunsResult;
   getShotAnnotationTypes: Array<ShotAnnotationTypeGql>;
   getShots: Array<ShotGql>;
@@ -2730,6 +2873,7 @@ export type Query = {
   getUsersMatching: Array<UserGql>;
   getVideo: VideoGql;
   getVideoMakePercentageIntervals: Array<MakePercentageIntervalGql>;
+  getVideoProcessings: Array<VideoProcessingGql>;
   getVideos: Array<VideoGql>;
   isChallengeDismissed: Scalars["Boolean"]["output"];
   myChallengeEntries: Array<ChallengeEntry>;
@@ -2741,6 +2885,11 @@ export type Query = {
   waitFor: Scalars["Float"]["output"];
 };
 
+export type QueryAssessVideoProcessingArgs = {
+  groundTruthProcessingId?: InputMaybe<Scalars["Int"]["input"]>;
+  processingId: Scalars["Int"]["input"];
+};
+
 export type QueryChallengeArgs = {
   id: Scalars["ID"]["input"];
 };
@@ -2796,7 +2945,9 @@ export type QueryGetOrderedShotsArgs = {
   countRespectsLimit?: Scalars["Boolean"]["input"];
   filterInput: FilterInput;
   ids?: InputMaybe<Array<Scalars["Int"]["input"]>>;
+  includeExcluded?: Scalars["Boolean"]["input"];
   limit?: Scalars["Int"]["input"];
+  processingId?: InputMaybe<Scalars["Int"]["input"]>;
   shotsOrdering?: InputMaybe<GetShotsOrdering>;
 };
 
@@ -2820,7 +2971,9 @@ export type QueryGetShotAnnotationTypesArgs = {
 export type QueryGetShotsArgs = {
   countRespectsLimit?: Scalars["Boolean"]["input"];
   filterInput: FilterInput;
+  includeExcluded?: Scalars["Boolean"]["input"];
   limit?: Scalars["Int"]["input"];
+  processingId?: InputMaybe<Scalars["Int"]["input"]>;
   shotsPagination?: InputMaybe<GetShotsPagination>;
 };
 
@@ -2832,7 +2985,9 @@ export type QueryGetShotsWithMetadataArgs = {
   countRespectsLimit?: Scalars["Boolean"]["input"];
   filterInput: FilterInput;
   ids?: InputMaybe<Array<Scalars["Int"]["input"]>>;
+  includeExcluded?: Scalars["Boolean"]["input"];
   limit?: Scalars["Int"]["input"];
+  processingId?: InputMaybe<Scalars["Int"]["input"]>;
   shotsPagination?: InputMaybe<GetShotsPagination>;
 };
 
@@ -2886,6 +3041,10 @@ export type QueryGetVideoMakePercentageIntervalsArgs = {
   videoId: Scalars["ID"]["input"];
 };
 
+export type QueryGetVideoProcessingsArgs = {
+  videoId: Scalars["Int"]["input"];
+};
+
 export type QueryGetVideosArgs = {
   videoIds: Array<Scalars["Int"]["input"]>;
 };
@@ -2904,6 +3063,18 @@ export type QueryWaitForArgs = {
   duration: Scalars["Float"]["input"];
 };
 
+export type QuotaStatusGql = {
+  __typename?: "QuotaStatusGQL";
+  canUpload: Scalars["Boolean"]["output"];
+  durationLimitSeconds?: Maybe<Scalars["Int"]["output"]>;
+  durationRemainingSeconds?: Maybe<Scalars["Float"]["output"]>;
+  durationUsedSeconds: Scalars["Float"]["output"];
+  maxVideoDurationSeconds?: Maybe<Scalars["Int"]["output"]>;
+  periodEnd: Scalars["DateTime"]["output"];
+  periodStart: Scalars["DateTime"]["output"];
+  tierName: Scalars["String"]["output"];
+};
+
 export enum ReactionEnum {
   Bullseye = "BULLSEYE",
   Heart = "HEART",
@@ -3170,6 +3341,12 @@ export enum SpinTypeEnum {
   Unknown = "UNKNOWN",
 }
 
+export type SplitGroundTruthShotReturn = {
+  __typename?: "SplitGroundTruthShotReturn";
+  createdShot: ShotGql;
+  originalShot: ShotGql;
+};
+
 export type StreamErrorGql = {
   __typename?: "StreamErrorGQL";
   message: Scalars["String"]["output"];
@@ -3478,6 +3655,39 @@ export type VideoMetadataInput = {
   videoName?: InputMaybe<Scalars["String"]["input"]>;
 };
 
+export type VideoProcessingAssessmentGql = {
+  __typename?: "VideoProcessingAssessmentGQL";
+  correctlyFlaggedFalsePositivePairs: Array<VideoProcessingAssessmentShotPairGql>;
+  correctlyFlaggedFalsePositives: Scalars["Int"]["output"];
+  extraFlaggedFalsePositiveShotIds: Array<Scalars["Int"]["output"]>;
+  extraFlaggedFalsePositives: Scalars["Int"]["output"];
+  groundTruthFalsePositiveCount: Scalars["Int"]["output"];
+  groundTruthPositiveShotCount: Scalars["Int"]["output"];
+  groundTruthProcessingId: Scalars["Int"]["output"];
+  groundTruthShotCount: Scalars["Int"]["output"];
+  groundTruthShotsFlaggedFalsePositive: Scalars["Int"]["output"];
+  groundTruthShotsFlaggedFalsePositivePairs: Array<VideoProcessingAssessmentShotPairGql>;
+  makeDisagreementPairs: Array<VideoProcessingAssessmentShotPairGql>;
+  makeDisagreements: Scalars["Int"]["output"];
+  matchedGroundTruthPairs: Array<VideoProcessingAssessmentShotPairGql>;
+  matchedGroundTruthShots: Scalars["Int"]["output"];
+  missedGroundTruthShotIds: Array<Scalars["Int"]["output"]>;
+  missedGroundTruthShots: Scalars["Int"]["output"];
+  processingId: Scalars["Int"]["output"];
+  sourceFalsePositiveCount: Scalars["Int"]["output"];
+  sourceShotCount: Scalars["Int"]["output"];
+  unexpectedShotIds: Array<Scalars["Int"]["output"]>;
+  unexpectedShots: Scalars["Int"]["output"];
+  unflaggedFalsePositivePairs: Array<VideoProcessingAssessmentShotPairGql>;
+  unflaggedFalsePositives: Scalars["Int"]["output"];
+};
+
+export type VideoProcessingAssessmentShotPairGql = {
+  __typename?: "VideoProcessingAssessmentShotPairGQL";
+  groundTruthShotId: Scalars["Int"]["output"];
+  processingShotId: Scalars["Int"]["output"];
+};
+
 export type VideoProcessingErrorGql = {
   __typename?: "VideoProcessingErrorGQL";
   endSegmentIndex?: Maybe<Scalars["Int"]["output"]>;
@@ -3491,6 +3701,8 @@ export type VideoProcessingGql = {
   errors: Array<VideoProcessingErrorGql>;
   framesProcessed?: Maybe<Scalars["Int"]["output"]>;
   id: Scalars["Int"]["output"];
+  labels: Array<Scalars["String"]["output"]>;
+  parentProcessingId?: Maybe<Scalars["Int"]["output"]>;
   progressPercentage?: Maybe<Scalars["Float"]["output"]>;
   status: ProcessingStatusEnum;
   statuses: Array<VideoProcessingStatusGql>;
@@ -3977,6 +4189,7 @@ export type GetDeployedConfigQuery = {
     minimumAllowedAppVersion: string;
     subscriptionGatingEnabled: boolean;
     defaultAndroidRecordingFormat: StreamSegmentTypeEnum;
+    bucketUrl: string;
     bannerMessages: Array<{
       __typename?: "BannerGQL";
       color: string;
@@ -4211,6 +4424,21 @@ export type VideoCardFieldsFragment = {
   }>;
 };
 
+export type GetVideoFeedSessionCountQueryVariables = Exact<{
+  limit?: Scalars["Int"]["input"];
+  filters?: InputMaybe<VideoFilterInput>;
+  includePrivate?: InputMaybe<IncludePrivateEnum>;
+  feedInput?: InputMaybe<VideoFeedInputGql>;
+}>;
+
+export type GetVideoFeedSessionCountQuery = {
+  __typename?: "Query";
+  getFeedVideos: {
+    __typename?: "VideoHistoryGQL";
+    videos: Array<{ __typename?: "VideoGQL"; id: number; totalShots: number }>;
+  };
+};
+
 export type GetVideoFeedQueryVariables = Exact<{
   limit?: Scalars["Int"]["input"];
   after?: InputMaybe<Scalars["String"]["input"]>;
@@ -8219,6 +8447,7 @@ export const GetDeployedConfigDocument = gql`
         priority
       }
       defaultAndroidRecordingFormat
+      bucketUrl
     }
   }
 `;
@@ -8519,6 +8748,95 @@ export type GetFeedQueryResult = Apollo.QueryResult<
   GetFeedQuery,
   GetFeedQueryVariables
 >;
+export const GetVideoFeedSessionCountDocument = gql`
+  query GetVideoFeedSessionCount(
+    $limit: Int! = 100
+    $filters: VideoFilterInput = null
+    $includePrivate: IncludePrivateEnum = MINE
+    $feedInput: VideoFeedInputGQL = null
+  ) {
+    getFeedVideos(
+      limit: $limit
+      filters: $filters
+      includePrivate: $includePrivate
+      feedInput: $feedInput
+    ) {
+      videos {
+        id
+        totalShots
+      }
+    }
+  }
+`;
+
+/**
+ * __useGetVideoFeedSessionCountQuery__
+ *
+ * To run a query within a React component, call `useGetVideoFeedSessionCountQuery` and pass it any options that fit your needs.
+ * When your component renders, `useGetVideoFeedSessionCountQuery` returns an object from Apollo Client that contains loading, error, and data properties
+ * you can use to render your UI.
+ *
+ * @param baseOptions options that will be passed into the query, supported options are listed on: https://www.apollographql.com/docs/react/api/react-hooks/#options;
+ *
+ * @example
+ * const { data, loading, error } = useGetVideoFeedSessionCountQuery({
+ *   variables: {
+ *      limit: // value for 'limit'
+ *      filters: // value for 'filters'
+ *      includePrivate: // value for 'includePrivate'
+ *      feedInput: // value for 'feedInput'
+ *   },
+ * });
+ */
+export function useGetVideoFeedSessionCountQuery(
+  baseOptions?: Apollo.QueryHookOptions<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useQuery<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >(GetVideoFeedSessionCountDocument, options);
+}
+export function useGetVideoFeedSessionCountLazyQuery(
+  baseOptions?: Apollo.LazyQueryHookOptions<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useLazyQuery<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >(GetVideoFeedSessionCountDocument, options);
+}
+export function useGetVideoFeedSessionCountSuspenseQuery(
+  baseOptions?: Apollo.SuspenseQueryHookOptions<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >,
+) {
+  const options = { ...defaultOptions, ...baseOptions };
+  return Apollo.useSuspenseQuery<
+    GetVideoFeedSessionCountQuery,
+    GetVideoFeedSessionCountQueryVariables
+  >(GetVideoFeedSessionCountDocument, options);
+}
+export type GetVideoFeedSessionCountQueryHookResult = ReturnType<
+  typeof useGetVideoFeedSessionCountQuery
+>;
+export type GetVideoFeedSessionCountLazyQueryHookResult = ReturnType<
+  typeof useGetVideoFeedSessionCountLazyQuery
+>;
+export type GetVideoFeedSessionCountSuspenseQueryHookResult = ReturnType<
+  typeof useGetVideoFeedSessionCountSuspenseQuery
+>;
+export type GetVideoFeedSessionCountQueryResult = Apollo.QueryResult<
+  GetVideoFeedSessionCountQuery,
+  GetVideoFeedSessionCountQueryVariables
+>;
 export const GetVideoFeedDocument = gql`
   query GetVideoFeed(
     $limit: Int! = 5

src/operations/config.gql

@@ -15,5 +15,6 @@ query getDeployedConfig {
       priority
     }
     defaultAndroidRecordingFormat
+    bucketUrl
   }
 }

src/operations/feed.gql

@@ -90,6 +90,26 @@ fragment VideoCardFields on VideoGQL {
   }
 }
 
+# Lightweight version of GetVideoFeed for counting sessions only
+query GetVideoFeedSessionCount(
+  $limit: Int! = 100
+  $filters: VideoFilterInput = null
+  $includePrivate: IncludePrivateEnum = MINE
+  $feedInput: VideoFeedInputGQL = null
+) {
+  getFeedVideos(
+    limit: $limit
+    filters: $filters
+    includePrivate: $includePrivate
+    feedInput: $feedInput
+  ) {
+    videos {
+      id
+      totalShots
+    }
+  }
+}
+
 query GetVideoFeed(
   $limit: Int! = 5
   $after: String = null

src/schema.gql

@@ -61,6 +61,8 @@ type Query {
     shotsOrdering: GetShotsOrdering = null
     limit: Int! = 500
     countRespectsLimit: Boolean! = false
+    processingId: Int = null
+    includeExcluded: Boolean! = false
   ): GetShotsResult!
   getShotsWithMetadata(
     filterInput: FilterInput!
@@ -68,12 +70,16 @@ type Query {
     shotsPagination: GetShotsPagination = null
     limit: Int! = 500
     countRespectsLimit: Boolean! = false
+    processingId: Int = null
+    includeExcluded: Boolean! = false
   ): GetShotsResult!
   getShots(
     filterInput: FilterInput!
     shotsPagination: GetShotsPagination = null
     limit: Int! = 500
     countRespectsLimit: Boolean! = false
+    processingId: Int = null
+    includeExcluded: Boolean! = false
   ): [ShotGQL!]!
   getShotsByIds(ids: [Int!]!): [ShotGQL!]!
   getUser(userId: Int!): UserGQL
@@ -97,6 +103,7 @@ type Query {
   ): UserRelationshipsResult!
   getAvailableSubscriptionOptions: StripeSubscriptionOptionsGQL!
   getUserSubscriptionStatus: UserSubscriptionStatusGQL!
+  getQuotaStatus: QuotaStatusGQL!
   getPlayTime(userId: Int!, filters: VideoFilterInput = null): UserPlayTimeGQL!
   getUserVideos(
     userId: Int = null
@@ -107,6 +114,11 @@ type Query {
   getUserTags(includeRetiredTags: Boolean = false): [TagGQL!]!
   getVideo(videoId: Int!, debuggingJson: JSON = null): VideoGQL!
   getVideos(videoIds: [Int!]!): [VideoGQL!]!
+  getVideoProcessings(videoId: Int!): [VideoProcessingGQL!]!
+  assessVideoProcessing(
+    processingId: Int!
+    groundTruthProcessingId: Int = null
+  ): VideoProcessingAssessmentGQL!
 }
 
 type AggregateResultGQL {
@@ -208,6 +220,7 @@ input FilterInput @oneOf {
   intendedPocketType: [PocketEnum!]
   shotDirection: [ShotDirectionEnum!]
   videoId: [Int!]
+  processingId: [Int!]
   userId: [Int!]
   runId: [Int!]
   username: [String!]
@@ -605,6 +618,8 @@ type IntPoint2D {
 
 type VideoProcessingGQL {
   id: Int!
+  parentProcessingId: Int
+  labels: [String!]!
   errors: [VideoProcessingErrorGQL!]!
   status: ProcessingStatusEnum!
   statuses: [VideoProcessingStatusGQL!]!
@@ -669,6 +684,7 @@ type DeployedConfigGQL {
   subscriptionGatingEnabled: Boolean!
   bannerMessages: [BannerGQL!]!
   defaultAndroidRecordingFormat: StreamSegmentTypeEnum!
+  bucketUrl: String!
 }
 
 type BannerGQL {
@@ -976,6 +992,17 @@ enum StripeSubscriptionStatusEnum {
   PAUSED
 }
 
+type QuotaStatusGQL {
+  tierName: String!
+  periodStart: DateTime!
+  periodEnd: DateTime!
+  durationUsedSeconds: Float!
+  durationLimitSeconds: Int
+  maxVideoDurationSeconds: Int
+  durationRemainingSeconds: Float
+  canUpload: Boolean!
+}
+
 type UserPlayTimeGQL {
   totalSeconds: Float!
 }
@@ -1000,6 +1027,37 @@ scalar JSON
     url: "https://ecma-international.org/wp-content/uploads/ECMA-404_2nd_edition_december_2017.pdf"
   )
 
+type VideoProcessingAssessmentGQL {
+  processingId: Int!
+  groundTruthProcessingId: Int!
+  sourceShotCount: Int!
+  sourceFalsePositiveCount: Int!
+  groundTruthShotCount: Int!
+  groundTruthPositiveShotCount: Int!
+  groundTruthFalsePositiveCount: Int!
+  matchedGroundTruthShots: Int!
+  missedGroundTruthShots: Int!
+  unexpectedShots: Int!
+  correctlyFlaggedFalsePositives: Int!
+  unflaggedFalsePositives: Int!
+  extraFlaggedFalsePositives: Int!
+  groundTruthShotsFlaggedFalsePositive: Int!
+  makeDisagreements: Int!
+  matchedGroundTruthPairs: [VideoProcessingAssessmentShotPairGQL!]!
+  missedGroundTruthShotIds: [Int!]!
+  unexpectedShotIds: [Int!]!
+  correctlyFlaggedFalsePositivePairs: [VideoProcessingAssessmentShotPairGQL!]!
+  unflaggedFalsePositivePairs: [VideoProcessingAssessmentShotPairGQL!]!
+  extraFlaggedFalsePositiveShotIds: [Int!]!
+  groundTruthShotsFlaggedFalsePositivePairs: [VideoProcessingAssessmentShotPairGQL!]!
+  makeDisagreementPairs: [VideoProcessingAssessmentShotPairGQL!]!
+}
+
+type VideoProcessingAssessmentShotPairGQL {
+  processingShotId: Int!
+  groundTruthShotId: Int!
+}
+
 type Mutation {
   createBucketSet(params: CreateBucketSetInput!): BucketSetGQL!
   createRuleSet(name: String!, description: String = null): RuleSet!
@@ -1061,6 +1119,19 @@ type Mutation {
     shotId: Int!
     fieldsToEdit: EditableShotFieldInputGQL!
   ): EditShotReturn!
+  createGroundTruthShot(input: CreateGroundTruthShotInputGQL!): ShotGQL!
+  updateGroundTruthShotFrameRange(
+    shotId: Int!
+    startFrame: Int = null
+    endFrame: Int = null
+    notes: String = null
+  ): ShotGQL!
+  splitGroundTruthShot(
+    shotId: Int!
+    splitFrame: Int!
+    notes: String = null
+  ): SplitGroundTruthShotReturn!
+  deleteGroundTruthShot(shotId: Int!): DeleteGroundTruthShotReturn!
   getProfileImageUploadLink(
     fileExt: String = ".png"
   ): GetProfileUploadLinkReturn!
@@ -1081,7 +1152,13 @@ type Mutation {
   findPrerecordTableLayout(b64Image: String!, videoId: Int!): HomographyInfoGQL
   createUploadStream(
     videoMetadata: VideoMetadataInput!
+    expectedDurationSeconds: Float = null
   ): CreateUploadStreamReturn!
+  getOrCreateGroundTruthProcessing(
+    videoId: Int!
+    sourceProcessingId: Int = null
+  ): VideoProcessingGQL!
+  promoteGroundTruthProcessing(processingId: Int!): VideoProcessingGQL!
   getUploadLink(videoId: Int!, segmentIndex: Int!): GetUploadLinkReturn!
   getHlsInitUploadLink(videoId: Int!): GetUploadLinkReturn!
   setSegmentDuration(
@@ -1158,12 +1235,32 @@ input EditableShotFieldInputGQL {
   shotDirection: ShotDirectionEnum
   spinType: SpinTypeEnum
   targetPocketAngleDirection: ShotDirectionEnum
+  intendedBallId: Int
+  intendedPocketId: PocketIdentifier
+  pathMetadataIndex: Int
   make: Boolean
   backcut: Boolean
   excludeFromStats: Boolean
   notes: String
 }
 
+input CreateGroundTruthShotInputGQL {
+  processingId: Int!
+  startFrame: Int!
+  endFrame: Int!
+  make: Boolean
+  notes: String
+}
+
+type SplitGroundTruthShotReturn {
+  originalShot: ShotGQL!
+  createdShot: ShotGQL!
+}
+
+type DeleteGroundTruthShotReturn {
+  deletedShotId: Int!
+}
+
 type GetProfileUploadLinkReturn {
   value: UploadLinkGetProfileUploadLinkErrors!
 }