| 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")
|
|
|
|
|
| @PromptServer.instance.routes.get("/pysssss/autocomplete")
|
| async def get_autocomplete(request):
|
| if os.path.isfile(file):
|
| return web.FileResponse(file)
|
| return web.Response(status=404)
|
|
|
|
|
| @PromptServer.instance.routes.post("/pysssss/autocomplete")
|
| async def update_autocomplete(request):
|
| with open(file, "w", encoding="utf-8") as f:
|
| f.write(await request.text())
|
| return web.Response(status=200)
|
|
|
|
|
| @PromptServer.instance.routes.get("/pysssss/loras")
|
| 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)))
|
|
|
