change init room logic

frontend/src/lib/App.svelte

@@ -29,7 +29,6 @@
 	}
 
 	const promptImgStorage = useObject('promptImgStorage');
-	$: console.log('isRendering', $isRenderingCanvas);
 
 	$: isLoading = $myPresence?.status === Status.loading || $isRenderingCanvas || false;
 	function onShowModal(e: CustomEvent) {

frontend/src/lib/Buttons/RoomsSelector.svelte

@@ -42,7 +42,7 @@
 
 <!-- svelte-ignore a11y-click-events-have-key-events -->
 <div class="min-w-[20ch]">
-	{#if loadingRooms}
+	{#if loadingRooms && selectedRoom}
 		<div
 			class="text-sm rounded-2xl md:text-smtext-gray-700 py-1 font-medium tracking-tight relative ring-1 ring-blue-500 px-2"
 			title="Choose a different room"

frontend/src/lib/PaintFrame.svelte

@@ -30,7 +30,7 @@
 	let frameElement: HTMLDivElement;
 	let dragEnabled = true;
 	let isDragging = false;
-	let firstLoading = true;
+	let firstLoading: boolean | null = null;
 
 	$: prompt = $myPresence?.currentPrompt;
 	$: isLoading =
@@ -41,7 +41,7 @@
 		y: transform.applyY(position.y)
 	};
 
-	$: if (!$isRenderingCanvas && firstLoading) {
+	$: if (firstLoading !== null && !$isRenderingCanvas && firstLoading) {
 		cleanMask();
 		firstLoading = false;
 	}
@@ -98,6 +98,8 @@
 			})
 			.call(maskingHandler() as any)
 			.call(cursorUpdate);
+
+		firstLoading = true;
 	});
 
 	function cursorUpdate(selection: any) {

frontend/src/lib/store.ts

@@ -7,7 +7,7 @@ export const canvasEl = writable<HTMLCanvasElement>();
 export const maskEl = writable<HTMLCanvasElement>();
 export const selectedRoomID = writable<string | null>();
 export const toggleAbout = writable<boolean>(false);
-export const isRenderingCanvas = writable<boolean>(true);
+export const isRenderingCanvas = writable<boolean>(false);
 export const showModal = writable<boolean>(false);
 export const canvasSize = writable<{
     width: number;

frontend/src/routes/+page.svelte

@@ -42,21 +42,25 @@
 		const rooms: RoomResponse[] = await res.json();
 		const emptyRoom = rooms.find((room) => room.users_count < MAX_CAPACITY) || null;
 
-		let roomAvailable = false;
+		let queriedRoom: string | null = null;
+		// init if roomid is set via param
 		if (roomidParam) {
-			const room = rooms.find((room) => room.room_id === roomidParam) || null;
-			roomAvailable = room ? room.users_count < MAX_CAPACITY : false;
-			if (room && roomAvailable) {
-				$selectedRoomID = room.room_id;
-				const state = { roomid: room.room_id };
-				const queryString = '?' + new URLSearchParams(state).toString();
-				window.history.replaceState(null, '', queryString);
-				window.parent.postMessage({ queryString: queryString }, '*');
+			// if room is unlisted, skip the check
+			if (roomidParam.startsWith('secret-')) {
+				queriedRoom = roomidParam;
+			} else {
+				// if room is listed, check if it's full
+				const room = rooms.find((room) => room.room_id === roomidParam) || null;
+				queriedRoom = room && room.users_count < MAX_CAPACITY ? room.room_id : null;
 			}
+		} else {
+			// if roomid is not set via param, select the first empty room
+			queriedRoom = emptyRoom ? emptyRoom.room_id : null;
 		}
-		if (emptyRoom && !roomAvailable) {
-			selectedRoomID.set(emptyRoom.room_id);
-			const state = { roomid: emptyRoom.room_id };
+		// if seleceted room is full, select the first empty room
+		if (queriedRoom) {
+			$selectedRoomID = queriedRoom;
+			const state = { roomid: queriedRoom };
 			const queryString = '?' + new URLSearchParams(state).toString();
 			window.history.replaceState(null, '', queryString);
 			window.parent.postMessage({ queryString: queryString }, '*');

requirements.txt

@@ -13,4 +13,5 @@ gradio==3.6
 boto3==1.24.93
 python-magic==0.4.27
 fastapi-utils==0.2.1
-shortuuid==1.0.9
+shortuuid==1.0.
+tqdm==4.64

stablediffusion-infinity/createRooms.py

@@ -53,6 +53,16 @@ async def create_room(db: sqlite3.Connection = Depends(get_db)):
     return all
 
 
+@app.get("/create-one-room/{room_id}")
+async def create_one_room(room_id: str):
+    payload = {"id": room_id, "defaultAccesses": ["room:write"]}
+
+    response = requests.post(f"https://api.liveblocks.io/v2/rooms",
+                             headers={"Authorization": f"Bearer {LIVEBLOCKS_SECRET}"}, json=payload)
+    data = response.json()
+    return data
+
+
 def createRoom(room_id, db):
     payload = {"id": room_id, "defaultAccesses": ["room:write"]}
 
@@ -117,4 +127,5 @@ async def sync_rooms(db: sqlite3.Connection = Depends(get_db)):
 
 
 if __name__ == "__main__":
-    uvicorn.run("createRooms:app", host="0.0.0.0", log_level="debug", reload=True)
+    uvicorn.run("createRooms:app", host="0.0.0.0",
+                log_level="debug", reload=True)