Update rust_highlight/src/lib.rs

rust_highlight/src/lib.rs

@@ -1,11 +1,12 @@
 use pyo3::prelude::*;
-use std::process::Command;
+use pyo3::wrap_pyfunction;
+use std::fs::File;
+use std::io::Write;
 use std::path::Path;
-use pyo3::wrap_pyfunction; 
-use std::fs::File; 
-use std::io::Write;  
+use std::process::Command;
 use uuid::Uuid;
 
+/// Renders a video by highlighting regions over time
 #[pyfunction]
 fn render_video(
     id: usize,
@@ -62,7 +63,7 @@ fn render_video(
             "-c:a", "aac",
             "-b:a", "192k",
             "-r", "60",
-            "-fps_mode", "vfr",  // Replaces deprecated -vsync
+            "-fps_mode", "vfr",
             "-t", &duration_str,
             &output_path,
         ])
@@ -80,23 +81,24 @@ fn render_video(
     }
 }
 
-
-#[pyfunction] 
-fn combine_clips(clips: Vec<String>) -> PyResult<String> { 
-    let paths: Vec<&str> = clips.iter().map(|s| s.as_str()).collect(); 
-    concat_videos(paths) 
+/// Combines multiple video clips into one
+#[pyfunction]
+fn combine_clips(clips: Vec<String>) -> PyResult<String> {
+    let paths: Vec<&str> = clips.iter().map(|s| s.as_str()).collect();
+    concat_videos(paths)
 }
 
-fn concat_videos(clips: Vec<&str>) -> PyResult<String> { 
-    // Step 1: Write file list 
-    let list_file = format!("/tmp/clips_{}.txt", Uuid::new_v4()); 
+fn concat_videos(clips: Vec<&str>) -> PyResult<String> {
+    // Step 1: Write file list
+    let list_file = format!("/tmp/clips_{}.txt", Uuid::new_v4());
     let mut file = File::create(&list_file)
-        .map_err(|e| pyo3::exceptions::PyIOError::new_err(e.to_string()))?; 
+        .map_err(|e| pyo3::exceptions::PyIOError::new_err(e.to_string()))?;
     for clip in &clips {
         writeln!(file, "file '{}'", clip)
-            .map_err(|e| pyo3::exceptions::PyIOError::new_err(e.to_string()))?; }
+            .map_err(|e| pyo3::exceptions::PyIOError::new_err(e.to_string()))?;
+    }
 
-// Step 2: Run FFmpeg
+    // Step 2: Run FFmpeg
     let output_file = format!("/tmp/final_video_{}.mp4", Uuid::new_v4());
     let status = Command::new("ffmpeg")
         .args(&[
@@ -106,22 +108,25 @@ fn concat_videos(clips: Vec<&str>) -> PyResult<String> {
             "-i", &list_file,
             "-c", "copy",
             &output_file,
-    ])
+        ])
         .status()
         .map_err(|e| pyo3::exceptions::PyRuntimeError::new_err(e.to_string()))?;
-    
+
     if status.success() {
         Ok(output_file)
     } else {
         Err(pyo3::exceptions::PyRuntimeError::new_err("FFmpeg command failed"))
     }
-    
-    }
+}
 
 #[pymodule]
 fn rust_highlight(_py: Python, m: &PyModule) -> PyResult<()> {
     m.add_function(wrap_pyfunction!(render_video, m)?)?;
-    m.add_function(wrap_pyfunction!(combine_clips, m)?)?; 
     Ok(())
 }
 
+#[pymodule]
+fn rust_combiner(_py: Python, m: &PyModule) -> PyResult<()> {
+    m.add_function(wrap_pyfunction!(combine_clips, m)?)?;
+    Ok(())
+}