Spaces:
Running
on
Zero
Running
on
Zero
from server import PromptServer | |
from aiohttp import web | |
import os | |
import folder_paths | |
dir = os.path.abspath(os.path.join(__file__, "../../user")) | |
if not os.path.exists(dir): | |
os.mkdir(dir) | |
file = os.path.join(dir, "autocomplete.txt") | |
async def get_autocomplete(request): | |
if os.path.isfile(file): | |
return web.FileResponse(file) | |
return web.Response(status=404) | |
async def update_autocomplete(request): | |
with open(file, "w", encoding="utf-8") as f: | |
f.write(await request.text()) | |
return web.Response(status=200) | |
async def get_loras(request): | |
loras = folder_paths.get_filename_list("loras") | |
return web.json_response(list(map(lambda a: os.path.splitext(a)[0], loras))) | |