working on the API

src/gaussian-splatting.mts

@@ -3,7 +3,7 @@ import { exec } from "child_process"
 /**
  * Interface for the options for the `runGaussianSplattingCUDA` function.
  */
-interface Options {
+export interface Options {
   /**
    * Data path for the training data.
    */

src/index.mts

@@ -1,7 +1,7 @@
-console.log("splatter-api is booting up")
-
 import express from "express"
 
+import { runGaussianSplattingCUDA, Options } from "./gaussian-splatting.mts"
+
 import { initFolders } from "./initFolders.mts"
 
 initFolders()
@@ -9,15 +9,36 @@ initFolders()
 const app = express()
 const port = 7860
 
-const queue = []
-const queueLimit = 4
+const maxActiveRequests = 4
+let activeRequests = 0
 
 app.use(express.json({limit: '50mb'}))
 app.use(express.urlencoded({limit: '50mb', extended: true}))
 
 app.post("/", async (req, res) => {
-  // TODO
-})
+  if(activeRequests >= maxActiveRequests) {
+    res.status(503).json({message: "Service Unavailable: Max concurrent requests reached. Please try again later"}).end();
+    return
+  }
+  activeRequests++
+  
+  let options: Options = req.body // Assuming the body contains the options
+  
+  try {
+    const result = await runGaussianSplattingCUDA(options);
+    res.status(200).json({
+      success: true,
+      message: result
+    }).end();
+  } catch (error) {
+    res.status(500).json({
+      error: "Couldn't generate gaussian splatting scene",
+      message: error
+    }).end()
+  }
+
+  activeRequests--
+});
 
 app.get("/", async (req, res) => {
   // this is what users will see in the space - but no need to show something scary
@@ -28,4 +49,5 @@ Splatter API is a micro-service used to generate gaussian splatting scenes.
   res.end()
 })
 
-app.listen(port, () => { console.log(`Open http://localhost:${port}`) })
+app.listen(port, () => { console.log(`Open http://localhost:${port}`) })
+