Update index.html

index.html

@@ -54,6 +54,7 @@
         document.body.appendChild(renderer.domElement);
 
         const camera = new THREE.PerspectiveCamera(75, window.innerWidth / window.innerHeight, 0.1, 1000);
+        camera.position.set(0, 1, 5);
         const controls = new THREE.OrbitControls(camera, renderer.domElement);
         controls.enableDamping = true;
         controls.dampingFactor = 0.05;
@@ -190,7 +191,8 @@
         const socketStatus = document.getElementById('socket-status');
         const statusText = document.getElementById('status');
         // This automatically grabs the current Hugging Face URL and connects securely
-        const ws = new WebSocket('wss://' + window.location.host + '/api/generate_stream');
+        const wsProtocol = window.location.protocol === 'https:' ? 'wss://' : 'ws://';
+        const ws = new WebSocket(wsProtocol + window.location.host + '/api/generate_stream');
 
         ws.onopen = () => {
             socketStatus.innerHTML = "🟢 Live TCP Connection";