Update src/main.rs

src/main.rs

@@ -1,5 +1,4 @@
 use axum::{routing::{get, post}, Router};
-use std::net::SocketAddr;
 use tower_http::services::ServeDir;
 use std::sync::Arc;
 use tokio::sync::broadcast;
@@ -16,15 +15,20 @@ pub struct AppState {
 #[tokio::main]
 async fn main() {
     tracing_subscriber::fmt::init();
+    
+    // Channel for streaming logs and browser frames to UI
     let (tx, _) = broadcast::channel(100);
     let state = Arc::new(AppState { tx });
 
     let app = Router::new()
         .route("/api/chat", post(agent::chat_handler))
         .route("/api/stream", get(sse::stream::sse_handler))
+        // Serve the React frontend from the dist folder
         .fallback_service(ServeDir::new("./frontend/dist"))
         .with_state(state);
 
-    let addr = SocketAddr::from(([0, 0, 0, 0], 7860));
-    axum::Server::bind(&addr).serve(app.into_make_service()).await.unwrap();
+    // Axum 0.7 way to start the server
+    let listener = tokio::net::TcpListener::bind("0.0.0.0:7860").await.unwrap();
+    println!("🚀 Agent Server running on http://0.0.0.0:7860");
+    axum::serve(listener, app).await.unwrap();
 }