karaoke-gen 0.76.25__tar.gz → 0.76.28__tar.gz

{karaoke_gen-0.76.25 → karaoke_gen-0.76.28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: karaoke-gen
-Version: 0.76.25
+Version: 0.76.28
 Summary: Generate karaoke videos with synchronized lyrics. Handles the entire process from downloading audio and lyrics to creating the final video with title screens.
 License: MIT
 License-File: LICENSE
@@ -172,7 +172,7 @@ Run Whisper directly on your local machine using whisper-timestamped. Works on C
 
 ```bash
 # Install with local Whisper support
-pip install karaoke-gen[local-whisper]
+pip install "karaoke-gen[local-whisper]"
 
 # Optional: Configure model size (tiny, base, small, medium, large)
 export WHISPER_MODEL_SIZE="medium"
@@ -194,7 +194,7 @@ export WHISPER_DEVICE="cpu"
 ```bash
 # Pre-install CPU-only PyTorch first
 pip install torch torchaudio --index-url https://download.pytorch.org/whl/cpu
-pip install karaoke-gen[local-whisper]
+pip install "karaoke-gen[local-whisper]"
 ```
 
 Local Whisper runs automatically as a fallback when no cloud transcription services are configured.
@@ -729,7 +729,7 @@ Whisper models are downloaded on first use (~1-3GB depending on size). If downlo
 #### whisper-timestamped Not Found
 If you get "whisper-timestamped is not installed":
 ```bash
-pip install karaoke-gen[local-whisper]
+pip install "karaoke-gen[local-whisper]"
 # Or install directly:
 pip install whisper-timestamped
 ```

{karaoke_gen-0.76.25 → karaoke_gen-0.76.28}/README.md

@@ -76,7 +76,7 @@ Run Whisper directly on your local machine using whisper-timestamped. Works on C
 
 ```bash
 # Install with local Whisper support
-pip install karaoke-gen[local-whisper]
+pip install "karaoke-gen[local-whisper]"
 
 # Optional: Configure model size (tiny, base, small, medium, large)
 export WHISPER_MODEL_SIZE="medium"
@@ -98,7 +98,7 @@ export WHISPER_DEVICE="cpu"
 ```bash
 # Pre-install CPU-only PyTorch first
 pip install torch torchaudio --index-url https://download.pytorch.org/whl/cpu
-pip install karaoke-gen[local-whisper]
+pip install "karaoke-gen[local-whisper]"
 ```
 
 Local Whisper runs automatically as a fallback when no cloud transcription services are configured.
@@ -633,7 +633,7 @@ Whisper models are downloaded on first use (~1-3GB depending on size). If downlo
 #### whisper-timestamped Not Found
 If you get "whisper-timestamped is not installed":
 ```bash
-pip install karaoke-gen[local-whisper]
+pip install "karaoke-gen[local-whisper]"
 # Or install directly:
 pip install whisper-timestamped
 ```

{karaoke_gen-0.76.25 → karaoke_gen-0.76.28}/lyrics_transcriber_temp/lyrics_transcriber/core/config.py

@@ -11,12 +11,13 @@ class TranscriberConfig:
     runpod_api_key: Optional[str] = None
     whisper_runpod_id: Optional[str] = None
 
-    # Local Whisper configuration
+    # Local Whisper configuration - reads from environment variables with sensible defaults
+    # Environment variables: WHISPER_MODEL_SIZE, WHISPER_DEVICE, WHISPER_CACHE_DIR, WHISPER_LANGUAGE
     enable_local_whisper: bool = True  # Enabled by default as fallback
-    local_whisper_model_size: str = "medium"  # tiny, base, small, medium, large
-    local_whisper_device: Optional[str] = None  # None for auto-detect, or cpu/cuda/mps
-    local_whisper_cache_dir: Optional[str] = None  # Custom model download location
-    local_whisper_language: Optional[str] = None  # None for auto-detect
+    local_whisper_model_size: str = field(default_factory=lambda: os.getenv("WHISPER_MODEL_SIZE", "medium"))
+    local_whisper_device: Optional[str] = field(default_factory=lambda: os.getenv("WHISPER_DEVICE"))
+    local_whisper_cache_dir: Optional[str] = field(default_factory=lambda: os.getenv("WHISPER_CACHE_DIR"))
+    local_whisper_language: Optional[str] = field(default_factory=lambda: os.getenv("WHISPER_LANGUAGE"))
 
 
 @dataclass

{karaoke_gen-0.76.25 → karaoke_gen-0.76.28}/lyrics_transcriber_temp/lyrics_transcriber/frontend/.gitignore

@@ -21,3 +21,4 @@ dist-ssr
 *.njsproj
 *.sln
 *.sw?
+test-results/

karaoke_gen-0.76.28/lyrics_transcriber_temp/lyrics_transcriber/frontend/e2e/agentic-corrections.spec.ts

@@ -0,0 +1,207 @@
+import { test, expect } from '@playwright/test';
+import * as path from 'path';
+import { fileURLToPath } from 'url';
+
+// Get __dirname equivalent in ESM
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+/**
+ * E2E tests for the agentic correction workflow in the lyrics transcriber frontend.
+ *
+ * These tests verify:
+ * 1. The UI loads correctly with agentic correction data
+ * 2. The AgenticCorrectionMetrics panel displays correctly
+ * 3. Corrected words are highlighted and clickable
+ * 4. The CorrectionDetailCard shows proper information
+ */
+
+// Helper function to load fixture data
+async function loadFixtureData(page: import('@playwright/test').Page) {
+  const fixturePath = path.join(__dirname, 'fixtures', 'agentic-correction-data.json');
+
+  // Create a file chooser promise before clicking
+  const fileChooserPromise = page.waitForEvent('filechooser');
+
+  // Click the Load File button
+  await page.getByRole('button', { name: /load file/i }).click();
+
+  // Handle the file chooser
+  const fileChooser = await fileChooserPromise;
+  await fileChooser.setFiles(fixturePath);
+
+  // Wait for data to load by asserting expected content appears
+  await expect(page.getByText('Hello,')).toBeVisible({ timeout: 5000 });
+}
+
+test.describe('Agentic Correction Workflow', () => {
+  test.beforeEach(async ({ page }) => {
+    // Navigate to the app
+    await page.goto('/');
+
+    // Wait for the initial load
+    await expect(page.getByText('Lyrics Correction Review')).toBeVisible();
+  });
+
+  test('should load the app in read-only mode', async ({ page }) => {
+    // Verify read-only mode alert is shown
+    await expect(page.getByText('Running in read-only mode')).toBeVisible();
+
+    // Verify Load File button is present
+    await expect(page.getByRole('button', { name: /load file/i })).toBeVisible();
+  });
+
+  test('should load correction data from JSON file', async ({ page }) => {
+    await loadFixtureData(page);
+
+    // Verify we're no longer in loading state
+    await expect(page.getByText('Loading Lyrics Correction Review...')).not.toBeVisible();
+
+    // Verify expected content from fixture is visible
+    await expect(page.getByText('Hello,')).toBeVisible();
+  });
+
+  test('should render transcription view after loading data', async ({ page }) => {
+    await loadFixtureData(page);
+
+    // Wait for transcription content to render
+    await expect(page.getByText('Hello,')).toBeVisible();
+
+    // Verify the Corrected Transcription header is visible
+    await expect(page.getByText('Corrected Transcription')).toBeVisible();
+  });
+});
+
+test.describe('UI Components', () => {
+  test('should show Load File button on initial load', async ({ page }) => {
+    await page.goto('/');
+
+    // The Load File button should be visible
+    const loadButton = page.getByRole('button', { name: /load file/i });
+    await expect(loadButton).toBeVisible();
+
+    // Should have upload icon
+    const uploadIcon = page.locator('svg[data-testid="UploadFileIcon"]');
+    await expect(uploadIcon).toBeVisible();
+  });
+
+  test('should show read-only mode banner', async ({ page }) => {
+    await page.goto('/');
+
+    // The read-only alert should be visible
+    await expect(page.getByRole('alert')).toBeVisible();
+    await expect(page.getByText(/read-only mode/i)).toBeVisible();
+  });
+
+  test('should have correction metrics component', async ({ page }) => {
+    await page.goto('/');
+
+    // The page structure should be there with Paper components
+    const metricsSection = page.locator('.MuiPaper-root');
+    await expect(metricsSection.first()).toBeVisible();
+  });
+});
+
+test.describe('File Upload Flow', () => {
+  test('should open file dialog when clicking Load File', async ({ page }) => {
+    await page.goto('/');
+
+    // Set up listener for file chooser
+    let fileChooserOpened = false;
+    page.on('filechooser', () => {
+      fileChooserOpened = true;
+    });
+
+    // Click the button
+    const loadButton = page.getByRole('button', { name: /load file/i });
+    await loadButton.click();
+
+    // Wait for the file chooser event to be processed
+    await page.waitForEvent('filechooser', { timeout: 5000 }).catch(() => {
+      // Event already fired
+    });
+
+    // Verify file chooser was triggered
+    expect(fileChooserOpened).toBe(true);
+  });
+});
+
+test.describe('Review Mode', () => {
+  test.beforeEach(async ({ page }) => {
+    await page.goto('/');
+    await expect(page.getByText('Lyrics Correction Review')).toBeVisible();
+  });
+
+  test('should show Review Mode toggle when agentic data is loaded', async ({ page }) => {
+    await loadFixtureData(page);
+
+    // Wait for content to load
+    await expect(page.getByText('Hello,')).toBeVisible();
+
+    // The Review Mode toggle should be visible when agentic corrections are present
+    // It will appear as "Review Off" chip initially (only in non-read-only mode with agentic data)
+    // Note: In read-only mode, the toggle won't appear
+    const reviewChip = page.getByText(/Review Off|Review Mode/i);
+    const chipCount = await reviewChip.count();
+
+    // Log for debugging purposes (will show in test output)
+    if (chipCount === 0) {
+      // Review toggle only shows in edit mode, not read-only mode
+      // This is expected behavior when loading files in read-only mode
+    }
+  });
+
+  test('should show batch actions panel when Review Mode is enabled', async ({ page }) => {
+    await loadFixtureData(page);
+
+    // Wait for content to load
+    await expect(page.getByText('Hello,')).toBeVisible();
+
+    // Find the Review Mode toggle (only visible in non-read-only mode)
+    const reviewToggle = page.getByText(/Review Off/i);
+
+    if (await reviewToggle.isVisible({ timeout: 2000 }).catch(() => false)) {
+      await reviewToggle.click();
+
+      // Wait for the batch actions panel to appear
+      await expect(page.getByRole('button', { name: /Accept High Confidence/i })).toBeVisible({ timeout: 5000 });
+      await expect(page.getByRole('button', { name: /Accept All/i })).toBeVisible();
+      await expect(page.getByRole('button', { name: /Revert All/i })).toBeVisible();
+    }
+  });
+
+  test('should render corrected words with original text preview', async ({ page }) => {
+    await loadFixtureData(page);
+
+    // Wait for transcription content to load
+    await expect(page.getByText('Hello,')).toBeVisible();
+
+    // Verify correction-related content is rendered
+    // The corrected word "now" should be visible (from fixture: "you're" -> "now")
+    await expect(page.getByText('now')).toBeVisible();
+  });
+
+  test('should toggle Review Mode on and off', async ({ page }) => {
+    await loadFixtureData(page);
+
+    // Wait for content to load
+    await expect(page.getByText('Hello,')).toBeVisible();
+
+    // Find the Review toggle (only visible in non-read-only mode)
+    const reviewOff = page.getByText(/Review Off/i);
+
+    if (await reviewOff.isVisible({ timeout: 2000 }).catch(() => false)) {
+      // Click to enable Review Mode
+      await reviewOff.click();
+
+      // Should now show "Review Mode" label (in filled state)
+      await expect(page.getByText(/Review Mode/i).first()).toBeVisible();
+
+      // Click to disable Review Mode
+      await page.getByText(/Review Mode/i).first().click();
+
+      // Should show "Review Off" again
+      await expect(page.getByText(/Review Off/i)).toBeVisible();
+    }
+  });
+});

karaoke_gen-0.76.28/lyrics_transcriber_temp/lyrics_transcriber/frontend/e2e/fixtures/agentic-correction-data.json

@@ -0,0 +1,226 @@
+{
+  "original_segments": [
+    {
+      "id": "seg-1",
+      "text": "Hello, is it me you're looking for?",
+      "words": [
+        {"id": "w1", "text": "Hello,", "start_time": 0.0, "end_time": 0.5},
+        {"id": "w2", "text": "is", "start_time": 0.5, "end_time": 0.7},
+        {"id": "w3", "text": "it", "start_time": 0.7, "end_time": 0.9},
+        {"id": "w4", "text": "me", "start_time": 0.9, "end_time": 1.1},
+        {"id": "w5", "text": "you're", "start_time": 1.1, "end_time": 1.3},
+        {"id": "w6", "text": "looking", "start_time": 1.3, "end_time": 1.6},
+        {"id": "w7", "text": "for?", "start_time": 1.6, "end_time": 1.9}
+      ],
+      "start_time": 0.0,
+      "end_time": 1.9
+    },
+    {
+      "id": "seg-2",
+      "text": "I can see it in your eyes",
+      "words": [
+        {"id": "w8", "text": "I", "start_time": 2.0, "end_time": 2.2},
+        {"id": "w9", "text": "can", "start_time": 2.2, "end_time": 2.4},
+        {"id": "w10", "text": "see", "start_time": 2.4, "end_time": 2.6},
+        {"id": "w11", "text": "it", "start_time": 2.6, "end_time": 2.8},
+        {"id": "w12", "text": "in", "start_time": 2.8, "end_time": 3.0},
+        {"id": "w13", "text": "your", "start_time": 3.0, "end_time": 3.2},
+        {"id": "w14", "text": "eyes", "start_time": 3.2, "end_time": 3.5}
+      ],
+      "start_time": 2.0,
+      "end_time": 3.5
+    },
+    {
+      "id": "seg-3",
+      "text": "I can see it in your smile",
+      "words": [
+        {"id": "w15", "text": "I", "start_time": 4.0, "end_time": 4.2},
+        {"id": "w16", "text": "can", "start_time": 4.2, "end_time": 4.4},
+        {"id": "w17", "text": "see", "start_time": 4.4, "end_time": 4.6},
+        {"id": "w18", "text": "it", "start_time": 4.6, "end_time": 4.8},
+        {"id": "w19", "text": "in", "start_time": 4.8, "end_time": 5.0},
+        {"id": "w20", "text": "your", "start_time": 5.0, "end_time": 5.2},
+        {"id": "w21", "text": "smile", "start_time": 5.2, "end_time": 5.5}
+      ],
+      "start_time": 4.0,
+      "end_time": 5.5
+    }
+  ],
+  "reference_lyrics": {
+    "genius": {
+      "segments": [
+        {
+          "id": "ref-seg-1",
+          "text": "Hello, is it me you're looking for?",
+          "words": [
+            {"id": "rw1", "text": "Hello,", "start_time": null, "end_time": null},
+            {"id": "rw2", "text": "is", "start_time": null, "end_time": null},
+            {"id": "rw3", "text": "it", "start_time": null, "end_time": null},
+            {"id": "rw4", "text": "me", "start_time": null, "end_time": null},
+            {"id": "rw5", "text": "you're", "start_time": null, "end_time": null},
+            {"id": "rw6", "text": "looking", "start_time": null, "end_time": null},
+            {"id": "rw7", "text": "for?", "start_time": null, "end_time": null}
+          ],
+          "start_time": null,
+          "end_time": null
+        }
+      ],
+      "metadata": {
+        "source": "genius",
+        "track_name": "Hello",
+        "artist_names": "Lionel Richie",
+        "album_name": "Can't Slow Down",
+        "duration_ms": 280000,
+        "explicit": false,
+        "language": "en",
+        "is_synced": false,
+        "lyrics_provider": "Genius",
+        "lyrics_provider_id": "12345",
+        "provider_metadata": {}
+      },
+      "source": "genius"
+    }
+  },
+  "anchor_sequences": [
+    {
+      "id": "anchor-1",
+      "transcribed_word_ids": ["w1", "w2", "w3", "w4"],
+      "transcription_position": 0,
+      "reference_positions": {"genius": 0},
+      "reference_word_ids": {"genius": ["rw1", "rw2", "rw3", "rw4"]},
+      "confidence": 0.95
+    }
+  ],
+  "gap_sequences": [
+    {
+      "id": "gap-1",
+      "transcribed_word_ids": ["w5", "w6", "w7"],
+      "transcription_position": 4,
+      "preceding_anchor_id": "anchor-1",
+      "following_anchor_id": null,
+      "reference_word_ids": {"genius": ["rw5", "rw6", "rw7"]}
+    }
+  ],
+  "resized_segments": [],
+  "corrections_made": 3,
+  "confidence": 0.85,
+  "corrections": [
+    {
+      "id": "corr-1",
+      "handler": "AgenticCorrector",
+      "original_word": "you're",
+      "corrected_word": "now",
+      "segment_id": "seg-1",
+      "word_id": "w5",
+      "corrected_word_id": "cw5",
+      "source": "genius",
+      "confidence": 0.92,
+      "reason": "Transcription error - misheard word, reference lyrics confirm it should be 'now' [SOUND_ALIKE] (confidence: 92%)",
+      "alternatives": {"your": 0.65},
+      "is_deletion": false,
+      "split_index": null,
+      "split_total": null,
+      "reference_positions": {"genius": 4},
+      "length": 1
+    },
+    {
+      "id": "corr-2",
+      "handler": "AgenticCorrector",
+      "original_word": "it",
+      "corrected_word": "",
+      "segment_id": "seg-2",
+      "word_id": "w11",
+      "corrected_word_id": null,
+      "source": "genius",
+      "confidence": 0.88,
+      "reason": "Extra filler word detected, not in reference lyrics [EXTRA_WORDS] (confidence: 88%)",
+      "alternatives": {},
+      "is_deletion": true,
+      "split_index": null,
+      "split_total": null,
+      "reference_positions": {},
+      "length": 1
+    },
+    {
+      "id": "corr-3",
+      "handler": "AgenticCorrector",
+      "original_word": "I",
+      "corrected_word": "",
+      "segment_id": "seg-3",
+      "word_id": "w15",
+      "corrected_word_id": null,
+      "source": "genius",
+      "confidence": 0.75,
+      "reason": "Extra word at start of line, not in reference [EXTRA_WORDS] (confidence: 75%)",
+      "alternatives": {},
+      "is_deletion": true,
+      "split_index": null,
+      "split_total": null,
+      "reference_positions": {},
+      "length": 1
+    }
+  ],
+  "corrected_segments": [
+    {
+      "id": "seg-1",
+      "text": "Hello, is it me you're looking for?",
+      "words": [
+        {"id": "w1", "text": "Hello,", "start_time": 0.0, "end_time": 0.5},
+        {"id": "w2", "text": "is", "start_time": 0.5, "end_time": 0.7},
+        {"id": "w3", "text": "it", "start_time": 0.7, "end_time": 0.9},
+        {"id": "w4", "text": "me", "start_time": 0.9, "end_time": 1.1},
+        {"id": "cw5", "text": "now", "start_time": 1.1, "end_time": 1.3, "created_during_correction": true},
+        {"id": "w6", "text": "looking", "start_time": 1.3, "end_time": 1.6},
+        {"id": "w7", "text": "for?", "start_time": 1.6, "end_time": 1.9}
+      ],
+      "start_time": 0.0,
+      "end_time": 1.9
+    },
+    {
+      "id": "seg-2",
+      "text": "I can see in your eyes",
+      "words": [
+        {"id": "w8", "text": "I", "start_time": 2.0, "end_time": 2.2},
+        {"id": "w9", "text": "can", "start_time": 2.2, "end_time": 2.4},
+        {"id": "w10", "text": "see", "start_time": 2.4, "end_time": 2.6},
+        {"id": "w12", "text": "in", "start_time": 2.8, "end_time": 3.0},
+        {"id": "w13", "text": "your", "start_time": 3.0, "end_time": 3.2},
+        {"id": "w14", "text": "eyes", "start_time": 3.2, "end_time": 3.5}
+      ],
+      "start_time": 2.0,
+      "end_time": 3.5
+    },
+    {
+      "id": "seg-3",
+      "text": "can see it in your smile",
+      "words": [
+        {"id": "w16", "text": "can", "start_time": 4.2, "end_time": 4.4},
+        {"id": "w17", "text": "see", "start_time": 4.4, "end_time": 4.6},
+        {"id": "w18", "text": "it", "start_time": 4.6, "end_time": 4.8},
+        {"id": "w19", "text": "in", "start_time": 4.8, "end_time": 5.0},
+        {"id": "w20", "text": "your", "start_time": 5.0, "end_time": 5.2},
+        {"id": "w21", "text": "smile", "start_time": 5.2, "end_time": 5.5}
+      ],
+      "start_time": 4.0,
+      "end_time": 5.5
+    }
+  ],
+  "metadata": {
+    "anchor_sequences_count": 1,
+    "gap_sequences_count": 1,
+    "total_words": 21,
+    "correction_ratio": 0.143,
+    "audio_filepath": "/path/to/test-audio.mp3",
+    "audio_hash": "test-audio-hash-123",
+    "available_handlers": [
+      {"id": "agentic", "name": "AgenticCorrector", "description": "AI-powered agentic correction", "enabled": true},
+      {"id": "sound_alike", "name": "SoundAlikeHandler", "description": "Fixes sound-alike errors", "enabled": true}
+    ],
+    "enabled_handlers": ["AgenticCorrector", "SoundAlikeHandler"]
+  },
+  "correction_steps": [],
+  "word_id_map": {
+    "w5": "cw5"
+  },
+  "segment_id_map": {}
+}

{karaoke_gen-0.76.25 → karaoke_gen-0.76.28}/lyrics_transcriber_temp/lyrics_transcriber/frontend/package.json

@@ -11,7 +11,10 @@
     "lint": "eslint .",
     "preview": "vite preview",
     "predeploy": "npm run build-prod",
-    "deploy": "gh-pages -d dist"
+    "deploy": "gh-pages -d dist",
+    "test": "playwright test",
+    "test:ui": "playwright test --ui",
+    "test:headed": "playwright test --headed"
   },
   "dependencies": {
     "@emotion/react": "^11.14.0",

{karaoke_gen-0.76.25 → karaoke_gen-0.76.28}/lyrics_transcriber_temp/lyrics_transcriber/frontend/playwright.config.ts

@@ -61,7 +61,7 @@ export default defineConfig({
 
   // Run local dev server before starting tests
   webServer: {
-    command: 'yarn dev',
+    command: 'npm run dev',
     url: 'http://localhost:5173',
     reuseExistingServer: !process.env.CI,
     timeout: 120000,

{karaoke_gen-0.76.25 → karaoke_gen-0.76.28}/lyrics_transcriber_temp/lyrics_transcriber/frontend/src/components/CorrectedWordWithActions.tsx

@@ -23,6 +23,7 @@ interface CorrectedWordWithActionsProps {
     onClick?: () => void
     backgroundColor?: string
     shouldFlash?: boolean
+    showActions?: boolean  // Controls whether inline action buttons are visible
 }
 
 const WordContainer = styled(Box, {
@@ -98,7 +99,8 @@ export default function CorrectedWordWithActions({
     onAccept,
     onClick,
     backgroundColor,
-    shouldFlash
+    shouldFlash,
+    showActions = true
 }: CorrectedWordWithActionsProps) {
     const theme = useTheme()
     const isMobile = useMediaQuery(theme.breakpoints.down('sm'))
@@ -115,7 +117,7 @@ export default function CorrectedWordWithActions({
             onClick={onClick}
         >
             <OriginalWordLabel>{originalWord}</OriginalWordLabel>
-            
+
             <Box
                 component="span"
                 sx={{
@@ -127,40 +129,42 @@ export default function CorrectedWordWithActions({
                 {word}
             </Box>
 
-            <ActionsContainer>
-                <Tooltip title="Revert to original" placement="top" arrow>
-                    <ActionButton
-                        size="small"
-                        onClick={(e) => handleAction(e, onRevert)}
-                        aria-label="revert correction"
-                    >
-                        <UndoIcon />
-                    </ActionButton>
-                </Tooltip>
-
-                <Tooltip title="Edit correction" placement="top" arrow>
-                    <ActionButton
-                        size="small"
-                        onClick={(e) => handleAction(e, onEdit)}
-                        aria-label="edit correction"
-                    >
-                        <EditIcon />
-                    </ActionButton>
-                </Tooltip>
+            {showActions && (
+                <ActionsContainer>
+                    <Tooltip title="Revert to original" placement="top" arrow>
+                        <ActionButton
+                            size="small"
+                            onClick={(e) => handleAction(e, onRevert)}
+                            aria-label="revert correction"
+                        >
+                            <UndoIcon />
+                        </ActionButton>
+                    </Tooltip>
 
-                {!isMobile && (
-                    <Tooltip title="Accept correction" placement="top" arrow>
+                    <Tooltip title="Edit correction" placement="top" arrow>
                         <ActionButton
                             size="small"
-                            onClick={(e) => handleAction(e, onAccept)}
-                            aria-label="accept correction"
-                            sx={{ color: 'success.main' }}
+                            onClick={(e) => handleAction(e, onEdit)}
+                            aria-label="edit correction"
                         >
-                            <CheckCircleOutlineIcon />
+                            <EditIcon />
                         </ActionButton>
                     </Tooltip>
-                )}
-            </ActionsContainer>
+
+                    {!isMobile && (
+                        <Tooltip title="Accept correction" placement="top" arrow>
+                            <ActionButton
+                                size="small"
+                                onClick={(e) => handleAction(e, onAccept)}
+                                aria-label="accept correction"
+                                sx={{ color: 'success.main' }}
+                            >
+                                <CheckCircleOutlineIcon />
+                            </ActionButton>
+                        </Tooltip>
+                    )}
+                </ActionsContainer>
+            )}
         </WordContainer>
     )
 }