Update app.py

app.py

@@ -29,6 +29,9 @@ class ConnectionManager:
         if user_id in self.active_connections:
             del self.active_connections[user_id]
 
+    def isOnline(self, user_id: int):
+        return user_id in self.active_connections
+
     async def send_personal_message(self, message: Dict, user_id: int):
         """Sends a JSON message to a single connected user."""
         if user_id in self.active_connections:
@@ -163,6 +166,13 @@ async def websocket_endpoint(websocket: WebSocket, user_id: int):
     online_status = {"type": "status_update", "user_id": user_id, "is_online": True}
     await manager.broadcast(online_status)
 
+    # Send the other user's status to the newly connected user
+    for other_user_id in VALID_USER_IDS:
+        if other_user_id != user_id:
+            if not manager.isOnline(other_user_id):
+                offline_status = {"type": "status_update", "user_id": other_user_id, "is_online": False, "last_seen": userLastOnlineTime.get(other_user_id, 0.0)}
+                await manager.send_personal_message(offline_status, user_id)
+
     # 3. <--- NEW IMPLEMENTATION HERE --->
     # Attempt to send any messages that accumulated while the user was offline.
     await send_undelivered_messages(user_id)