Upload folder via script

Dockerfile

@@ -13,10 +13,6 @@ RUN sh /uv-installer.sh && rm /uv-installer.sh
 # Ensure the installed binary is on the PATH
 ENV PATH="/root/.local/bin:${PATH}"
 
-# Reproducible venv inside the image (no host .venv symlinks; see .dockerignore)
-ENV UV_COMPILE_BYTECODE=1
-ENV UV_LINK_MODE=copy
-
 # Set the work directory inside the container
 WORKDIR /app
 
@@ -35,5 +31,5 @@ ENV VELAI_PORT=7860
 
 EXPOSE 7860
 
-# uv run uses the .venv from `uv sync` above (.dockerignore keeps host .venv out)
-CMD ["uv", "run", "--frozen", "--no-dev", "python", "main.py"]
+# Default command to start your app
+CMD ["uv", "run", "--no-sync", "--no-dev", "python", "main.py"]

README.md

@@ -37,8 +37,8 @@ VELAI is designed for educators, facilitators, and creators who want a lightweig
 - Open the browser at `http://127.0.0.1:7860`. If a password is configured, enter it to access the canvas.
 
 **Run with Docker**
-- Build the image: `docker build -t VELAI .`.
-- Start a container: `docker run -p 7860:7860 -e VELAI_APP_PASSWORD=yourpassword -e VELAI_STORAGE_SECRET=changeme -v $(pwd)/.storage:/app/.storage VELAI`.
+- Build the image: `docker build -t velai .`.
+- Start a container: `docker run -p 7860:7860 -e VELAI_APP_PASSWORD=yourpassword -e VELAI_STORAGE_SECRET=changeme -e VELAI_ADMIN_PASSWORD=youradminpassword -e FAL_KEY=falapikey -v $(pwd)/.storage:/app/.storage velai`.
 - Visit `http://127.0.0.1:7860` to begin exploring.
 
 ## First session walkthrough

velai/app_context.py

@@ -77,10 +77,7 @@ async def get_or_create_app_context(client: Client) -> AppContext:
     # init storage
     app_storage = DocumentStorage(NiceGuiKeyValueStorage[dict[str, Any]](NiceGuiStorageType.General))
     user_storage = DocumentStorage(NiceGuiKeyValueStorage[dict[str, Any]](NiceGuiStorageType.User))
-    blob_storage: BlobStorage = FileSystemBlobStorage(
-        user_id=session_id, root_dir=environment.VELAI_BLOB_STORAGE_PATH, document_storage=user_storage
-    )
-    async_blob_storage = AsyncBlobStorageAdapter(blob_storage)
+    async_blob_storage = create_blob_storage_for_session(session_id)
 
     # create and load app config
     config = _create_config(app_storage)
@@ -115,10 +112,15 @@ async def current_app_context() -> AppContext:
         raise AppContextNotInitializedError("AppContext not initialized") from exc
 
 
-async def app_context_from_request(request: Request) -> AppContext:
+def _session_id_from_request(request: Request) -> str:
     session_id = request.session.get("id")
     if not session_id:
         raise AppContextNotInitializedError("missing session")
+    return session_id
+
+
+async def app_context_from_request(request: Request) -> AppContext:
+    session_id = _session_id_from_request(request)
 
     try:
         wrapper = await _app_contexts.get(session_id)
@@ -127,6 +129,33 @@ async def app_context_from_request(request: Request) -> AppContext:
         raise AppContextNotInitializedError("AppContext not initialized") from exc
 
 
+def create_blob_storage_for_session(session_id: str) -> AsyncBlobStorageAdapter:
+    """Filesystem-backed blob storage for a browser session (no in-memory AppContext required)."""
+    user_storage = DocumentStorage(NiceGuiKeyValueStorage[dict[str, Any]](NiceGuiStorageType.User))
+    blob_storage: BlobStorage = FileSystemBlobStorage(
+        user_id=session_id,
+        root_dir=environment.VELAI_BLOB_STORAGE_PATH,
+        document_storage=user_storage,
+    )
+    return AsyncBlobStorageAdapter(blob_storage)
+
+
+async def blob_storage_from_request(request: Request) -> AsyncBlobStorageAdapter:
+    """
+    Resolve blob storage for HTTP requests (e.g. <img src="/storage/...">).
+
+    Image loads are separate HTTP requests and may arrive while the WebSocket is
+    reconnecting or on another worker; they must not depend on in-memory AppContext.
+    """
+    session_id = _session_id_from_request(request)
+
+    if await _app_contexts.has(session_id):
+        wrapper = await _app_contexts.get(session_id)
+        return wrapper.context.blob_storage
+
+    return create_blob_storage_for_session(session_id)
+
+
 def register_app_context_cleanup(client: Client) -> None:
     session_id = _browser_session_id()
 

velai/session.py

@@ -152,9 +152,12 @@ class GraphSession:
         return await self.restore_from_server_storage()
 
     async def save_to_server_storage(self) -> None:
-        data = self.to_json()
+        try:
+            ctx = await app_context.current_app_context()
+        except app_context.AppContextNotInitializedError:
+            return
 
-        ctx = await app_context.current_app_context()
+        data = self.to_json()
         ctx.user_storage[GRAPH_STORAGE_KEY] = data
 
     async def restore_from_server_storage(self) -> bool:

velai/storage/storage_endpoint.py

@@ -65,10 +65,10 @@ def register():
 
     @app.get(storage_endpoint)
     async def download_blob(request: Request, blob_id: str, original_name: str) -> StreamingResponse:
-        ctx = await app_context.app_context_from_request(request)
+        blob_storage = await app_context.blob_storage_from_request(request)
 
         try:
-            blob = await ctx.blob_storage.require(blob_id)
+            blob = await blob_storage.require(blob_id)
         except KeyError as exc:
             raise HTTPException(status_code=404, detail="blob not found") from exc