router.py

from aiohttp import web
import os
import folder_paths

TEMP_DIR = folder_paths.get_temp_directory()
BRIDGE_DIR = os.path.join(TEMP_DIR, "tensor_bridge")
os.makedirs(BRIDGE_DIR, exist_ok=True)


async def upload_tensor_handler(request):
    try:
        reader = await request.multipart()
        channel_name = "default"
        file_written = False
        bytes_written = 0

        field = await reader.next()
        while field:
            if field.name == 'channel':
                value = await field.read(decode=True)
                channel_name = value.decode("utf-8")
            elif field.name == 'file':
                filename = f"{channel_name}.pt"
                filepath = os.path.join(BRIDGE_DIR, filename)

                # Write stream to disk
                with open(filepath, "wb") as f:
                    while True:
                        chunk = await field.read_chunk()
                        if not chunk: break
                        f.write(chunk)
                        bytes_written += len(chunk)
                file_written = True
            field = await reader.next()

        if file_written:
            return web.json_response({"status": "success", "channel": channel_name, "size": bytes_written})
        return web.json_response({"status": "error", "message": "No file field found"}, status=400)

    except Exception as e:
        print(f"[TensorBridge] Upload Error: {e}")
        return web.json_response({"status": "error", "message": str(e)}, status=500)