diff --git a/vite.config.ts b/vite.config.ts
index 8b0f57b..6c70beb 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -1,7 +1,27 @@
-import { defineConfig } from 'vite'
-import react from '@vitejs/plugin-react'
+import react from "@vitejs/plugin-react";
+import { defineConfig } from "vite";
 
-// https://vite.dev/config/
+const allowedHosts = [".develop.taylordb.ai", "localhost", "127.0.0.1"];
+
+// https://vitejs.dev/config/
 export default defineConfig({
-  plugins: [react()],
-})
+  plugins: [
+    react(),
+    {
+      name: "host-validation",
+      configureServer(server) {
+        server.middlewares.use((req, res, next) => {
+          const host = req.headers.host?.split(":")[0];
+          if (host && allowedHosts.some((allowed) => host.endsWith(allowed))) {
+            return next();
+          }
+          res.writeHead(403, { "Content-Type": "text/plain" });
+          res.end(`Forbidden host: ${host}`);
+        });
+      },
+    },
+  ],
+  server: {
+    host: "0.0.0.0", // Listen on all network interfaces
+  },
+});