Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import os, base64, shutil, random
 from pathlib import Path
 
+# 🚀 1. Load Aframe and Scripts: Let's get the party started with Aframe and friends! 🎉
 @st.cache_data
 def load_aframe_and_extras():
     return """
@@ -17,6 +18,7 @@ def load_aframe_and_extras():
     let maxRotSpeed = 2;
     let rotAccel = 0.1;
 
+    // 🖱️ 1.1 Draggable Component: Click and drag like a pro! 🏋️
     AFRAME.registerComponent('draggable', {
       init: function () {
         this.el.addEventListener('mousedown', this.onMouseDown.bind(this));
@@ -66,9 +68,11 @@ def load_aframe_and_extras():
       }
     });
 
+    // ⛹️ 1.2 Bouncing and Lights: Keep the scene lively! ✨
     AFRAME.registerComponent('bouncing', {/* … */});
     AFRAME.registerComponent('moving-light', {/* … */});
 
+    // 📸 1.3 Move Camera: Snap to your favorite quadrant views! 🖼️
     function moveCamera(view) {
       var rig = document.querySelector('#rig');
       var gw = 8, gh = 8;
@@ -78,7 +82,7 @@ def load_aframe_and_extras():
       var quadrantOffset = Math.max(gw, gh) / 4; // 2 units to center on quadrants
       var pos, rot;
 
-      rot = {x: -90, y: 0, z: 0}; // Always top-down view
+      rot = {x: -90, y: 0, z: 0}; // Always top-down view 📉
 
       switch(view) {
         case 'center':
@@ -113,6 +117,7 @@ def load_aframe_and_extras():
       rig.setAttribute('rotation', rot);
     }
 
+    // ✈️ 1.4 Fly Camera: Soar through the scene with style! 🛩️
     function flyCamera(action) {
       var rig = document.querySelector('#rig');
       var pos = rig.getAttribute('position');
@@ -169,6 +174,7 @@ def load_aframe_and_extras():
       rig.setAttribute('position', pos);
     }
 
+    // 🔄 1.5 Update Camera: Keep the camera flying smoothly! 🕊️
     function updateCameraPosition() {
       var rig = document.querySelector('#rig');
       var pos = rig.getAttribute('position');
@@ -189,6 +195,7 @@ def load_aframe_and_extras():
 
     requestAnimationFrame(updateCameraPosition);
 
+    // 🎯 1.6 Fire Raycast: Zap objects with a laser beam! ⚡
     function fireRaycast() {
       var cam = document.querySelector('[camera]');
       var dir = new THREE.Vector3(); cam.object3D.getWorldDirection(dir);
@@ -206,6 +213,7 @@ def load_aframe_and_extras():
       }
     }
 
+    // ⌨️ 1.7 Keyboard Listener: Take control with your keys! 🎮
     document.addEventListener('keydown', e => {
       switch(e.key.toLowerCase()){
         case '1': moveCamera('center'); break;
@@ -232,10 +240,12 @@ def load_aframe_and_extras():
     </script>
     """
 
+# 🔐 2. Encode File: Sneakily encode files into base64! 🕵️
 @st.cache_data
 def encode_file(path):
     with open(path,'rb') as f: return base64.b64encode(f.read()).decode()
 
+# 🏰 3. Create Aframe Entity: Build magical 3D models! 🪄
 def create_aframe_entity(stem, ext, pos):
     ry = random.uniform(0,360)
     if ext == 'obj':
@@ -248,6 +258,7 @@ def create_aframe_entity(stem, ext, pos):
                 'class="raycastable" draggable></a-entity>')
     return ''
 
+# 🗺️ 4. Generate Tilemap: Craft the world one tile at a time! 🌍
 @st.cache_data
 def generate_tilemap(files, dirpath, gw=8, gh=8):
     img_exts   = ['webp','png','jpeg','jpg']
@@ -258,6 +269,7 @@ def generate_tilemap(files, dirpath, gw=8, gh=8):
     models = [f for f in files if f.split('.')[-1] in model_exts]
     vids   = [f for f in files if f.split('.')[-1] in vid_exts]
 
+    # 📦 4.1 Load Assets: Pack up all the goodies! 🎁
     assets = "<a-assets>"
     for f in files:
         stem = Path(f).stem; ext=f.split('.')[-1]
@@ -273,7 +285,9 @@ def generate_tilemap(files, dirpath, gw=8, gh=8):
                        'loop autoplay muted></video>')
     assets += "</a-assets>"
 
+    # 🏞️ 4.2 Build the Scene: Lay out the tiles and models! 🏡
     entities = ""
+    model_counts = {f: 0 for f in models}  # Track spawned model counts 📊
     if vids:
         v = vids[0]; s = Path(v).stem
         entities += (f'<a-video src="#{s}" width="{gw}" height="{gh}" '
@@ -295,11 +309,14 @@ def generate_tilemap(files, dirpath, gw=8, gh=8):
         if models:
           m = random.choice(models); ext = m.split('.')[-1]; s = Path(m).stem
           entities += create_aframe_entity(s, ext, f"{x} 0 {z}")
-    return assets, entities
+          model_counts[m] += 1  # Increment count for this model 📈
+    return assets, entities, model_counts
 
+# 🎨 5. Main Function: Paint the canvas of your 3D world! 🖌️
 def main():
     st.set_page_config(layout="wide")
     with st.sidebar:
+        # 🧭 5.1 Camera Views UI: Pick your perfect angle! 📐
         st.markdown("### 🧭 Camera Views")
         st.markdown("**Select Quadrant** 📷")
         cols = st.columns(3)
@@ -315,22 +332,32 @@ def main():
         cols[1].button("⬇️ Bottom", on_click=lambda: st.session_state.update({'camera_view': 'back'}))
         cols[2].button("↘️ Bottom-Right", on_click=lambda: st.session_state.update({'camera_view': 'angle4'}))
 
+        # 📂 5.2 File Uploader: Drop your media treasures here! 💎
         st.markdown("### ➕ Add Media Files")
-        ups = st.file_uploader("Add files (png, jpeg, obj, glb, mp4, etc.):", accept_multiple_files=True)
+        ups = st.file_uploader(
+            "Add files (🖼️ png/jpeg, 🏰 obj/glb/gltf, 📹 mp4, etc.):",
+            accept_multiple_files=True
+        )
         st.markdown("### 📋 Uploaded Model Files")
         directory = st.text_input("Path:", ".", key="dir")
         if os.path.isdir(directory):
             files = [f for f in os.listdir(directory) if f.split('.')[-1] in ['obj', 'glb', 'gltf']]
             if files:
                 for i, f in enumerate(files, 1):
-                    st.markdown(f"{i}. {f}")
+                    # Add emoji based on file type and show spawn count 🖌️
+                    ext = f.split('.')[-1]
+                    emoji = "🏰" if ext == 'obj' else "🏯"  # 🏰 for obj, 🏯 for glb/gltf
+                    count = st.session_state.get('model_counts', {}).get(f, 0)
+                    st.markdown(f"{i}. {emoji} {f} (Spawned: {count})")
             else:
-                st.markdown("No model files found.")
+                st.markdown("No model files found. 🕸️")
 
+    # 🚨 5.3 Directory Check: Make sure the path exists! 🛤️
     if not os.path.isdir(directory):
-        st.sidebar.error("Invalid directory")
+        st.sidebar.error("Invalid directory 🚫")
         return
 
+    # 📥 5.4 Handle Uploads: Bring in the new files! 📦
     types = ['obj','glb','gltf','webp','png','jpeg','jpg','mp4']
     if ups:
       for up in ups:
@@ -338,14 +365,16 @@ def main():
         if ext in types:
           with open(os.path.join(directory,up.name),'wb') as f:
             shutil.copyfileobj(up,f)
-          st.sidebar.success(f"Uploaded {up.name}")
+          emoji = "🖼️" if ext in ['webp', 'png', 'jpeg', 'jpg'] else "🏰" if ext in ['obj', 'glb', 'gltf'] else "📹"
+          st.sidebar.success(f"Uploaded {emoji} {up.name}")
         else:
-          st.sidebar.warning(f"Skipped {up.name}")
+          st.sidebar.warning(f"Skipped {up.name} 🚧")
 
     files = [f for f in os.listdir(directory) if f.split('.')[-1] in types]
 
     spot_h = max(8,8)*1.5
 
+    # 🌐 5.5 Build the Scene: Create your 3D masterpiece! 🎬
     scene = f"""
     <a-scene embedded style="height:100vh; width:100vw; position:fixed; top:0; left:0;">
       <a-entity id="rig" position="0 12 0" rotation="-90 0 0">
@@ -362,9 +391,12 @@ def main():
       <a-text id="score" value="Score:0" position="-1.5 2 -3" scale="0.5 0.5 0.5" color="white"></a-text>
     """
 
-    assets, ents = generate_tilemap(files, directory, 8, 8)
+    # 🏗️ 5.6 Add Assets and Entities: Fill the world with wonders! 🌟
+    assets, ents, model_counts = generate_tilemap(files, directory, 8, 8)
+    st.session_state['model_counts'] = model_counts  # Store counts for display 📊
     scene += assets + ents + "</a-scene>"
 
+    # 🎥 5.7 Apply Camera Actions: Lights, camera, action! 🎬
     view_cmd = st.session_state.get('camera_view', 'center')
     if view_cmd:
       scene += f"<script>moveCamera('{view_cmd}');</script>"
@@ -382,5 +414,6 @@ def main():
       height=1000
     )
 
+# 🏁 6. Run the Show: Let’s launch this 3D adventure! 🎭
 if __name__ == "__main__":
     main()