Add health check endpoint and fix port

Dockerfile

@@ -12,4 +12,5 @@ RUN apt-get update && apt-get install -y ca-certificates && rm -rf /var/lib/apt/
 COPY --from=builder /app/target/release/rustvital-amd /usr/local/bin/rustvital-amd
 EXPOSE 7860
 ENV RUST_LOG=info
+ENV PORT=7860
 CMD ["rustvital-amd"]

src/main.rs

@@ -1,27 +1,29 @@
-use axum::{routing::post, Router};
+use axum::{routing::{get, post}, Router};
 use tower_http::trace::TraceLayer;
 use tracing_subscriber::EnvFilter;
 
-// Use the library crate to get the handler
 use rustvital_amd::handlers;
 
 #[tokio::main]
 async fn main() -> anyhow::Result<()> {
-    // Load .env file (ignore if missing)
     dotenvy::dotenv().ok();
 
-    // Initialise tracing (respects RUST_LOG from env)
     tracing_subscriber::fmt()
         .with_env_filter(EnvFilter::from_default_env().add_directive("rustvital_amd=debug".parse()?))
         .init();
 
-    tracing::info!("Starting RustVital-AMD server");
+    let port = std::env::var("PORT").unwrap_or_else(|_| "3000".to_string());
+    let addr = format!("0.0.0.0:{}", port);
+
+    tracing::info!("Starting RustVital-AMD server on {}", addr);
 
     let app = Router::new()
+        .route("/", get(|| async { "RustVital-AMD is running" }))
+        .route("/health", get(|| async { "healthy" }))
         .route("/triage", post(handlers::triage::handle))
         .layer(TraceLayer::new_for_http());
 
-    let listener = tokio::net::TcpListener::bind("0.0.0.0:3000").await?;
+    let listener = tokio::net::TcpListener::bind(&addr).await?;
     axum::serve(listener, app).await?;
     Ok(())
 }