0153b2c8a8
* update ci method & fix key page bug * add list model method * refine * Update workflow-ci.yml * Support the use of uploaded Lora (#85) * Support the use of uploaded Lora * refine * refine * refine * refine * refine * reformat * add test workflow * fix error * fix error * fix vaild error * refine * refine * remove Redundant warning * skip flux lora * use ini file instead of cookies in daily use (#91) * debug ci * refine * refine * get key from cookies * set key to ini file * print error msg when set api key failed * revert menu json file * revert code, becuase of the error reoccur * remove raise error, instead of using warning * rever workflow ci yaml * refine test output * add workflow dspatch event * refine * fix encode_comfy_image & decode_comfy_image * [feat]set api key home page refresh * update dev_config.yaml * add FluxGuidance node * fix error * black --------- Co-authored-by: Yao Chi <later@usopp.net> Co-authored-by: of-HaiQin <5372604@qq.com>
80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
import os
|
|
import uuid
|
|
from pathlib import Path
|
|
|
|
import server
|
|
from aiohttp import web
|
|
|
|
import bizyair
|
|
from bizyair.common import load_api_key, create_api_key_file
|
|
|
|
|
|
API_KEY = None
|
|
html_file_path = Path(os.path.dirname(os.path.abspath(__file__))) / "set_api_key.html"
|
|
with open(html_file_path, "r", encoding="utf-8") as htmlfile:
|
|
set_api_key_html = htmlfile.read()
|
|
|
|
|
|
has_key, api_key = load_api_key()
|
|
if has_key:
|
|
API_KEY = api_key
|
|
|
|
|
|
@server.PromptServer.instance.routes.get("/bizyair/set-api-key")
|
|
async def set_api_key_page(request):
|
|
return web.Response(text=set_api_key_html, content_type="text/html")
|
|
|
|
|
|
@server.PromptServer.instance.routes.post("/bizyair/set_api_key")
|
|
async def set_api_key(request):
|
|
global API_KEY
|
|
try:
|
|
data = await request.post()
|
|
api_key = data.get("api_key")
|
|
if api_key:
|
|
create_api_key_file(api_key)
|
|
API_KEY = api_key
|
|
bizyair.set_api_key(API_KEY)
|
|
return web.Response(text="ok")
|
|
else:
|
|
error_msg = (
|
|
"No token provided, please refer to cloud.siliconflow.cn to get the key"
|
|
)
|
|
print("set_api_key:", erro_msg)
|
|
return web.Response(
|
|
text=error_msg,
|
|
status=400,
|
|
)
|
|
except Exception as e:
|
|
print(f"set api key error: {str(e)}")
|
|
return web.Response(text=str(e), status=500)
|
|
|
|
|
|
@server.PromptServer.instance.routes.get("/bizyair/get_api_key")
|
|
async def get_api_key(request):
|
|
global API_KEY
|
|
try:
|
|
has_key, api_key = load_api_key()
|
|
if has_key:
|
|
API_KEY = api_key
|
|
bizyair.set_api_key(API_KEY)
|
|
return web.Response(text="Key has been loaded from the api_key.ini file")
|
|
else:
|
|
api_key = request.cookies.get("api_key")
|
|
if not api_key:
|
|
print("No api key found in cookies")
|
|
return web.Response(
|
|
text="No api key found in cookies, please refer to cloud.siliconflow.cn to get the key",
|
|
status=404,
|
|
)
|
|
API_KEY = api_key
|
|
bizyair.set_api_key(API_KEY)
|
|
return web.Response(text="Key has been loaded from the cookies")
|
|
|
|
except Exception as e:
|
|
return web.Response(text="str(e)", status=500)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|