Update app.py

app.py

@@ -35,7 +35,7 @@ with col2:
 # Session State Initialization
 def init_session_state():
     defaults = {
-        'server_running': False, 'server_task': None, 'active_connections': {},
+        'server_running': False, 'active_connections': {},
         'chat_history': [], 'last_chat_update': 0, 'username': None,
         'tts_voice': "en-US-AriaNeural", 'audio_cache': {}
     }
@@ -122,16 +122,12 @@ async def broadcast_message(message, room_id):
             if client_id in st.session_state.active_connections[room_id]:
                 del st.session_state.active_connections[room_id][client_id]
 
-async def run_websocket_server():
+async def start_websocket_server():
     if not st.session_state.get('server_running', False):
         server = await websockets.serve(websocket_handler, '0.0.0.0', 8765)
         st.session_state['server_running'] = True
-        await server.wait_closed()
-
-def start_websocket_server():
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-    loop.run_until_complete(run_websocket_server())
+        st.session_state['server'] = server  # Store server reference
+        await asyncio.Future()  # Run forever until cancelled
 
 # Chat Functions
 async def save_chat_entry(username, message):
@@ -216,7 +212,7 @@ html_code = f"""
         document.getElementById('gameContainer').appendChild(renderer.domElement);
 
         // Lighting
-        const ambientLight = new THREE.AmbientLight(0xffffff, 0.1); // Dim ambient for night effect
+        const ambientLight = new THREE.AmbientLight(0xffffff, 0.1);
         scene.add(ambientLight);
 
         // Ground
@@ -335,8 +331,8 @@ html_code = f"""
             // Add city lights at intersections
             for (let x = -citySize; x <= citySize; x++) {{
                 for (let z = -citySize; z <= citySize; z++) {{
-                    const light = new THREE.PointLight(0xffff99, 1, 20); // Yellowish light, intensity 1, distance 20
-                    light.position.set(x * spacing, 5, z * spacing); // 5 units above ground
+                    const light = new THREE.PointLight(0xffff99, 1, 20);
+                    light.position.set(x * spacing, 5, z * spacing);
                     light.castShadow = true;
                     scene.add(light);
                     lights.push(light);
@@ -345,7 +341,7 @@ html_code = f"""
         }}
 
         function evolveCity() {{
-            if (Math.random() < 0.1) {{ // 10% chance to evolve per frame
+            if (Math.random() < 0.1) {{
                 const citySize = 5, spacing = 15;
                 const x = (Math.random() * 2 - 1) * citySize * spacing;
                 const z = (Math.random() * 2 - 1) * citySize * spacing;
@@ -559,7 +555,7 @@ html_code = f"""
             cycleTime += delta;
             const cycleDuration = 120;
             const angle = (cycleTime / cycleDuration) * Math.PI * 2;
-            const intensity = Math.max(0, Math.sin(angle)) * 0.5 + 0.5; // Varies between 0.5 and 1
+            const intensity = Math.max(0, Math.sin(angle)) * 0.5 + 0.5;
             lights.forEach(light => light.intensity = intensity);
             const dayColor = new THREE.Color(0x87CEEB);
             const nightColor = new THREE.Color(0x001133);
@@ -634,10 +630,15 @@ if audio_bytes:
     message = "Voice message received"
     asyncio.run(save_chat_entry(st.session_state.username, message))
 
-# Start WebSocket Server
-if not st.session_state.get('server_task'):
-    st.session_state.server_task = threading.Thread(target=start_websocket_server, daemon=True)
-    st.session_state.server_task.start()
+# Start WebSocket Server in Main Event Loop
+async def main_async():
+    if not st.session_state.get('server_running', False):
+        await start_websocket_server()
+
+# Run the WebSocket server in the main event loop
+loop = asyncio.get_event_loop()
+if not st.session_state.get('server_running', False):
+    loop.run_until_complete(main_async())
 
 st.sidebar.write("""
 ### How to Play