chore: port, redirect, cors config

.env.example

@@ -1,3 +1,4 @@
 OPENAI_API_KEY=
-QDRANT_URL=         #Default: http://localhost:6334
-RUST_LOG=           #Logging levels: "error", "warn", "info", "debug", "trace"
+QDRANT_URL=         #Defaults to http://localhost:6334
+RUST_LOG=           #Logging levels: "error", "warn", "info", "debug", "trace"
+WEBSERVER_PORT=     #Defaults to 3000

Cargo.toml

@@ -30,3 +30,4 @@ actix-rt = "2"
 tracing-actix-web = "0.7"
 env_logger = "0.10"
 tokio = "1"
+actix-cors = "0.6.4"

src/constants.rs

@@ -5,7 +5,7 @@ pub const EMBEDDINGS_DIMENSION: usize = 384;
 
 //Actix-web
 pub const SSE_CHANNEL_BUFFER_SIZE: usize = 1;
-pub const ACTIX_WEB_SERVER_PORT: usize = 3000;
+pub const HOME_ROUTE_REDIRECT_URL: &str = "https://opensauced.pizza";
 
 //OpenAI
 pub const CHAT_COMPLETION_TEMPERATURE: f64 = 0.7;

src/main.rs

@@ -8,8 +8,9 @@ mod routes;
 mod utils;
 use std::{path::Path, sync::Arc};
 
-use actix_web::{web, App, HttpResponse, HttpServer};
-use constants::ACTIX_WEB_SERVER_PORT;
+use actix_cors::Cors;
+use actix_web::{web, App, HttpServer};
+use constants::HOME_ROUTE_REDIRECT_URL;
 use env_logger::Env;
 use tracing_actix_web::TracingLogger;
 
@@ -21,18 +22,22 @@ async fn main() -> std::io::Result<()> {
 
     let model: Arc<embeddings::Onnx> = Arc::new(embeddings::Onnx::new(Path::new("model")).unwrap());
     let db: Arc<db::QdrantDB> = Arc::new(db::QdrantDB::initialize().unwrap());
+    let port = std::env::var("WEBSERVER_PORT").unwrap_or("3000".into());
+    let port = port.parse::<u16>().expect("Invalid WEBSERVER_PORT");
 
     HttpServer::new(move || {
+
         App::new()
+            .wrap(Cors::permissive())
             .wrap(TracingLogger::default())
-            .route("/", web::get().to(HttpResponse::Ok))
+            .service(web::redirect("/", HOME_ROUTE_REDIRECT_URL))
             .service(routes::embeddings)
             .service(routes::query)
             .service(routes::repo)
             .app_data(web::Data::new(model.clone()))
             .app_data(web::Data::new(db.clone()))
     })
-    .bind(("0.0.0.0", ACTIX_WEB_SERVER_PORT as u16))?
+    .bind(("0.0.0.0", port))?
     .run()
     .await
 }