Update src/browser/controller.rs

src/browser/controller.rs

@@ -1,24 +1,28 @@
 use chromiumoxide::browser::{Browser, BrowserConfig};
+use chromiumoxide::page::Page;
 use futures_util::StreamExt;
-use tokio::sync::broadcast;
 
-pub async fn run_browser(tx: broadcast::Sender<String>) {
+pub async fn init_browser() -> (Browser, Page) {
     let (browser, mut handler) = Browser::launch(
         BrowserConfig::builder()
             .arg("--no-sandbox")
-            .build().unwrap()
-    ).await.unwrap();
+            .arg("--disable-setuid-sandbox")
+            .arg("--headless")
+            .build()
+            .unwrap(),
+    )
+    .await
+    .unwrap();
 
+    // Critical: Keep the browser handler running in the background
     tokio::spawn(async move {
-        while let Some(h) = handler.next().await { h.unwrap(); }
+        while let Some(h) = handler.next().await {
+            if let Err(e) = h {
+                eprintln!("Browser handler error: {}", e);
+            }
+        }
     });
 
-    let page = browser.new_page("https://www.google.com").await.unwrap();
-    loop {
-        if let Ok(img) = page.screenshot().await {
-            let b64 = base64::encode(img);
-            let _ = tx.send(format!("{{\"type\":\"frame\", \"data\":\"{}\"}}", b64));
-        }
-        tokio::time::sleep(std::time::Duration::from_millis(200)).await;
-    }
+    let page = browser.new_page("about:blank").await.unwrap();
+    (browser, page)
 }