From 6e23650a99765cf15084a8d13c0af218fbd3af6a Mon Sep 17 00:00:00 2001
From: Ajay <dev@ajay.app>
Date: Tue, 22 Feb 2022 01:01:27 -0500
Subject: [PATCH] Add category pill right away

Fix #1199
---
 src/content.ts | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/content.ts b/src/content.ts
index 0b8233dd..26a2b772 100644
--- a/src/content.ts
+++ b/src/content.ts
@@ -1934,6 +1934,11 @@ async function sendSubmitMessage() {
         sponsorTimesSubmitting = [];
 
         updatePreviewBar();
+
+        const fullVideoSegment = sponsorTimes.filter((time) => time.actionType === ActionType.Full)[0];
+        if (fullVideoSegment) {
+            categoryPill?.setSegment(fullVideoSegment);
+        }
     } else {
         // Show that the upload failed
         playerButtons.submit.button.style.animation = "unset";