Update video2.py

video2.py

@@ -32,54 +32,28 @@ def audio_func(id,lines):
   return duration,audio_path
     
 # --- CONFIGURATION ---
-def video_func(id,lines):
-  print(id,lines[id])
-  duration,audio_path = audio_func(id,lines)
-  IMAGE_PATH = os.path.join(IMAGE_DIR,f"slide{id}.png")   # Ensure this path is correct
-  VIDEO_DURATION = duration               # seconds
-  HIGHLIGHT_COLOR = (255, 255, 0)   # Yellow highlight
-  HIGHLIGHT_OPACITY = 0.5           # Semi-transparent
+def video_func(id, lines):
+    duration, audio_path = audio_func(id, lines)
+    image_path = os.path.join(IMAGE_DIR, f"slide{id}.png")
+    img = Image.open(image_path)
+    data = pytesseract.image_to_data(img, output_type=pytesseract.Output.DICT)
 
-  # --- OCR STEP ---
-  img = Image.open(IMAGE_PATH)
-  data = pytesseract.image_to_data(img, output_type=pytesseract.Output.DICT)
+    words = []
+    for i in range(len(data['text'])):
+        txt = data['text'][i].strip()
+        if txt and int(data['conf'][i]) > 60:
+            box = (
+                data['left'][i],
+                data['top'][i],
+                data['width'][i],
+                data['height'][i],
+            )
+            words.append((txt, box))
 
-  # Extract words and their positions
-  words = []
-  for i in range(len(data['text'])):
-      word = data['text'][i].strip()
-      if word and int(data['conf'][i]) > 60:
-          x, y, w, h = data['left'][i], data['top'][i], data['width'][i], data['height'][i]
-          words.append({'text': word, 'box': (x, y, w, h)})
-
-  # --- BASE IMAGE CLIP ---
-  image_clip = ImageClip(IMAGE_PATH).set_duration(VIDEO_DURATION)
-
-  # --- HIGHLIGHT WORDS ONE BY ONE ---
-  n_words = len(words)
-  highlight_duration = VIDEO_DURATION / n_words
-
-  highlight_clips = []
-
-  for i, word in enumerate(words):
-      x, y, w, h = word['box']
-      start = i * highlight_duration
-      end = start + highlight_duration
-
-      # Create highlight rectangle
-      rect = ColorClip(size=(w, h), color=HIGHLIGHT_COLOR)
-      rect = rect.set_opacity(HIGHLIGHT_OPACITY).set_position((x, y)).set_start(start).set_end(end)
-
-      highlight_clips.append(rect)
-
-  # --- FINAL VIDEO --
-
-  final_clip = CompositeVideoClip([image_clip] + highlight_clips)
-  audio = AudioFileClip(audio_path)
-  final_clip = final_clip.set_audio(audio)
-  clip_name = "clip"+str(id)+".mp4"
-  video_path=os.path.join(CLIPS_DIR,clip_name)
-  final_clip.write_videofile(video_path, fps=24,audio=True)
+    clip_file = rust_highlight.render_video(
+        id, image_path, audio_path, duration, words
+    )
+    print(f"Created {clip_file}")
 
 def video_com(lines):
     video_path = f"/tmp/video_{uuid.uuid4().hex}.mp4"