Update src/main.rs

src/main.rs

@@ -1,20 +1,31 @@
-use axum::{routing::{get, post}, Router, response::sse::{Event, Sse}};
+use axum::{routing::{get, post}, Router};
 use std::net::SocketAddr;
 use tower_http::services::ServeDir;
+use tower_http::cors::CorsLayer;
+
 mod agent;
 mod browser;
+mod sse;
+mod ui;
 
 #[tokio::main]
 async fn main() {
+    tracing_subscriber::fmt::init();
+
     let app = Router::new()
-        // API for Chat
+        // API for the Chat interface
         .route("/api/chat", post(agent::chat_handler))
-        // SSE Stream for Browser Frames & Logs
-        .route("/api/stream", get(agent::stream_handler))
-        // Serve the React Frontend
-        .fallback_service(ServeDir::new("dist"));
+        // 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());
 
     let addr = SocketAddr::from(([0, 0, 0, 0], 7860));
-    println!("Server starting on http://localhost:7860");
-    axum::Server::bind(&addr).serve(app.into_make_service()).await.unwrap();
+    println!("🚀 Agent Server running on http://localhost:7860");
+    
+    axum::Server::bind(&addr)
+        .serve(app.into_make_service())
+        .await
+        .unwrap();
 }