Update src/main.rs

src/main.rs

@@ -0,0 +1,20 @@
+use axum::{routing::{get, post}, Router, response::sse::{Event, Sse}};
+use std::net::SocketAddr;
+use tower_http::services::ServeDir;
+mod agent;
+mod browser;
+
+#[tokio::main]
+async fn main() {
+    let app = Router::new()
+        // API for Chat
+        .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"));
+
+    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();
+}