Update utils/subtitle_engine.py

utils/subtitle_engine.py

@@ -1,31 +1,65 @@
+import re
+from .logger import logger
+
 class SubtitleEngine:
     def __init__(self, words_per_frame=4):
+        """
+        words_per_frame: Number of words visible on screen at once.
+        TikTok style usually favors 3-5 words for readability.
+        """
         self.words_per_frame = words_per_frame
 
     def create_highlight_chunks(self, segments):
         """
-        Creates frames where 3-5 words are visible, but the current 
-        spoken word is highlighted.
+        Processes faster-whisper segments into frames optimized for 
+        Pango highlighting and rhythmic 'pop' effects.
         """
         highlight_frames = []
+        
         for segment in segments:
             if not hasattr(segment, 'words') or not segment.words:
                 continue
             
             words = segment.words
-            # Break words into frames of size N
+            
+            # Group words into frames of size N (default 4)
             for i in range(0, len(words), self.words_per_frame):
                 frame_words = words[i : i + self.words_per_frame]
                 
-                # For each word in the frame, create a unique timestamp entry
-                # where that specific word is the "active" one
+                # Each word in this frame needs its own 'frame' entry in the 
+                # renderer so that the highlight moves as the word is spoken.
                 for active_index, active_word in enumerate(frame_words):
+                    
+                    # Ensure start/end times are logical
+                    start = active_word.start
+                    end = active_word.end
+                    
+                    # If it's the last word in the frame, we can hold it 
+                    # slightly longer or until the next frame starts
+                    if active_index == len(frame_words) - 1:
+                        # Small buffer to prevent flicker between frames
+                        end += 0.1 
+
                     highlight_frames.append({
-                        "full_frame_text": [w.word.strip().upper() for w in frame_words],
+                        "full_frame_text": [self._clean_text(w.word) for w in frame_words],
                         "active_word_index": active_index,
-                        "start": active_word.start,
-                        "end": active_word.end
+                        "start": start,
+                        "end": end
                     })
+        
+        logger.info(f"Generated {len(highlight_frames)} highlight frames for rendering.")
         return highlight_frames
 
-sub_engine = SubtitleEngine()
+    def _clean_text(self, text):
+        """
+        Standardizes text for TikTok style: 
+        - All caps
+        - No leading/trailing whitespace
+        - Strips special characters that might break Pango markup
+        """
+        clean = text.strip().upper()
+        # Escaping characters that break XML/Pango if necessary
+        clean = clean.replace("&", "&amp;").replace("<", "&lt;").replace(">", "&gt;")
+        return clean
+
+sub_engine = SubtitleEngine(words_per_frame=4)