Update server.ts

server.ts

@@ -10,6 +10,14 @@ import { GoogleGenAI, Type } from "@google/genai";
 import { createProxyMiddleware } from 'http-proxy-middleware';
 import pv from 'prismarine-viewer';
 
+process.on('uncaughtException', (err) => {
+  console.error('Uncaught Exception:', err);
+});
+
+process.on('unhandledRejection', (reason, promise) => {
+  console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+});
+
 const app = express();
 const httpServer = createServer(app);
 const io = new Server(httpServer, {
@@ -18,7 +26,7 @@ const io = new Server(httpServer, {
   },
 });
 
-const PORT = Number(process.env.PORT) || 7860;
+const PORT = 3000;
 
 let bot: mineflayer.Bot | null = null;
 let ai: GoogleGenAI | null = null;
@@ -50,6 +58,7 @@ function stopAfkMode() {
     if (bot) {
       bot.pathfinder.setGoal(null);
       bot.chat('/gamemode creative @s');
+      bot.creative.stopFlying();
     }
     log("AFK mode stopped.", "info");
   }
@@ -66,6 +75,7 @@ async function startAfkMode() {
   setTimeout(() => {
     if (!bot) return;
     bot.chat('/tp @s ~ ~20 ~'); // Teleport up into the air
+    bot.creative.startFlying();
   }, 500);
 
   afkInterval = setInterval(() => {
@@ -170,6 +180,11 @@ async function startServer() {
       const { host, port, version, username, apiKey, model } = config;
 
       if (bot) {
+        if ((bot as any).viewer) {
+          try {
+            (bot as any).viewer.close();
+          } catch (e) {}
+        }
         bot.quit();
         log("Existing bot disconnected");
       }
@@ -227,7 +242,8 @@ async function startServer() {
           if (!bot?.entity || isMlgWatering) return;
           
           // If falling fast (velocity.y is negative)
-          if (bot.entity.velocity.y < -0.6 && !bot.entity.onGround && !bot.entity.isInWater) {
+          const inWater = bot.blockAt(bot.entity.position)?.name === 'water' || bot.blockAt(bot.entity.position)?.name === 'flowing_water';
+          if (bot.entity.velocity.y < -0.6 && !bot.entity.onGround && !inWater) {
             const waterBucket = bot.inventory.items().find(item => item.name === 'water_bucket');
             if (!waterBucket) return;
 
@@ -915,9 +931,9 @@ async function startServer() {
                yaw: bot.entity.yaw,
                pitch: bot.entity.pitch
              });
-             const host = process.env.PUBLIC_URL || "http://localhost:7860";
-             bot.chat(`Here is the screenshot: /screenshot/${id}`);
-             log(`Screenshot taken: /screenshot/${id}`, "success");
+             const hostUrl = process.env.PUBLIC_URL || "http://localhost:7860";
+             bot.chat(`Here is the live 3D view: ${hostUrl}/efir/`);
+             log(`Live 3D view requested: ${hostUrl}/efir/`, "success");
            } else if (action.length >= 4) {
              const [x, y, z, rawBlock] = action;
              const parseCoord = (c: any) => {
@@ -1017,6 +1033,19 @@ async function startServer() {
       appType: "spa",
     });
     app.use(vite.middlewares);
+    
+    app.use('*', async (req, res, next) => {
+      try {
+        const url = req.originalUrl;
+        const fs = await import('fs');
+        let template = fs.readFileSync(path.resolve(process.cwd(), 'index.html'), 'utf-8');
+        template = await vite.transformIndexHtml(url, template);
+        res.status(200).set({ 'Content-Type': 'text/html' }).end(template);
+      } catch (e) {
+        vite.ssrFixStacktrace(e as Error);
+        next(e);
+      }
+    });
   } else {
     const distPath = path.join(process.cwd(), "dist");
     app.use(express.static(distPath));