Update app.py

app.py

@@ -250,7 +250,7 @@ EDITOR_TEMPLATE = '''
         let currentRotation = 0;
         const gridSize = 1;
         let levelData = { floors: {}, walls: {}, objects: {}, buildingDefinitions: {}, buildingInstances: [] };
-        let drawMode = 'single'; // 'single' or 'area'
+        let drawMode = 'single';
         let isDrawingArea = false;
         let areaStartPoint = new THREE.Vector3();
 
@@ -599,6 +599,7 @@ EDITOR_TEMPLATE = '''
             if (previewMesh) {
                 scene.remove(previewMesh);
                 previewMesh.geometry.dispose();
+                if(previewMesh.material.map) previewMesh.material.map.dispose();
                 previewMesh.material.dispose();
             }
 
@@ -626,6 +627,7 @@ EDITOR_TEMPLATE = '''
                 previewMesh = new THREE.Mesh(geometry, material);
                 previewMesh.userData.isBuilding = false;
             }
+            previewMesh.rotation.y = currentRotation;
             scene.add(previewMesh);
         }
 
@@ -742,6 +744,7 @@ EDITOR_TEMPLATE = '''
         }
         
         function rotatePreview(direction) {
+            if (!previewMesh) return;
             currentRotation += (Math.PI / 2) * direction;
             previewMesh.rotation.y = currentRotation;
         }
@@ -896,7 +899,7 @@ EDITOR_TEMPLATE = '''
 
         function togglePlayMode() {
             const blocker = document.getElementById('blocker');
-            if(isPlayMode) { // Exit play mode
+            if(isPlayMode) {
                 isPlayMode = false;
                 document.exitPointerLock?.();
                 const uiContainer = document.getElementById('ui-container');
@@ -913,7 +916,7 @@ EDITOR_TEMPLATE = '''
                 camera.lookAt(0,0,0);
                 orbitControls.target.set(0, 0, 0);
                 onWindowResize(); 
-            } else { // Enter play mode
+            } else {
                 blocker.style.display = 'block';
             }
         }
@@ -973,6 +976,8 @@ EDITOR_TEMPLATE = '''
                 
                 clearLevel();
                 Object.assign(levelData, result.data);
+                if (!levelData.buildingDefinitions) levelData.buildingDefinitions = {};
+                if (!levelData.buildingInstances) levelData.buildingInstances = [];
                 updateCustomToolsUI();
                 updateLevelGeometry();
                 document.getElementById('project-name').value = projectName;