Update app.py

app.py

@@ -1,95 +1,4 @@
-import streamlit as st
-import os
-import base64
-from pathlib import Path
-import shutil
-import random
-
-# 🌈 Load A-Frame and custom components
-@st.cache_data
-def load_aframe_and_extras():
-    return """
-    <script src="https://aframe.io/releases/1.2.0/aframe.min.js"></script>
-    <script src="https://unpkg.com/aframe-event-set-component@5.0.0/dist/aframe-event-set-component.min.js"></script>
-    <script>
-    // 🕹️ Make objects draggable
-    AFRAME.registerComponent('draggable', {
-      // ... (draggable component code remains the same)
-    });
-
-    // 🦘 Make objects bounce
-    AFRAME.registerComponent('bouncing', {
-      // ... (bouncing component code remains the same)
-    });
-
-    // 💡 Create moving light sources
-    AFRAME.registerComponent('moving-light', {
-      // ... (moving-light component code remains the same)
-    });
-
-    // 📷 Camera controls
-    function moveCamera(direction) {
-      var camera = document.querySelector('[camera]');
-      var rig = document.querySelector('#rig');
-      var pos = rig.getAttribute('position');
-      var rot = rig.getAttribute('rotation');
-      var speed = 0.5;
-      var rotationSpeed = 5;
-      
-      switch(direction) {
-        case 'up':
-          pos.z -= speed;
-          break;
-        case 'down':
-          pos.z += speed;
-          break;
-        case 'left':
-          pos.x -= speed;
-          break;
-        case 'right':
-          pos.x += speed;
-          break;
-        case 'rotateLeft':
-          rot.y += rotationSpeed;
-          break;
-        case 'rotateRight':
-          rot.y -= rotationSpeed;
-          break;
-        case 'reset':
-          pos = {x: 0, y: 10, z: 0};
-          rot = {x: -90, y: 0, z: 0};
-          break;
-        case 'ground':
-          pos = {x: 0, y: 1.6, z: 0};
-          rot = {x: 0, y: 0, z: 0};
-          break;
-      }
-      
-      rig.setAttribute('position', pos);
-      rig.setAttribute('rotation', rot);
-    }
-
-    // ⌨️ Keyboard controls
-    document.addEventListener('keydown', function(event) {
-      switch(event.key.toLowerCase()) {
-        case 'q':
-          moveCamera('rotateLeft');
-          break;
-        case 'e':
-          moveCamera('rotateRight');
-          break;
-        case 'z':
-          moveCamera('reset');
-          break;
-        case 'c':
-          moveCamera('ground');
-          break;
-      }
-    });
-    </script>
-    """
-
-# ... (The rest of the functions remain the same)
+# ... (previous code remains the same)
 
 # 🎭 Main function to run the Streamlit app
 def main():
@@ -133,7 +42,37 @@ def main():
         st.markdown("### 📁 Directory")
         directory = st.text_input("Enter path:", ".", key="directory_input")
 
-    # ... (The rest of the main function remains the same)
+    if not os.path.isdir(directory):
+        st.sidebar.error("Invalid directory path")
+        return
+
+    file_types = ['obj', 'glb', 'webp', 'png', 'mp4']
+    
+    if uploaded_files:
+        for uploaded_file in uploaded_files:
+            file_extension = Path(uploaded_file.name).suffix.lower()[1:]
+            if file_extension in file_types:
+                with open(os.path.join(directory, uploaded_file.name), "wb") as f:
+                    shutil.copyfileobj(uploaded_file, f)
+                st.sidebar.success(f"Uploaded: {uploaded_file.name}")
+            else:
+                st.sidebar.warning(f"Skipped unsupported file: {uploaded_file.name}")
+
+    files = [f for f in os.listdir(directory) if f.split('.')[-1] in file_types]
+
+    aframe_scene = f"""
+    <a-scene embedded style="height: 600px; width: 100%;">
+      <a-entity id="rig" position="0 {max(grid_width, grid_height)} 0" rotation="-90 0 0">
+        <a-camera fov="60" look-controls wasd-controls cursor="rayOrigin: mouse" raycaster="objects: .raycastable"></a-camera>
+      </a-entity>
+      <a-sky color="#87CEEB"></a-sky>
+      <a-entity moving-light="color: #FFD700; speed: 0.07 0.05 0.06; bounds: 4 3 4" position="2 2 -2"></a-entity>
+      <a-entity moving-light="color: #FF6347; speed: 0.06 0.08 0.05; bounds: 4 3 4" position="-2 1 2"></a-entity>
+      <a-entity moving-light="color: #00CED1; speed: 0.05 0.06 0.07; bounds: 4 3 4" position="0 3 0"></a-entity>
+    """
+
+    assets, entities = generate_tilemap(files, directory, grid_width, grid_height, max_unique_models=5)
+    aframe_scene += assets + entities + "</a-scene>"
 
     camera_move = st.session_state.get('camera_move', None)
     if camera_move: