minor, get back

app.py

@@ -21,6 +21,29 @@ def main() -> None:
         # add_sphere_btn = gr.Button("Add Random Sphere")
         iframe_html = gr.HTML("")
 
+        # @demo.load(outputs=[iframe_html])
+        # def start_server(request: gr.Request):
+        #     assert request.session_hash is not None
+        #     viser_manager.start_server(request.session_hash)
+
+        #     # Use the request's base URL if available
+        #     host = request.headers["host"]
+
+        #     # Determine protocol (use HTTPS for HuggingFace Spaces or other secure environments)
+        #     protocol = (
+        #         "https"
+        #         if request.headers.get("x-forwarded-proto") == "https"
+        #         else "http"
+        #     )
+            
+        #     # Add three random spheres automatically
+        #     server = viser_manager.get_server(request.session_hash)
+        #     visualize_st4rtrack(server=server)
+
+        #     return f"""
+        #     <iframe src="{protocol}://{host}/viser/{request.session_hash}/" width="100%" height="500px" frameborder="0" style="display: block;"></iframe>
+        #     """
+        
         @demo.load(outputs=[iframe_html])
         def start_server(request: gr.Request):
             assert request.session_hash is not None
@@ -38,12 +61,27 @@ def main() -> None:
             
             # Add three random spheres automatically
             server = viser_manager.get_server(request.session_hash)
-            visualize_st4rtrack(server=server)
+            for _ in range(3):
+                add_random_sphere(server)
 
             return f"""
             <iframe src="{protocol}://{host}/viser/{request.session_hash}/" width="100%" height="500px" frameborder="0" style="display: block;"></iframe>
             """
 
+        # @add_sphere_btn.click
+        def add_random_sphere(server):
+            # Add icosphere with random properties
+            server.scene.add_icosphere(
+                name=f"sphere_{random.randint(1, 10000)}",
+                position=(
+                    random.uniform(-1, 1),
+                    random.uniform(-1, 1),
+                    random.uniform(-1, 1),
+                ),
+                radius=random.uniform(0.05, 0.2),
+                color=(random.random(), random.random(), random.random()),
+            )
+
         @demo.unload
         def stop(request: gr.Request):
             assert request.session_hash is not None