Update rust_highlight/src/lib.rs

rust_highlight/src/lib.rs

@@ -1,20 +1,28 @@
+use pyo3::prelude::*;
 use std::process::Command;
 use std::path::Path;
 
-pub fn render_video(
+#[pyfunction]
+fn render_video(
     id: usize,
-    image_path: &str,
-    audio_path: &str,
+    image_path: String,
+    audio_path: String,
     duration: f64,
     words: Vec<(String, (u32, u32, u32, u32))>,
-    output_dir: &str,
-) -> Result<String, String> {
+    output_dir: String,
+) -> PyResult<String> {
     // Validate paths
-    if !Path::new(image_path).exists() {
-        return Err(format!("Image not found: {}", image_path));
+    if !Path::new(&image_path).exists() {
+        return Err(pyo3::exceptions::PyFileNotFoundError::new_err(format!(
+            "Image not found: {}",
+            image_path
+        )));
     }
-    if !Path::new(audio_path).exists() {
-        return Err(format!("Audio not found: {}", audio_path));
+    if !Path::new(&audio_path).exists() {
+        return Err(pyo3::exceptions::PyFileNotFoundError::new_err(format!(
+            "Audio not found: {}",
+            audio_path
+        )));
     }
 
     let output_path = format!("{}/clip{}.mp4", output_dir, id);
@@ -30,7 +38,6 @@ pub fn render_video(
         let start = i as f64 * highlight_duration;
         let end = start + highlight_duration;
 
-        // Create highlight box
         filter_complex.push_str(&format!(
             "[{last_out}]drawbox=x={x}:y={y}:w={w}:h={h}:color=yellow@0.5:t=fill:enable='between(t,{start:.3},{end:.3})'[v{i}];",
             last_out = last_output,
@@ -48,39 +55,41 @@ pub fn render_video(
     // Remove trailing semicolon
     filter_complex.pop();
 
-    // FFmpeg command with optimized settings
+    // FFmpeg command
     let status = Command::new("ffmpeg")
         .args(&[
-            "-y", // Overwrite output
-            "-hwaccel", "auto", // Enable hardware acceleration
+            "-y",
+            "-hwaccel", "auto",
             "-loop", "1",
-            "-i", image_path,
-            "-i", audio_path,
+            "-i", &image_path,
+            "-i", &audio_path,
             "-filter_complex", &filter_complex,
             "-map", &last_output,
             "-map", "1:a",
             "-c:v", "libx264",
-            "-preset", "fast", // Balance between speed and compression
-            "-crf", "18", // High quality (lower = better quality)
+            "-preset", "fast",
+            "-crf", "18",
             "-pix_fmt", "yuv420p",
-            "-movflags", "+faststart", // For web streaming
+            "-movflags", "+faststart",
             "-c:a", "aac",
-            "-b:a", "192k", // Higher audio bitrate
-            "-r", "60", // Higher framerate for smoother animation
+            "-b:a", "192k",
+            "-r", "60",
             "-t", &duration_str,
-            "-vsync", "2", // Frame rate conversion method
+            "-vsync", "2",
             &output_path,
         ])
         .status();
 
     match status {
         Ok(exit_status) if exit_status.success() => Ok(output_path),
-        Ok(_) => Err("FFmpeg command failed".to_string()),
-        Err(e) => Err(format!("Failed to execute FFmpeg: {}", e)),
+        Ok(_) => Err(pyo3::exceptions::PyRuntimeError::new_err("FFmpeg command failed")),
+        Err(e) => Err(pyo3::exceptions::PyRuntimeError::new_err(format!(
+            "Failed to execute FFmpeg: {}",
+            e
+        ))),
     }
 }
 
-
 #[pymodule]
 fn rust_highlight(_py: Python, m: &PyModule) -> PyResult<()> {
     m.add_function(wrap_pyfunction!(render_video, m)?)?;