lyrics-transcriber 0.41.0__py3-none-any.whl → 0.43.0__py3-none-any.whl

lyrics_transcriber/correction/handlers/levenshtein.py

@@ -38,23 +38,38 @@ class LevenshteinHandler(GapCorrectionHandler):
         self.similarity_threshold = similarity_threshold
         self.logger = logger or logging.getLogger(__name__)
 
-    def can_handle(self, gap: GapSequence) -> Tuple[bool, Dict[str, Any]]:
+    def can_handle(self, gap: GapSequence, data: Optional[Dict[str, Any]] = None) -> Tuple[bool, Dict[str, Any]]:
         """Check if we can handle this gap - we'll try if there are reference words."""
-        if not gap.reference_words:
+        if not data or "word_map" not in data:
+            self.logger.error("No word_map provided in data")
+            return False, {}
+
+        word_map = data["word_map"]
+
+        if not gap.reference_word_ids:
             self.logger.debug("No reference words available")
             return False, {}
 
-        if not gap.words:
+        if not gap.transcribed_word_ids:
             self.logger.debug("No gap words available")
             return False, {}
 
         # Check if any word has sufficient similarity to reference
-        for i, word in enumerate(gap.words):
-            for ref_words in gap.reference_words.values():
-                if i < len(ref_words):
-                    similarity = self._get_string_similarity(word, ref_words[i])
+        for i, word_id in enumerate(gap.transcribed_word_ids):
+            if word_id not in word_map:
+                continue
+            word = word_map[word_id]
+
+            for source, ref_word_ids in gap.reference_word_ids.items():
+                if i < len(ref_word_ids):
+                    ref_word_id = ref_word_ids[i]
+                    if ref_word_id not in word_map:
+                        continue
+                    ref_word = word_map[ref_word_id]
+
+                    similarity = self._get_string_similarity(word.text, ref_word.text)
                     if similarity >= self.similarity_threshold:
-                        self.logger.debug(f"Found similar word: '{word}' -> '{ref_words[i]}' ({similarity:.2f})")
+                        self.logger.debug(f"Found similar word: '{word.text}' -> '{ref_word.text}' ({similarity:.2f})")
                         return True, {}
 
         self.logger.debug("No words meet similarity threshold")
@@ -62,58 +77,85 @@ class LevenshteinHandler(GapCorrectionHandler):
 
     def handle(self, gap: GapSequence, data: Optional[Dict[str, Any]] = None) -> List[WordCorrection]:
         """Try to correct words based on string similarity."""
+        if not data or "word_map" not in data:
+            self.logger.error("No word_map provided in data")
+            return []
+
+        word_map = data["word_map"]
         corrections = []
 
         # Process each word in the gap
-        for i, word in enumerate(gap.words):
+        for i, word_id in enumerate(gap.transcribed_word_ids):
+            if word_id not in word_map:
+                continue
+            word = word_map[word_id]
+
             # Skip if word is empty or just punctuation
-            if not word.strip():
+            if not word.text.strip():
                 continue
 
             # Skip exact matches
-            if any(i < len(ref_words) and word.lower() == ref_words[i].lower() for ref_words in gap.reference_words.values()):
-                self.logger.debug(f"Skipping exact match: '{word}'")
+            exact_match = False
+            for source, ref_word_ids in gap.reference_word_ids.items():
+                if i < len(ref_word_ids):
+                    ref_word_id = ref_word_ids[i]
+                    if ref_word_id in word_map:
+                        ref_word = word_map[ref_word_id]
+                        if word.text.lower() == ref_word.text.lower():
+                            exact_match = True
+                            break
+            if exact_match:
                 continue
 
             # Find matching reference words at this position
-            matches = {}  # word -> (sources, similarity)
-            for source, ref_words in gap.reference_words.items():
-                ref_words_original = gap.reference_words_original[source]  # Get original formatted words
-                if i >= len(ref_words):
+            matches: Dict[str, Tuple[List[str], float, str]] = {}  # word -> (sources, similarity, word_id)
+
+            for source, ref_word_ids in gap.reference_word_ids.items():
+                if i >= len(ref_word_ids):
                     continue
 
-                ref_word = ref_words[i]
-                ref_word_original = ref_words_original[i]  # Get original formatted word
-                similarity = self._get_string_similarity(word, ref_word)
+                ref_word_id = ref_word_ids[i]
+                if ref_word_id not in word_map:
+                    continue
+                ref_word = word_map[ref_word_id]
+
+                similarity = self._get_string_similarity(word.text, ref_word.text)
 
                 if similarity >= self.similarity_threshold:
-                    self.logger.debug(f"Found match: '{word}' -> '{ref_word}' ({similarity:.2f})")
-                    if ref_word_original not in matches:  # Use original formatted word as key
-                        matches[ref_word_original] = ([], similarity)
-                    matches[ref_word_original][0].append(source)
+                    self.logger.debug(f"Found match: '{word.text}' -> '{ref_word.text}' ({similarity:.2f})")
+                    if ref_word.text not in matches:
+                        matches[ref_word.text] = ([], similarity, ref_word_id)
+                    matches[ref_word.text][0].append(source)
 
             # Create correction for best match if any found
             if matches:
-                best_match, (sources, similarity) = max(
+                best_match, (sources, similarity, ref_word_id) = max(
                     matches.items(), key=lambda x: (len(x[1][0]), x[1][1])  # Sort by number of sources, then similarity
                 )
 
-                source_confidence = len(sources) / len(gap.reference_words)
+                source_confidence = len(sources) / len(gap.reference_word_ids)
                 final_confidence = similarity * source_confidence
 
-                # Calculate reference positions for matching sources
-                reference_positions = WordOperations.calculate_reference_positions(gap, sources)
+                # Calculate reference positions
+                reference_positions = WordOperations.calculate_reference_positions(gap, anchor_sequences=data.get("anchor_sequences", []))
 
-                self.logger.debug(f"Creating correction: {word} -> {best_match} (confidence: {final_confidence})")
+                self.logger.debug(f"Creating correction: {word.text} -> {best_match} (confidence: {final_confidence})")
                 corrections.append(
-                    WordOperations.create_word_replacement_correction(
-                        original_word=word,
-                        corrected_word=best_match,  # Using original formatted word
+                    WordCorrection(
+                        original_word=word.text,
+                        corrected_word=best_match,
+                        segment_index=0,
                         original_position=gap.transcription_position + i,
-                        source=", ".join(sources),
                         confidence=final_confidence,
-                        reason=f"LevenshteinHandler: String similarity ({final_confidence:.2f})",
+                        source=", ".join(sources),
+                        reason=f"String similarity ({final_confidence:.2f})",
+                        alternatives={k: len(v[0]) for k, v in matches.items()},
+                        is_deletion=False,
                         reference_positions=reference_positions,
+                        length=1,
+                        handler="LevenshteinHandler",
+                        word_id=word_id,
+                        corrected_word_id=ref_word_id,
                     )
                 )
 

lyrics_transcriber/correction/handlers/llm.py

@@ -0,0 +1,290 @@
+from typing import List, Optional, Tuple, Dict, Any, Union
+import logging
+import json
+from ollama import chat
+from datetime import datetime
+from pathlib import Path
+
+from lyrics_transcriber.types import GapSequence, WordCorrection
+from lyrics_transcriber.correction.handlers.base import GapCorrectionHandler
+from lyrics_transcriber.correction.handlers.word_operations import WordOperations
+
+
+class LLMHandler(GapCorrectionHandler):
+    """Uses an LLM to analyze and correct gaps by comparing with reference lyrics."""
+
+    def __init__(self, logger: Optional[logging.Logger] = None, cache_dir: Optional[Union[str, Path]] = None):
+        super().__init__(logger)
+        self.logger = logger or logging.getLogger(__name__)
+        self.model = "deepseek-r1:7b"
+        self.cache_dir = Path(cache_dir) if cache_dir else None
+
+    def _format_prompt(self, gap: GapSequence, data: Optional[Dict[str, Any]] = None) -> str:
+        """Format the prompt for the LLM with context about the gap and reference lyrics."""
+        word_map = data.get("word_map", {})
+        metadata = data.get("metadata", {}) if data else {}
+
+        if not word_map:
+            self.logger.error("No word_map provided in data")
+            return ""
+
+        # Format transcribed words with their IDs
+        transcribed_words = [{"id": word_id, "text": word_map[word_id].text} for word_id in gap.transcribed_word_ids if word_id in word_map]
+
+        prompt = (
+            "You are a lyrics correction expert. You will be given transcribed lyrics that may contain errors "
+            "and reference lyrics from multiple sources. Your task is to analyze each word in the transcribed text "
+            "and suggest specific corrections based on the reference lyrics.\n\n"
+            "Each word has a unique ID. When suggesting corrections, you must specify the ID of the word being corrected. "
+            "This ensures accuracy in applying your corrections.\n\n"
+            "For each correction, specify:\n"
+            "1. The word ID being corrected\n"
+            "2. The correction type ('replace', 'split', 'combine', or 'delete')\n"
+            "3. The corrected text\n"
+            "4. Your confidence level\n"
+            "5. The reason for the correction\n\n"
+        )
+
+        # Add song context if available
+        if metadata and metadata.get("artist") and metadata.get("title"):
+            prompt += f"Song: {metadata['title']}\nArtist: {metadata['artist']}\n\n"
+
+        # Format transcribed words with IDs
+        prompt += "Transcribed words:\n"
+        for word in transcribed_words:
+            prompt += f"- ID: {word['id']}, Text: '{word['text']}'\n"
+
+        prompt += "\nReference lyrics from different sources:\n"
+
+        # Add each reference source with words and their IDs
+        for source, word_ids in gap.reference_word_ids.items():
+            reference_words = [{"id": word_id, "text": word_map[word_id].text} for word_id in word_ids if word_id in word_map]
+            prompt += f"\n{source} immediate context:\n"
+            for word in reference_words:
+                prompt += f"- ID: {word['id']}, Text: '{word['text']}'\n"
+
+            # Add full lyrics if available
+            if metadata and metadata.get("full_reference_texts", {}).get(source):
+                prompt += f"\nFull {source} lyrics:\n{metadata['full_reference_texts'][source]}\n"
+
+        # Add context about surrounding anchors if available
+        if gap.preceding_anchor_id:
+            preceding_anchor = next((a.anchor for a in data.get("anchor_sequences", []) if a.anchor.id == gap.preceding_anchor_id), None)
+            if preceding_anchor:
+                anchor_words = [
+                    {"id": word_id, "text": word_map[word_id].text}
+                    for word_id in preceding_anchor.transcribed_word_ids
+                    if word_id in word_map
+                ]
+                prompt += "\nPreceding correct words:\n"
+                for word in anchor_words:
+                    prompt += f"- ID: {word['id']}, Text: '{word['text']}'\n"
+
+        prompt += (
+            "\nProvide corrections in the following JSON format:\n"
+            "{\n"
+            '  "corrections": [\n'
+            "    {\n"
+            '      "word_id": "id_of_word_to_correct",\n'
+            '      "type": "replace|split|combine|delete",\n'
+            '      "corrected_text": "new text",\n'
+            '      "reference_word_id": "id_from_reference_lyrics",  // Optional, use when matching a specific reference word\n'
+            '      "confidence": 0.9,\n'
+            '      "reason": "explanation of correction"\n'
+            "    }\n"
+            "  ]\n"
+            "}\n\n"
+            "Important rules:\n"
+            "1. Always include the word_id for each correction\n"
+            "2. For 'split' type, corrected_text should contain the space-separated words\n"
+            "3. For 'combine' type, word_id should be the first word to combine\n"
+            "4. Include reference_word_id when the correction matches a specific reference word\n"
+            "5. Only suggest corrections when you're confident they improve the lyrics\n"
+            "6. Preserve any existing words that match the reference lyrics\n"
+            "7. Respond ONLY with the JSON object, no other text"
+        )
+
+        return prompt
+
+    def can_handle(self, gap: GapSequence, data: Optional[Dict[str, Any]] = None) -> Tuple[bool, Dict[str, Any]]:
+        """LLM handler can attempt to handle any gap with reference words."""
+        if not gap.reference_word_ids:
+            self.logger.debug("No reference words available")
+            return False, {}
+
+        return True, {}
+
+    def _write_debug_info(self, prompt: str, response: str, gap_index: int, audio_file_hash: Optional[str] = None) -> None:
+        """Write prompt and response to debug files."""
+        if not self.cache_dir:
+            self.logger.warning("No cache directory provided, skipping LLM debug output")
+            return
+
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        debug_dir = self.cache_dir / "llm_debug"
+        debug_dir.mkdir(exist_ok=True, parents=True)
+
+        hash_prefix = f"{audio_file_hash}_" if audio_file_hash else ""
+        filename = debug_dir / f"llm_debug_{hash_prefix}{gap_index}_{timestamp}.txt"
+
+        debug_content = "=== LLM PROMPT ===\n" f"{prompt}\n\n" "=== LLM RESPONSE ===\n" f"{response}\n"
+
+        try:
+            with open(filename, "w", encoding="utf-8") as f:
+                f.write(debug_content)
+        except IOError as e:
+            self.logger.error(f"Failed to write LLM debug file: {e}")
+
+    def handle(self, gap: GapSequence, data: Optional[Dict[str, Any]] = None) -> List[WordCorrection]:
+        """Process the gap using the LLM and create corrections based on its response."""
+        if not data or "word_map" not in data:
+            self.logger.error("No word_map provided in data")
+            return []
+
+        word_map = data["word_map"]
+        transcribed_words = [word_map[word_id].text for word_id in gap.transcribed_word_ids if word_id in word_map]
+
+        # Calculate reference positions using the centralized method
+        reference_positions = (
+            WordOperations.calculate_reference_positions(gap, anchor_sequences=data.get("anchor_sequences", [])) or {}
+        )  # Ensure empty dict if None
+
+        prompt = self._format_prompt(gap, data)
+        if not prompt:
+            return []
+
+        # Get a unique index for this gap based on its position
+        gap_index = gap.transcription_position
+
+        try:
+            self.logger.debug(f"Processing gap words: {transcribed_words}")
+            self.logger.debug(f"Reference word IDs: {gap.reference_word_ids}")
+
+            response = chat(model=self.model, messages=[{"role": "user", "content": prompt}], format="json")
+
+            # Write debug info to files
+            self._write_debug_info(prompt, response.message.content, gap_index, audio_file_hash=data.get("audio_file_hash"))
+
+            try:
+                corrections_data = json.loads(response.message.content)
+            except json.JSONDecodeError as e:
+                self.logger.error(f"Failed to parse LLM response as JSON: {e}")
+                self.logger.error(f"Raw response content: {response.message.content}")
+                return []
+
+            # Check if corrections exist and are non-empty
+            if not corrections_data.get("corrections"):
+                self.logger.debug("No corrections suggested by LLM")
+                return []
+
+            corrections = []
+            for correction in corrections_data["corrections"]:
+                # Validate word_id exists in gap
+                if correction["word_id"] not in gap.transcribed_word_ids:
+                    self.logger.error(f"LLM suggested correction for word_id {correction['word_id']} which is not in the gap")
+                    continue
+
+                # Get original word from word map
+                original_word = word_map[correction["word_id"]]
+                position = gap.transcription_position + gap.transcribed_word_ids.index(correction["word_id"])
+
+                self.logger.debug(f"Processing correction: {correction}")
+
+                if correction["type"] == "replace":
+                    self.logger.debug(
+                        f"Creating replacement: '{original_word.text}' -> '{correction['corrected_text']}' " f"at position {position}"
+                    )
+                    corrections.append(
+                        WordOperations.create_word_replacement_correction(
+                            original_word=original_word.text,
+                            corrected_word=correction["corrected_text"],
+                            original_position=position,
+                            source="LLM",
+                            confidence=correction["confidence"],
+                            reason=correction["reason"],
+                            handler="LLMHandler",
+                            reference_positions=reference_positions,
+                            original_word_id=correction["word_id"],
+                            corrected_word_id=correction.get("reference_word_id"),
+                        )
+                    )
+                elif correction["type"] == "split":
+                    split_words = correction["corrected_text"].split()
+                    self.logger.debug(f"Creating split: '{original_word.text}' -> {split_words} " f"at position {position}")
+
+                    # Get reference word IDs if provided
+                    reference_word_ids = correction.get("reference_word_ids", [None] * len(split_words))
+
+                    corrections.extend(
+                        WordOperations.create_word_split_corrections(
+                            original_word=original_word.text,
+                            reference_words=split_words,
+                            original_position=position,
+                            source="LLM",
+                            confidence=correction["confidence"],
+                            reason=correction["reason"],
+                            handler="LLMHandler",
+                            reference_positions=reference_positions,
+                            original_word_id=correction["word_id"],
+                            corrected_word_ids=reference_word_ids,
+                        )
+                    )
+                elif correction["type"] == "combine":
+                    # Get all word IDs to combine
+                    word_ids_to_combine = []
+                    current_idx = gap.transcribed_word_ids.index(correction["word_id"])
+                    words_needed = len(correction["corrected_text"].split())
+
+                    if current_idx + words_needed <= len(gap.transcribed_word_ids):
+                        word_ids_to_combine = gap.transcribed_word_ids[current_idx : current_idx + words_needed]
+                    else:
+                        self.logger.error(f"Not enough words available to combine at position {position}")
+                        continue
+
+                    words_to_combine = [word_map[word_id].text for word_id in word_ids_to_combine]
+
+                    self.logger.debug(
+                        f"Creating combine: {words_to_combine} -> '{correction['corrected_text']}' " f"at position {position}"
+                    )
+
+                    corrections.extend(
+                        WordOperations.create_word_combine_corrections(
+                            original_words=words_to_combine,
+                            reference_word=correction["corrected_text"],
+                            original_position=position,
+                            source="LLM",
+                            confidence=correction["confidence"],
+                            combine_reason=correction["reason"],
+                            delete_reason=f"Part of combining words: {correction['reason']}",
+                            handler="LLMHandler",
+                            reference_positions=reference_positions,
+                            original_word_ids=word_ids_to_combine,
+                            corrected_word_id=correction.get("reference_word_id"),
+                        )
+                    )
+                elif correction["type"] == "delete":
+                    self.logger.debug(f"Creating deletion: '{original_word.text}' at position {position}")
+                    corrections.append(
+                        WordCorrection(
+                            original_word=original_word.text,
+                            corrected_word="",
+                            segment_index=0,
+                            original_position=position,
+                            confidence=correction["confidence"],
+                            source="LLM",
+                            reason=correction["reason"],
+                            alternatives={},
+                            is_deletion=True,
+                            handler="LLMHandler",
+                            reference_positions=reference_positions,
+                            word_id=correction["word_id"],
+                            corrected_word_id=None,  # Deleted words don't need a corrected ID
+                        )
+                    )
+
+            self.logger.debug(f"Created {len(corrections)} corrections: {[f'{c.original_word}->{c.corrected_word}' for c in corrections]}")
+            return corrections
+
+        except Exception as e:
+            self.logger.error(f"Unexpected error in LLM handler: {e}")
+            return []

lyrics_transcriber/correction/handlers/no_space_punct_match.py

@@ -20,88 +20,133 @@ class NoSpacePunctuationMatchHandler(GapCorrectionHandler):
         # Remove all punctuation including apostrophes
         return re.sub(r"[^\w\s]", "", text)
 
-    def can_handle(self, gap: GapSequence) -> Tuple[bool, Dict[str, Any]]:
+    def can_handle(self, gap: GapSequence, data: Optional[Dict[str, Any]] = None) -> Tuple[bool, Dict[str, Any]]:
         # Must have reference words
-        if not gap.reference_words:
-            self.logger.debug("No reference words available.")
+        if not gap.reference_word_ids:
+            self.logger.debug("No reference word IDs available.")
             return False, {}
 
+        # Get word lookup map from data
+        if not data or "word_map" not in data:
+            self.logger.error("No word_map provided in data")
+            return False, {}
+
+        word_map = data["word_map"]
+
+        # Get the actual words from word IDs
+        gap_words = []
+        for word_id in gap.transcribed_word_ids:
+            if word_id not in word_map:
+                self.logger.error(f"Word ID {word_id} not found in word_map")
+                return False, {}
+            gap_words.append(word_map[word_id].text)
+
         # Get the gap text without spaces and punctuation
-        gap_text = self._remove_spaces_and_punct(gap.words)
+        gap_text = self._remove_spaces_and_punct(gap_words)
 
         # Check if any reference source matches when spaces and punctuation are removed
-        for words in gap.reference_words.values():
-            ref_text = self._remove_spaces_and_punct(words)
+        for source, ref_word_ids in gap.reference_word_ids.items():
+            ref_words = []
+            for word_id in ref_word_ids:
+                if word_id not in word_map:
+                    self.logger.error(f"Reference word ID {word_id} not found in word_map")
+                    continue
+                ref_words.append(word_map[word_id].text)
+
+            if not ref_words:
+                continue
+
+            ref_text = self._remove_spaces_and_punct(ref_words)
             if gap_text == ref_text:
                 self.logger.debug("Found a matching reference source with spaces and punctuation removed.")
-                return True, {}
+                return True, {
+                    "matching_source": source,
+                    "reference_word_ids": ref_word_ids,
+                    "word_map": word_map,
+                }
 
         self.logger.debug("No matching reference source found with spaces and punctuation removed.")
         return False, {}
 
     def handle(self, gap: GapSequence, data: Optional[Dict[str, Any]] = None) -> List[WordCorrection]:
-        corrections = []
+        """Handle the gap using no-space punctuation matching."""
+        if not data:
+            can_handle, data = self.can_handle(gap)
+            if not can_handle:
+                return []
 
-        # Find the matching source (we know there is at least one from can_handle)
-        gap_text = self._remove_spaces_and_punct(gap.words)
-        matching_source = None
-        reference_words = None
-        reference_words_original = None
-        for source, words in gap.reference_words.items():
-            if self._remove_spaces_and_punct(words) == gap_text:
-                matching_source = source
-                reference_words = words
-                reference_words_original = gap.reference_words_original[source]
-                self.logger.debug(f"Using source '{source}' for corrections.")
-                break
+        corrections = []
+        matching_source = data["matching_source"]
+        reference_word_ids = data["reference_word_ids"]
+        word_map = data["word_map"]
 
         # Calculate reference positions for the matching source
         reference_positions = WordOperations.calculate_reference_positions(gap, [matching_source])
 
         # Handle cases where number of words differ
-        if len(gap.words) > len(reference_words):
+        if len(gap.transcribed_word_ids) > len(reference_word_ids):
             # Multiple transcribed words -> fewer reference words
+            # Get the actual words from word IDs
+            gap_words = [word_map[word_id].text for word_id in gap.transcribed_word_ids]
+            ref_word = word_map[reference_word_ids[0]].text
+
             corrections.extend(
                 WordOperations.create_word_combine_corrections(
-                    original_words=gap.words,
-                    reference_word=reference_words_original[0],
+                    original_words=gap_words,
+                    reference_word=ref_word,
                     original_position=gap.transcription_position,
                     source=matching_source,
                     confidence=1.0,
-                    combine_reason="NoSpacePunctuationMatchHandler: Words combined based on text match",
-                    delete_reason="NoSpacePunctuationMatchHandler: Word removed as part of text match combination",
+                    combine_reason="Words combined based on text match",
+                    delete_reason="Word removed as part of text match combination",
                     reference_positions=reference_positions,
+                    handler="NoSpacePunctuationMatchHandler",
+                    original_word_ids=gap.transcribed_word_ids,
+                    corrected_word_id=reference_word_ids[0],  # Use the reference word's ID
                 )
             )
-            self.logger.debug(f"Combined words into '{reference_words_original[0]}'.")
+            self.logger.debug(f"Combined words into '{ref_word}'.")
 
-        elif len(gap.words) < len(reference_words):
+        elif len(gap.transcribed_word_ids) < len(reference_word_ids):
             # Single transcribed word -> multiple reference words
+            # Get the actual words
+            gap_word = word_map[gap.transcribed_word_ids[0]].text
+            ref_words = [word_map[word_id].text for word_id in reference_word_ids]
+
             corrections.extend(
                 WordOperations.create_word_split_corrections(
-                    original_word=gap.words[0],
-                    reference_words=reference_words_original,
+                    original_word=gap_word,
+                    reference_words=ref_words,
                     original_position=gap.transcription_position,
                     source=matching_source,
                     confidence=1.0,
-                    reason="NoSpacePunctuationMatchHandler: Split word based on text match",
+                    reason="Split word based on text match",
                     reference_positions=reference_positions,
+                    handler="NoSpacePunctuationMatchHandler",
+                    original_word_id=gap.transcribed_word_ids[0],
+                    corrected_word_ids=reference_word_ids,  # Use the reference word IDs
                 )
             )
-            self.logger.debug(f"Split word '{gap.words[0]}' into {reference_words_original}.")
+            self.logger.debug(f"Split word '{gap_word}' into {ref_words}.")
 
         else:
             # One-to-one replacement
-            for i, (orig_word, ref_word, ref_word_original) in enumerate(zip(gap.words, reference_words, reference_words_original)):
-                if orig_word.lower() != ref_word.lower():
+            for i, (orig_word_id, ref_word_id) in enumerate(zip(gap.transcribed_word_ids, reference_word_ids)):
+                orig_word = word_map[orig_word_id]
+                ref_word = word_map[ref_word_id]
+
+                if orig_word.text.lower() != ref_word.text.lower():
                     correction = WordOperations.create_word_replacement_correction(
-                        original_word=orig_word,
-                        corrected_word=ref_word_original,
+                        original_word=orig_word.text,
+                        corrected_word=ref_word.text,
                         original_position=gap.transcription_position + i,
                         source=matching_source,
                         confidence=1.0,
-                        reason=f"NoSpacePunctuationMatchHandler: Source '{matching_source}' matched when spaces and punctuation removed",
+                        reason=f"Source '{matching_source}' matched when spaces and punctuation removed",
                         reference_positions=reference_positions,
+                        handler="NoSpacePunctuationMatchHandler",
+                        original_word_id=orig_word_id,
+                        corrected_word_id=ref_word_id,
                     )
                     corrections.append(correction)
                     self.logger.debug(f"Correction made: {correction}")