liveblocks auth api

frontend/.env.development.example

@@ -1,3 +1,3 @@
 PUBLIC_WS_INPAINTING="ws://0.0.0.0:7860/gradio/queue/join"
 PUBLIC_UPLOADS="https://d26smi9133w0oo.cloudfront.net/uploads"
-PUBLIC_UPLOAD_URL="/server/uploadfile/"
+PUBLIC_API_BASE="/server/api"

frontend/.env.example

@@ -1,3 +1,3 @@
 PUBLIC_WS_INPAINTING="wss://spaces.huggingface.tech/huggingface-projects/stable-diffusion-multiplayer/gradio/queue/join"
 PUBLIC_UPLOADS="https://d26smi9133w0oo.cloudfront.net/uploads"
-PUBLIC_UPLOAD_URL="/embed/huggingface-projects/stable-diffusion-multiplayer/uploadfile/"
+PUBLIC_API_BASE="/embed/huggingface-projects/stable-diffusion-multiplayer/api"

frontend/src/lib/constants.ts

@@ -5,24 +5,5 @@ export const COLORS = Object.values(colors)
 	.map((e) => e['200'])
 	.slice(0, 18);
 
-// all animal emojis list
-export const EMOJIS = [
-	'🐶',
-	'🐱',
-	'🐭',
-	'🐹',
-	'🐰',
-	'🦊',
-	'🐻',
-	'🐼',
-	'🐨',
-	'🐯',
-	'🦁',
-	'🐮',
-	'🐲',
-	'🌚',
-	'🌝',
-	'🌞',
-	'🌛',
-	'🌜'
-];
+export const EMOJIS = ['🐝', '🐌', '🐞', '🐜', '🦋', '🐛', '🐝', '🐞', '🦟', '🦗', '🕷', '🦂', '🐢', '🐍', '🦎', '🦖', '🦕', '🐙', '🦑', '🐠', '🐟', '🐡', '🐬', '🦈', '🐳', '🐋', '🐊', '🐅', '🐆', '🦓', '🦍', '🦧', '🐘', '🦛', '🦏', '🐪', '🐫', '🦒', '🐃', '🐂', '🐄', '🐎', '🐖',
+	'🐏', '🐑', '🐐', '🐕', '🐩', '🐈', '🐓', '🦃', '🦅', '🦆', '🦢', '🦉', '🦚', '🦜', '🦇', '🐁', '🐀', '🐿', '🐇', '🐿', '🦔', '🦇', '🐻', '🐻', '🐨', '🐼', '🐵', '🙈', '🙉', '🙊', '🐒', '🐉', '🐲', '🦕', '🦖', '🐊', '🐢', '🦎', '🐍', '🐦', '🐧', '🦅', '🦆', '🦉', '🦇']

frontend/src/lib/utils.ts

@@ -1,4 +1,4 @@
-import { PUBLIC_UPLOAD_URL } from '$env/static/public';
+import { PUBLIC_API_BASE } from '$env/static/public';
 
 export function base64ToBlob(base64image: string): Promise<Blob> {
 	return new Promise((resolve) => {
@@ -32,7 +32,7 @@ export async function uploadImage(imagBlob: Blob, prompt: string, key: string):
 	const formData = new FormData()
 	formData.append('file', file)
 
-	const response = await fetch(PUBLIC_UPLOAD_URL, {
+	const response = await fetch(PUBLIC_API_BASE + "/uploadfile", {
 		method: 'POST',
 		body: formData
 	});

frontend/src/routes/+layout.ts

@@ -1 +1,6 @@
-export const prerender = true;
+import { PUBLIC_API_BASE } from '$env/static/public';
+
+export async function load({ fetch }) {
+    const rooms = await fetch(PUBLIC_API_BASE + "/rooms").then((res) => res.json());
+    return { rooms };
+}

frontend/src/routes/+page.svelte

@@ -15,11 +15,18 @@
 	import LiveblocksProvider from '$lib/liveblocks/LiveblocksProvider.svelte';
 	import RoomProvider from '$lib/liveblocks/RoomProvider.svelte';
 	import App from '$lib/App.svelte';
+	import type { PageData } from './$types';
+	import { PUBLIC_API_BASE } from '$env/static/public';
+
+	export let data: PageData;
 
 	let roomId: string;
 	let loaded = false;
 	let client: Client;
 
+	$: {
+		console.log('data changed', data);
+	}
 	onMount(() => {
 		document.addEventListener('wheel', (e) => e.preventDefault(), { passive: false });
 
@@ -29,7 +36,8 @@
 
 		// Connect to the authentication API for Liveblocks
 		client = createClient({
-			publicApiKey: 'pk_test_JlUZGH3kQmhmZQiqU2l8eIi5'
+			// publicApiKey: 'pk_test_JlUZGH3kQmhmZQiqU2l8eIi5'
+			authEndpoint: PUBLIC_API_BASE + '/auth'
 		});
 
 		loaded = true;

stablediffusion-infinity/app.py

@@ -3,7 +3,7 @@ import os
 
 from pathlib import Path
 import uvicorn
-from fastapi import FastAPI, BackgroundTasks, HTTPException, UploadFile, Depends, status
+from fastapi import FastAPI, BackgroundTasks, HTTPException, UploadFile, Depends, status, Request
 from fastapi.staticfiles import StaticFiles
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi_utils.tasks import repeat_every
@@ -21,6 +21,7 @@ import boto3
 import magic
 import sqlite3
 import requests
+import uuid
 
 AWS_ACCESS_KEY_ID = os.getenv('AWS_ACCESS_KEY_ID')
 AWS_SECRET_KEY = os.getenv('AWS_SECRET_KEY')
@@ -36,6 +37,7 @@ FILE_TYPES = {
     'image/jpeg': 'jpg',
 }
 DB_PATH = Path("rooms.db")
+
 app = FastAPI()
 
 print("DB_PATH", DB_PATH)
@@ -55,7 +57,6 @@ def get_db():
         db.close()
 
 
-
 s3 = boto3.client(service_name='s3',
                   aws_access_key_id=AWS_ACCESS_KEY_ID,
                   aws_secret_access_key=AWS_SECRET_KEY)
@@ -115,6 +116,7 @@ def get_model():
     # model["img2img"]
 
 
+# init model on startup
 # get_model()
 
 
@@ -280,13 +282,32 @@ async def sync_rooms():
         print(e)
         print("Rooms update failed")
 
-@app.get('/rooms')
+
+@app.get('/api/rooms')
 async def get_rooms(db: sqlite3.Connection = Depends(get_db)):
     rooms = db.execute("SELECT * FROM rooms").fetchall()
     return rooms
 
 
-@app.post('/uploadfile/')
+@app.post('/api/auth')
+async def autorize(request: Request):
+    data = await request.json()
+    room = data["room"]
+    payload = {
+        "userId": str(uuid.uuid4()),
+        "userInfo": {
+            "name": "Anon"
+        }}
+
+    response = requests.post(f"https://api.liveblocks.io/v2/rooms/{room}/authorize",
+                             headers={"Authorization": f"Bearer {LIVEBLOCKS_SECRET}"}, json=payload)
+    if response.status_code == 200:
+        return response.json()
+    else:
+        raise Exception(response.status_code, response.text)
+
+
+@ app.post('/api/uploadfile/')
 async def create_upload_file(background_tasks: BackgroundTasks, file: UploadFile):
     contents = await file.read()
     file_size = len(contents)