Update src/main.rs

src/main.rs

@@ -1,31 +1,30 @@
 use axum::{routing::{get, post}, Router};
 use std::net::SocketAddr;
 use tower_http::services::ServeDir;
-use tower_http::cors::CorsLayer;
+use std::sync::Arc;
+use tokio::sync::broadcast;
 
 mod agent;
 mod browser;
 mod sse;
 mod ui;
 
+pub struct AppState {
+    pub tx: broadcast::Sender<String>,
+}
+
 #[tokio::main]
 async fn main() {
     tracing_subscriber::fmt::init();
+    let (tx, _) = broadcast::channel(100);
+    let state = Arc::new(AppState { tx });
 
     let app = Router::new()
-        // API for the Chat interface
         .route("/api/chat", post(agent::chat_handler))
-        // SSE Stream for Browser frames and Agent logs
-        .route("/api/stream", get(sse::sse_handler))
-        // Serve the frontend (Agent can modify this folder!)
-        .fallback_service(ServeDir::new("./dist"))
-        .layer(CorsLayer::permissive());
+        .route("/api/stream", get(sse::stream::sse_handler))
+        .fallback_service(ServeDir::new("./frontend/dist"))
+        .with_state(state);
 
     let addr = SocketAddr::from(([0, 0, 0, 0], 7860));
-    println!("🚀 Agent Server running on http://localhost:7860");
-    
-    axum::Server::bind(&addr)
-        .serve(app.into_make_service())
-        .await
-        .unwrap();
+    axum::Server::bind(&addr).serve(app.into_make_service()).await.unwrap();
 }