Update rust_highlight/src/lib.rs

rust_highlight/src/lib.rs

@@ -1,56 +1,12 @@
-use pyo3::prelude::*;
-use std::process::Command;
-
-#[pyfunction]
-fn render_video(
-    id: usize,
-    image_path: &str,
-    audio_path: &str,
-    duration: f64,
-    words: Vec<(String, (u32, u32, u32, u32))>,
-) -> PyResult<String> {
-    let clip = format!("clip{}.mp4", id);
-    let duration = duration.to_string();
-
-    let mut vf = String::new();
-    let n = words.len() as f64;
-    for (i, (_, (x, y, w, h))) in words.iter().enumerate() {
-        let start = (i as f64) * (duration.parse::<f64>().unwrap() / n);
-        let end = start + (duration.parse::<f64>().unwrap() / n);
-        vf.push_str(&format!(
-            "drawbox=x={}:y={}:w={}:h={}:[email protected]:enable='between(t,{:.3},{:.3})',",
-            x, y, w, h, start, end
-        ));
-    }
-    vf = vf.trim_end_matches(',').to_string();
-
-    let status = Command::new("ffmpeg")
-        .args(&[
-            "-y",
-            "-loop", "1",
-            "-i", image_path,
-            "-i", audio_path,
-            "-vf", &vf,
-            "-t", &duration,
-            "-c:v", "libx264",
-            "-pix_fmt", "yuv420p",
-            "-c:a", "aac",
-            &clip,
-        ])
-        .status()
-        .expect("ffmpeg failed");
-
-    if status.success() {
-        Ok(clip)
-    } else {
-        Err(PyErr::new::<pyo3::exceptions::PyRuntimeError, _>(
-            "FFmpeg failed",
-        ))
-    }
-}
-
-#[pymodule]
-fn rust_highlight(py: Python, m: &PyModule) -> PyResult<()> {
-    m.add_function(wrap_pyfunction!(render_video, m)?)?;
-    Ok(())
+use pyo3::prelude::*;
+
+#[pyfunction]
+fn rust_hello() -> PyResult<String> {
+    Ok("Hello from Rust!".to_string())
+}
+
+#[pymodule]
+fn rust_highlight(_py: Python, m: &PyModule) -> PyResult<()> {
+    m.add_function(wrap_pyfunction!(rust_hello, m)?)?;
+    Ok(())
 }