Add category pill right away

Fix #1199

1 files changed, 5 insertions, 0 deletions

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";