Update rust_highlight/src/lib.rs

rust_highlight/src/lib.rs

@@ -1,7 +1,10 @@
+```rust
 // rust_highlight/src/lib.rs
 use pyo3::prelude::*;
+use pyo3::types::PyModule;
 use opencv::core::{Mat, Point, Scalar, CV_8UC3};
 use opencv::imgproc::{circle, get_text_size, line, put_text, HersheyFonts, LineTypes};
+use opencv::prelude::*;
 use std::process::{Command, Stdio};
 use std::io::Write;
 use std::time::Instant;
@@ -47,7 +50,7 @@ fn generate_video_clip(id: usize, text: String, audio_path: String, duration: f6
 
     // Wrap text
     let text_area_width = width - 2 * margin_x;
-    let (wrapped_lines, line_styles) = wrap_text_cv(&text, font, default_font_scale, default_thickness, text_area_width);
+    let (wrapped_lines, line_styles) = wrap_text_cv(&text, font, default_font_scale, default_thickness, text_area_width, header_font_scale, header_thickness);
 
     let full_text = wrapped_lines.join("\n");
     if full_text.is_empty() {
@@ -139,7 +142,6 @@ fn generate_video_clip(id: usize, text: String, audio_path: String, duration: f6
                     anim_offset,
                     width as i32,
                     height as i32,
-                    &wrapped_lines,
                     &line_styles,
                     &y_positions,
                     margin_x,
@@ -213,17 +215,17 @@ fn generate_video_clip(id: usize, text: String, audio_path: String, duration: f6
     Ok(Some(final_video_path))
 }
 
-fn wrap_text_cv(text: &str, font: i32, default_font_scale: f64, default_thickness: i32, max_width: i32) -> (Vec<String>, Vec<bool>) {
+fn wrap_text_cv(text: &str, font: i32, default_font_scale: f64, default_thickness: i32, max_width: i32, header_font_scale: f64, header_thickness: i32) -> (Vec<String>, Vec<bool>) {
     let mut wrapped_lines: Vec<String> = Vec::new();
     let mut styles: Vec<bool> = Vec::new();
     for para in text.lines() {
         let trimmed = para.trim_start();
-        let mut is_header = trimmed.starts_with("###");
+        let is_header = trimmed.starts_with("###");
         let mut para_str = para.to_string();
         if is_header {
             para_str = trimmed[3..].trim().to_string();
         }
-        let font_scale = if is_header { header_font_scale } else { default_font_scale };  // Use globals or pass
+        let font_scale = if is_header { header_font_scale } else { default_font_scale };
         let thickness = if is_header { header_thickness } else { default_thickness };
         if para_str.is_empty() {
             wrapped_lines.push("".to_string());
@@ -242,7 +244,6 @@ fn wrap_text_cv(text: &str, font: i32, default_font_scale: f64, default_thicknes
                 if !cur.is_empty() {
                     wrapped_lines.push(cur);
                     styles.push(is_header);
-                    cur = String::new();
                 }
                 let mut base_line_single = 0;
                 let size_single = get_text_size(w, font, font_scale, thickness, &mut base_line_single).unwrap();
@@ -338,7 +339,8 @@ fn render_frame(
 }
 
 #[pymodule]
-fn rust_highlight(_py: Python, m: &PyModule) -> PyResult<()> {
+fn rust_highlight(py: Python<'_>, m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_function(wrap_pyfunction!(generate_video_clip, m)?)?;
     Ok(())
-}
+}
+```