1491 lines
49 KiB
Python
1491 lines
49 KiB
Python
import os
|
||
|
||
from aiohttp import web
|
||
from server import PromptServer
|
||
|
||
from .ai_server.siliconflow import getModelList, translateObject
|
||
from .ai_translator.ai_translator import (
|
||
add_openai_setting,
|
||
delete_openai_setting,
|
||
initialize_config,
|
||
set_select_openai,
|
||
update_openai_settings,
|
||
)
|
||
from .cloud_warehouse.save_history import get_package_paths
|
||
|
||
# from .translate.api_translate import installTranslateApi,applyTranslateApi,translateText
|
||
from .cloud_warehouse.warehouse import get_main_warehouse, get_warehouse_tree
|
||
from .dao.dao import install_cloud_file_db
|
||
from .fast_autocomplete.autocomplete import fuzzy_search
|
||
from .history.collect_history_contl import *
|
||
from .history.history_contl import *
|
||
from .prompt_api.danbooru import *
|
||
from .prompt_api.lora_info import *
|
||
from .prompt_api.lora_networks import *
|
||
from .prompt_api.random_tag_template import *
|
||
from .prompt_api.tags_manager import *
|
||
from .translate.local_translate import translate_phrase
|
||
from .translate.new_translate_api import api_service_translate
|
||
from .translate.openai_translate import openai_translate
|
||
from .user_init.user_init import *
|
||
from .prompt_api.tag_image_manager import get_tag_image_info, image_dir, thumb_dir, delete_tag_image
|
||
from .prompt_api.comfyui_workflow import get_generation_options
|
||
from .prompt_api.tag_image_queue import enqueue_generation, get_task_status, get_task_status_by_tuuid, cancel_all_tasks, start_worker
|
||
|
||
static_path = os.path.join(os.path.dirname(__file__), "../../dist/")
|
||
dir = os.path.join(os.path.dirname(__file__), "../../dist/javascript/")
|
||
baseUrl = "/weilin/prompt_ui/api/"
|
||
|
||
|
||
@PromptServer.instance.routes.get("/weilin/prompt_ui/webjs")
|
||
async def _getWeiLinPromptUIWebJs(request):
|
||
# 合并JS（按文件名排序，确保补丁在最后加载）
|
||
js = ""
|
||
for file in sorted(os.listdir(dir)):
|
||
if file.endswith(".js"):
|
||
with open(os.path.join(dir, file), encoding="utf-8") as f:
|
||
js += f.read() + "\n"
|
||
return web.Response(status=200, text=js, content_type="application/javascript")
|
||
|
||
|
||
@PromptServer.instance.routes.get("/weilin/prompt_ui/file/{file_path:.*}")
|
||
async def _getWeiLinPromptUIFile(request):
|
||
file_path = request.match_info.get("file_path", "")
|
||
full_path = os.path.join(static_path, file_path)
|
||
# 检查文件路径是否存在
|
||
if os.path.isfile(full_path):
|
||
return web.FileResponse(full_path)
|
||
raise web.HTTPNotFound()
|
||
|
||
|
||
# ============================================= Lora列表获取 ============================================
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "get_lora_list")
|
||
async def _get_extra_networks(request):
|
||
return web.json_response({"data": await get_extra_networks(auto_fetch=False)})
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "get_lora_load_all")
|
||
async def _get_extra_networks_load_all(request):
|
||
return web.json_response({"data": await get_extra_networks(auto_fetch=True)})
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "get_lora_load_status")
|
||
async def _get_lora_load_status(request):
|
||
return web.json_response({"data": loading_status})
|
||
|
||
|
||
# 新写法 日期2025-04-14 拆分获取的方式
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "get_lora_folder_list")
|
||
async def _get_lora_folder_list(request):
|
||
return web.json_response({"data": get_lora_folder()})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "get_lora_list_by_range")
|
||
async def _get_lora_folder_list(request):
|
||
data = await request.json()
|
||
return web.json_response({"data": await get_rang_for_extra_networks(data["range"])})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "get_lora_list_by_search")
|
||
async def _get_lora_folder_list(request):
|
||
data = await request.json()
|
||
return web.json_response({"data": await search_lora_files(data["search"])})
|
||
|
||
|
||
# =======================================================================================================
|
||
|
||
# ============================================= Lora 信息 ============================================
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "lorainfo/api/loras/info")
|
||
async def _api_save_lora_data(request):
|
||
"""Saves data to a lora by name."""
|
||
api_response = {"status": 200}
|
||
lora_file = get_param(request, "file")
|
||
if lora_file is None:
|
||
api_response["status"] = "404"
|
||
api_response["error"] = "No Lora found at path"
|
||
else:
|
||
post = await request.post()
|
||
await set_model_info_partial(lora_file, json.loads(post.get("json")))
|
||
info_data = await get_model_info(lora_file)
|
||
api_response["data"] = info_data
|
||
return web.json_response(api_response)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "lorainfo/api/delete/loras/info/filed")
|
||
async def _remove_user_diy_fields(request):
|
||
"""Delete data to a lora by name."""
|
||
api_response = {"status": 200}
|
||
lora_file = get_param(request, "file")
|
||
if lora_file is None:
|
||
api_response["status"] = "404"
|
||
api_response["error"] = "No Lora found at path"
|
||
else:
|
||
data = await request.json()
|
||
# print(post.get("json"))
|
||
await remove_user_diy_fields(lora_file, data["json"])
|
||
info_data = await get_model_info(lora_file)
|
||
api_response["data"] = info_data
|
||
return web.json_response(api_response)
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "lorainfo/api/loras/info")
|
||
async def _api_get_loras_info(request):
|
||
"""Returns a list loras info; either all or a single if provided a 'file' param."""
|
||
lora_file = get_param(request, "file")
|
||
maybe_fetch_metadata = lora_file is not None
|
||
if not is_param_falsy(request, "light"):
|
||
maybe_fetch_metadata = False
|
||
api_response = await get_loras_info_response(
|
||
request, maybe_fetch_metadata=maybe_fetch_metadata
|
||
)
|
||
return web.json_response(api_response)
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "lorainfo/api/loras/info/clear")
|
||
async def _delete_lora_info(request):
|
||
"""Clears lora info from the filesystem for the provided file."""
|
||
api_response = {"status": 200}
|
||
lora_file = get_param(request, "file")
|
||
del_info = not is_param_falsy(request, "del_info")
|
||
del_metadata = not is_param_falsy(request, "del_metadata")
|
||
del_civitai = not is_param_falsy(request, "del_civitai")
|
||
if lora_file is None:
|
||
api_response["status"] = "404"
|
||
api_response["error"] = "No Lora file provided"
|
||
elif (
|
||
lora_file == "ALL"
|
||
): # Force the user to supply file=ALL to trigger all clearing.
|
||
lora_files = folder_paths.get_filename_list("loras")
|
||
for lora_file in lora_files:
|
||
await delete_model_info(
|
||
lora_file,
|
||
del_info=del_info,
|
||
del_metadata=del_metadata,
|
||
del_civitai=del_civitai,
|
||
)
|
||
else:
|
||
await delete_model_info(
|
||
lora_file,
|
||
del_info=del_info,
|
||
del_metadata=del_metadata,
|
||
del_civitai=del_civitai,
|
||
)
|
||
return web.json_response(api_response)
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "lorainfo/api/loras/info/refresh")
|
||
async def _refresh_get_loras_info(request):
|
||
"""Refreshes lora info; either all or a single if provided a 'file' param."""
|
||
api_response = await get_loras_info_response(
|
||
request, maybe_fetch_civitai=True, maybe_fetch_metadata=True
|
||
)
|
||
return web.json_response(api_response)
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "lorainfo/api/loras/img")
|
||
async def _api_get_loras_info_img(request):
|
||
"""Returns an image response if one exists for the lora."""
|
||
lora_file = get_param(request, "file")
|
||
lora_path = folder_paths.get_full_path("loras", lora_file)
|
||
if not path_exists(lora_path):
|
||
lora_path = os.path.abspath(lora_path)
|
||
|
||
img_path = None
|
||
for ext in ["jpg", "png", "jpeg", "gif", "webp"]:
|
||
try_path = f"{os.path.splitext(lora_path)[0]}.{ext}"
|
||
if path_exists(try_path):
|
||
img_path = try_path
|
||
break
|
||
|
||
if not path_exists(img_path):
|
||
api_response = {}
|
||
api_response["status"] = "404"
|
||
api_response["error"] = "No Lora found at path"
|
||
return web.json_response(api_response)
|
||
|
||
return web.FileResponse(img_path)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "lorainfo/api/loras/set/img")
|
||
async def _upload_image(request):
|
||
post = await request.post()
|
||
data = image_upload(post)
|
||
return web.json_response(data)
|
||
|
||
|
||
# ======================================================================================================================
|
||
|
||
# ================================= Tag管理 =================================
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "prompt/get_group_tags")
|
||
async def _get_group_tags(request):
|
||
try:
|
||
data = await get_group_tags() # 添加await
|
||
return web.json_response({"data": data})
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "prompt/get_group_tags_paginated")
|
||
async def _get_group_tags_paginated(request):
|
||
try:
|
||
# 支持分页参数
|
||
page = int(request.rel_url.query.get("page", 1))
|
||
page_size = int(request.rel_url.query.get("page_size", 500))
|
||
|
||
data = await get_group_tags_paginated(page, page_size)
|
||
return web.json_response(data)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/add_new_f_group")
|
||
async def _add_new_f_group(request):
|
||
data = await request.json()
|
||
resp = {"code": 202}
|
||
try:
|
||
resp = await add_new_node_group(data["name"], data["color"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response(resp)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/edit_f_group")
|
||
async def _edit_s_group(request):
|
||
data = await request.json()
|
||
resp = {"code": 202}
|
||
try:
|
||
resp = await edit_node_group(data["id_index"], data["name"], data["color"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response(resp)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/delete_f_group")
|
||
async def _delete_s_group(request):
|
||
data = await request.json()
|
||
try:
|
||
delete_node_group(data["p_uuid"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/add_new_s_group")
|
||
async def _add_new_s_group(request):
|
||
data = await request.json()
|
||
resp = {"code": 202}
|
||
try:
|
||
resp = await add_new_group(
|
||
data["key"], data["name"], data["color"], data["p_uuid"]
|
||
)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response(resp)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/edit_s_group")
|
||
async def _edit_new_s_group(request):
|
||
data = await request.json()
|
||
resp = {"code": 202}
|
||
try:
|
||
resp = await edit_child_node_group(
|
||
data["id_index"], data["name"], data["color"]
|
||
)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response(resp)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/delete_s_group")
|
||
async def _delete_new_s_group(request):
|
||
data = await request.json()
|
||
try:
|
||
delete_child_node_group(data["g_uuid"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/new_tags")
|
||
async def _new_tags(request):
|
||
data = await request.json()
|
||
try:
|
||
await add_group_tag(
|
||
data["text"], data["desc"], data["id_index"], data["color"], data["g_uuid"]
|
||
)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/edit_tags")
|
||
async def _edit_tags(request):
|
||
data = await request.json()
|
||
try:
|
||
await edit_group_tag(
|
||
data["text"], data["desc"], data["id_index"], data["color"]
|
||
)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/delete_tags")
|
||
async def _delete_tags(request):
|
||
data = await request.json()
|
||
try:
|
||
await delete_group_tag(data["id_index"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/batch_delete_tags")
|
||
async def _batch_delete_tags(request):
|
||
data = await request.json()
|
||
try:
|
||
await batch_delete_group_tags(data["id_indexs"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/move_tag")
|
||
async def _move_tag(request):
|
||
data = await request.json()
|
||
try:
|
||
result = await move_tag(
|
||
data["id_index"], data["reference_id_index"], data["position"]
|
||
)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": result})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/get_groups_list")
|
||
async def _get_groups_list(request):
|
||
try:
|
||
result = await get_groups_list()
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": result})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/run_sql_text")
|
||
async def _run_sql_text(request):
|
||
data = await request.json()
|
||
try:
|
||
result = run_sql_text(data["sql"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response(result)
|
||
|
||
|
||
# 2025-05-06 新增 移动分类功能
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/move_group")
|
||
async def _move_group(request):
|
||
data = await request.json()
|
||
try:
|
||
result = await move_group(
|
||
data["id_index"], data["reference_id_index"], data["position"]
|
||
)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": result})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/move_subgroup")
|
||
async def _move_subgroup(request):
|
||
data = await request.json()
|
||
try:
|
||
result = await move_subgroup(
|
||
data["id_index"], data["reference_id_index"], data["position"]
|
||
)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": result})
|
||
|
||
|
||
# 2025-05-06 新增 查询功能
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/get_tag_groups")
|
||
async def _get_tag_groups(request):
|
||
resp = {"code": 202}
|
||
try:
|
||
resp = await get_tag_groups()
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response(resp)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/get_tag_subgroups")
|
||
async def _get_tag_subgroups(request):
|
||
data = await request.json()
|
||
resp = {"code": 202}
|
||
try:
|
||
resp = await get_tag_subgroups(data["p_uuid"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response(resp)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/get_tag_tags")
|
||
async def _get_tag_tags(request):
|
||
data = await request.json()
|
||
resp = {"code": 202}
|
||
try:
|
||
resp = await get_tag_tags(data["g_uuid"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response(resp)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/search_tags")
|
||
async def _search_tags(request):
|
||
data = await request.json()
|
||
resp = {"code": 202}
|
||
try:
|
||
resp = await search_tags(data["keyword"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response(resp)
|
||
|
||
|
||
# =====================================================================================================
|
||
|
||
# ================================= 历史记录功能 =================================
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "prompt/history/get_history")
|
||
async def _get_history(request):
|
||
try:
|
||
data = await read_history()
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"data": data})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/history/add_history")
|
||
async def _add_history(request):
|
||
data = await request.json()
|
||
try:
|
||
new_entry = await add_history(data["tag"])
|
||
except Exception as e:
|
||
import traceback
|
||
|
||
print(f"Error in add_history: {e}")
|
||
traceback.print_exc()
|
||
return web.json_response({"error": str(e)}, status=500)
|
||
|
||
return web.json_response({"data": new_entry})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/history/delete_history")
|
||
async def _delete_history(request):
|
||
data = await request.json()
|
||
try:
|
||
delete_info = await delete_history(data["id_index"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": delete_info})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/history/batch_delete_history")
|
||
async def _batch_delete_history(request):
|
||
data = await request.json()
|
||
try:
|
||
delete_info = await batch_delete_history(data["id_index"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": delete_info})
|
||
|
||
|
||
# =====================================================================================================
|
||
|
||
# ================================= 收藏历史记录功能 =================================
|
||
|
||
|
||
@PromptServer.instance.routes.get(
|
||
baseUrl + "prompt/collect_history/get_collect_history"
|
||
)
|
||
async def _get_collect_history(request):
|
||
try:
|
||
data = await read_collect_history()
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"data": data})
|
||
|
||
|
||
@PromptServer.instance.routes.post(
|
||
baseUrl + "prompt/collect_history/add_collect_history"
|
||
)
|
||
async def _add_collect_history(request):
|
||
data = await request.json()
|
||
try:
|
||
new_entry = await add_collect_history(
|
||
data["tag"], data.get("name", ""), data.get("color", "")
|
||
)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"data": new_entry})
|
||
|
||
|
||
@PromptServer.instance.routes.post(
|
||
baseUrl + "prompt/collect_history/delete_collect_history"
|
||
)
|
||
async def _delete_collect_history(request):
|
||
data = await request.json()
|
||
try:
|
||
delete_info = await delete_collect_history(data["id_index"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": delete_info})
|
||
|
||
|
||
@PromptServer.instance.routes.post(
|
||
baseUrl + "prompt/collect_history/batch_delete_collect_history"
|
||
)
|
||
async def _batch_delete_collect_history(request):
|
||
data = await request.json()
|
||
try:
|
||
delete_info = await batch_delete_collect_history(data["id_indices"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": delete_info})
|
||
|
||
|
||
@PromptServer.instance.routes.post(
|
||
baseUrl + "prompt/collect_history/edit_collect_history"
|
||
)
|
||
async def _edit_collect_history(request):
|
||
data = await request.json()
|
||
try:
|
||
edit_info = await edit_collect_history(
|
||
data["id_index"], data.get("name"), data.get("color"), data.get("tag")
|
||
)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": edit_info})
|
||
|
||
|
||
# =====================================================================================================
|
||
|
||
# ================================= 用户初始化设置 =================================
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "user_init/get_settings")
|
||
async def _get_user_settings(request):
|
||
try:
|
||
data = read_init_file()
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"data": data})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "user_init/set_user_lang")
|
||
async def _set_user_lang(request):
|
||
data = await request.json()
|
||
try:
|
||
set_user_lang(data["user_lang"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
# =====================================================================================================
|
||
|
||
# ================================= 翻译功能 =================================
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/local/translate")
|
||
async def _translate_phrase(request):
|
||
data = await request.json()
|
||
phrase = data.get("phrase", "")
|
||
try:
|
||
translated = await translate_phrase(phrase)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"original": phrase, "translated": translated})
|
||
|
||
|
||
# =====================================================================================================
|
||
|
||
# ================================= 快速补全功能 =================================
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/fast/autocomplete")
|
||
async def _autocomplete(request):
|
||
data = await request.json()
|
||
query = data.get("query", "")
|
||
try:
|
||
limit = get_auto_limit_setting() # 获取设置的参数信息
|
||
results = await fuzzy_search(query, limit)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"data": results})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "get/setting/get_auto_limit_setting")
|
||
async def _get_auto_limit_setting(request):
|
||
try:
|
||
limit = get_auto_limit_setting() # 获取设置的参数信息
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"data": limit})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "update/setting/update_auto_limit_setting")
|
||
async def _update_auto_limit_setting(request):
|
||
data = await request.json()
|
||
try:
|
||
update_auto_limit_setting(data.get("limit")) # 获取设置的参数信息
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
# =====================================================================================================
|
||
|
||
# ================================= OpenAI 功能 =================================
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "prompt/openai/get_settings")
|
||
async def _get_openai_setting(request):
|
||
return web.json_response({"data": initialize_config()})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/openai/update_settings")
|
||
async def _update_openai_settings(request):
|
||
data = await request.json()
|
||
index = data.get("index")
|
||
api_key = data.get("api_key")
|
||
base_url = data.get("base_url")
|
||
model = data.get("model")
|
||
try:
|
||
update_openai_settings(index, api_key, base_url, model)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/openai/add_setting")
|
||
async def _add_openai_setting(request):
|
||
data = await request.json()
|
||
api_key = data.get("api_key")
|
||
base_url = data.get("base_url")
|
||
model = data.get("model")
|
||
try:
|
||
add_openai_setting(api_key, base_url, model)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/openai/delete_setting")
|
||
async def _delete_openai_setting(request):
|
||
data = await request.json()
|
||
index = data.get("index")
|
||
try:
|
||
delete_openai_setting(index)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "prompt/openai/set_select")
|
||
async def _set_select_openai(request):
|
||
data = await request.json()
|
||
index = data.get("index")
|
||
try:
|
||
set_select_openai(index)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
# =====================================================================================================
|
||
|
||
# =================================== 启动面板 无法实现 暂时不考虑 =============================================
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "panel/start")
|
||
async def _start_panel(request):
|
||
try:
|
||
return web.json_response(
|
||
{"status": "success", "message": "Panel started on port 9898"}
|
||
)
|
||
except Exception as e:
|
||
print(f"Error starting panel: {e}")
|
||
return web.json_response({"status": "error", "message": str(e)}, status=500)
|
||
|
||
|
||
# =====================================================================================================
|
||
|
||
# =================================== 翻译API库 =============================================
|
||
# @PromptServer.instance.routes.post(baseUrl+"translate/get/packages/state")
|
||
# async def _checkHasInstallTranslateApi(request):
|
||
# apply = applyTranslateApi()
|
||
# if apply == False :
|
||
# return web.json_response({"info": 'fail'})
|
||
# return web.json_response({"info": 'ok'})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "translate/get/setting")
|
||
async def _getTranslaterSettingData(request):
|
||
try:
|
||
data = get_translate_setting()
|
||
except Exception as e:
|
||
print(f"[WeiLin] Error in get_translate_setting: {e}")
|
||
import traceback
|
||
|
||
traceback.print_exc()
|
||
return web.json_response({"data": "translater", "error": str(e)}, status=200)
|
||
|
||
return web.json_response({"data": data})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "translate/apply_setting")
|
||
async def _apply_translater_setting(request):
|
||
data = await request.json()
|
||
setting = data["setting"]
|
||
|
||
if setting == "openai":
|
||
# 可选：这里加一层检测，比如确认 openai 的 api_key 是否已配置
|
||
from .ai_translator.ai_translator import initialize_config
|
||
|
||
cfg = initialize_config()
|
||
if not cfg.get("api_key"):
|
||
return web.json_response({"info": "fail", "msg": "OpenAI API key 未配置"})
|
||
|
||
try:
|
||
update_translate_setting(setting)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
# @PromptServer.instance.routes.post(baseUrl+"translate/install/translaterpackage")
|
||
# async def _apply_translater_install_package(request):
|
||
# try:
|
||
# installTranslateApi()
|
||
# except Exception as e:
|
||
# print(f"Error: {e}")
|
||
# return web.Response(status=500)
|
||
# return web.json_response({"info": 'ok'})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "translate/get/tran_setting")
|
||
async def _get_tanslater_tsetting(request):
|
||
try:
|
||
data = get_translate_settings()
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"data": data})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "translate/save_tran/setting")
|
||
async def _save_translater_setting(request):
|
||
data = await request.json()
|
||
try:
|
||
update_translate_settings(
|
||
data["service"], data["source_lang"], data["target_lang"]
|
||
)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
# 翻译文本
|
||
# POST /translate/tran/text
|
||
@PromptServer.instance.routes.post(baseUrl + "translate/tran/text")
|
||
async def _tanslater_text(request):
|
||
data = await request.json()
|
||
try:
|
||
# 已有方法：读取翻译器选择 & 详细设置
|
||
# 'translater' / 'openai' / （兼容）'network' ...
|
||
current_setting = get_translate_setting()
|
||
data_setting = get_translate_settings() # service/source_lang/target_lang
|
||
|
||
text = data["text"]
|
||
data["str_object"]
|
||
if current_setting == "openai":
|
||
# OpenAI 只需要目标语种来写 prompt
|
||
# result = await openai_translate(strObjectData, data_setting['translate_target_lang'])
|
||
result = await openai_translate(text, data_setting["translate_target_lang"])
|
||
elif current_setting == "other_ai_plate":
|
||
aiInfoData = get_ai_info_setting()
|
||
if aiInfoData.get("base_url") == "https://api.siliconflow.cn/v1":
|
||
# result = await translateObject(strObjectData, data_setting['translate_target_lang'])
|
||
result = await translateObject(
|
||
text, data_setting["translate_target_lang"]
|
||
)
|
||
else:
|
||
# 仍然走 translators（含 'translater' 与历史的 'network'）
|
||
result = api_service_translate(
|
||
text,
|
||
data_setting["translate_source_lang"],
|
||
data_setting["translate_target_lang"],
|
||
data_setting["translate_service"],
|
||
)
|
||
return web.json_response({"data": result})
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
# POST /translate/tran/input （输入框反向翻译：目标 -> 源）
|
||
@PromptServer.instance.routes.post(baseUrl + "translate/tran/input")
|
||
async def _tanslater_input_text(request):
|
||
data = await request.json()
|
||
try:
|
||
current_setting = get_translate_setting()
|
||
data_setting = get_translate_settings()
|
||
text = data["text"]
|
||
data["str_object"]
|
||
if current_setting == "openai":
|
||
# 反向时把“源语种代码”当成目标，复用同一 prompt 模板
|
||
# result = await openai_translate(strObjectData, data_setting['translate_source_lang'])
|
||
result = await openai_translate(text, data_setting["translate_source_lang"])
|
||
elif current_setting == "other_ai_plate":
|
||
aiInfoData = get_ai_info_setting()
|
||
if aiInfoData.get("base_url") == "https://api.siliconflow.cn/v1":
|
||
# result = await translateObject(strObjectData, data_setting['translate_source_lang'])
|
||
result = await translateObject(
|
||
text, data_setting["translate_source_lang"]
|
||
)
|
||
else:
|
||
result = api_service_translate(
|
||
text,
|
||
data_setting["translate_target_lang"],
|
||
data_setting["translate_source_lang"],
|
||
data_setting["translate_service"],
|
||
)
|
||
return web.json_response({"data": result})
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
# =====================================================================================================
|
||
|
||
# =================================================== 云仓库获取 =======================================
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "cloud/get/main")
|
||
async def _cloud_get_main(request):
|
||
try:
|
||
return web.json_response({"data": get_main_warehouse()})
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "cloud/get/tree")
|
||
async def _cloud_get_tree(request):
|
||
data = await request.json()
|
||
try:
|
||
return web.json_response({"data": get_warehouse_tree(data["path"])})
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "cloud/download/file")
|
||
async def _cloud_get_download_file(request):
|
||
data = await request.json()
|
||
try:
|
||
return web.json_response(
|
||
{"data": install_cloud_file_db(data["path"], data["paths"])}
|
||
)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "cloud/get/local/package")
|
||
async def _cloud_get_local_package(request):
|
||
try:
|
||
return web.json_response({"data": get_package_paths()})
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
# =====================================================================================================
|
||
|
||
# =================================================== 随机Tag模板 =======================================
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "random_template/get_template_list")
|
||
async def _get_template_list(request):
|
||
try:
|
||
return web.json_response(get_template_list())
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "random_template/save_template")
|
||
async def _save_template(request):
|
||
data = await request.json()
|
||
try:
|
||
return web.json_response(save_template(data["data"]))
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "random_template/update_template")
|
||
async def _update_template(request):
|
||
data = await request.json()
|
||
try:
|
||
return web.json_response(update_template(data["name"], data["data"]))
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "random_template/delete_template")
|
||
async def _delete_template(request):
|
||
data = await request.json()
|
||
try:
|
||
update_random_template_setting("")
|
||
return web.json_response(delete_template(data["name"]))
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "random_template/get_template_data")
|
||
async def _get_template_data(request):
|
||
data = await request.json()
|
||
try:
|
||
return web.json_response(get_template_data(data["name"]))
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "get/setting/get_random_template_setting")
|
||
async def _get_random_template_setting(request):
|
||
try:
|
||
setting = get_random_template_setting() # 获取设置的参数信息
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"data": setting})
|
||
|
||
|
||
@PromptServer.instance.routes.post(
|
||
baseUrl + "update/setting/update_random_template_setting"
|
||
)
|
||
async def _update_random_template_setting(request):
|
||
data = await request.json()
|
||
try:
|
||
update_random_template_setting(data.get("path")) # 获取设置的参数信息
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "random_template/go_random_template")
|
||
async def _go_random_template(request):
|
||
try:
|
||
setting = get_random_template_setting() # 获取设置的参数信息
|
||
if setting == "":
|
||
return web.json_response({"code": 300, "info": "请先应用一个模板"})
|
||
return web.json_response(go_radom_template(setting))
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "random_template/go_random_template_path")
|
||
async def _go_random_template_path(request):
|
||
data = await request.json()
|
||
try:
|
||
if data.get("name") == "":
|
||
return web.json_response({"code": 300, "info": "请先应用一个模板"})
|
||
return web.json_response(go_radom_template(data.get("name")))
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
# =====================================================================================================
|
||
|
||
# ================================= Danbooru标签管理 =================================
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "danbooru/get_tags")
|
||
async def _get_danbooru_tags(request):
|
||
data = await request.json()
|
||
try:
|
||
tags = await get_danbooru_tags(
|
||
data.get("search"), data.get("page", 1), data.get("limit", 100)
|
||
)
|
||
return web.json_response({"data": tags})
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "danbooru/add_tag")
|
||
async def _add_danbooru_tag(request):
|
||
data = await request.json()
|
||
try:
|
||
await add_danbooru_tag(data)
|
||
return web.json_response({"code": 200})
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "danbooru/update_tag")
|
||
async def _update_danbooru_tag(request):
|
||
data = await request.json()
|
||
try:
|
||
await update_danbooru_tag(data["id"], data["update_data"])
|
||
return web.json_response({"code": 200})
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "danbooru/delete_tag")
|
||
async def _delete_danbooru_tag(request):
|
||
data = await request.json()
|
||
try:
|
||
await delete_danbooru_tag(data["id"])
|
||
return web.json_response({"code": 200})
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "danbooru/batch_delete_tag")
|
||
async def _batch_delete_danbooru_tag(request):
|
||
data = await request.json()
|
||
try:
|
||
await delete_danbooru_tags_batch(data["values"])
|
||
return web.json_response({"code": 200})
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "danbooru/get_tag_by_id")
|
||
async def _get_danbooru_tag_by_id(request):
|
||
data = await request.json()
|
||
try:
|
||
tag = await get_danbooru_tag_by_id(data["id"])
|
||
return web.json_response({"data": tag})
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "danbooru/run_sql_text")
|
||
async def _run_danbooru_sql_text(request):
|
||
data = await request.json()
|
||
try:
|
||
result = run_danbooru_sql_text(data["sql"])
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
|
||
return web.json_response(result)
|
||
|
||
|
||
# ===================================================================================
|
||
|
||
# ============================================ AI平台对接 ============================================
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "ai_server/get_settings")
|
||
async def _get_ai_server_setting(request):
|
||
return web.json_response({"data": get_ai_info_setting()})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "ai_server/update_settings")
|
||
async def _update_ai_server_settings(request):
|
||
data = await request.json()
|
||
try:
|
||
# 如果前端传递 {"ai_info_setting": {...}}，则取内部 dict
|
||
if "ai_info_setting" in data:
|
||
update_ai_info_setting(data["ai_info_setting"])
|
||
else:
|
||
update_ai_info_setting(data)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return web.Response(status=500)
|
||
return web.json_response({"info": "ok"})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "ai_server/get_ai_models")
|
||
def _get_ai_server_get_ai_models(request):
|
||
return web.json_response({"data": getModelList()})
|
||
|
||
|
||
# ============================================ AI平台对接 End ============================================
|
||
|
||
|
||
# ============================================ 标签图片管理 ============================================
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "tag_image/options")
|
||
async def _get_tag_image_options(request):
|
||
return web.json_response({"data": get_generation_options()})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "tag_image/generate")
|
||
async def _generate_tag_image(request):
|
||
data = await request.json()
|
||
t_uuid = data.get("t_uuid")
|
||
params = data.get("params", {})
|
||
regenerate = data.get("regenerate", False)
|
||
|
||
if not t_uuid:
|
||
return web.json_response({"error": "t_uuid is required"}, status=400)
|
||
|
||
# Validate required params
|
||
required = ["checkpoint", "width", "height", "sampler_name", "scheduler", "steps", "cfg", "seed", "positive", "negative"]
|
||
for key in required:
|
||
if key not in params:
|
||
return web.json_response({"error": f"Missing required param: {key}"}, status=400)
|
||
|
||
try:
|
||
if regenerate:
|
||
await delete_tag_image(t_uuid)
|
||
task_id = await enqueue_generation(t_uuid, params)
|
||
return web.json_response({"data": {"task_id": task_id}})
|
||
except RuntimeError as e:
|
||
return web.json_response({"error": str(e)}, status=409)
|
||
except Exception as e:
|
||
print(f"Error enqueuing generation: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "tag_image/batch_generate")
|
||
async def _batch_generate_tag_images(request):
|
||
data = await request.json()
|
||
tags = data.get("tags", [])
|
||
params = data.get("params", {})
|
||
|
||
if not tags or not isinstance(tags, list):
|
||
return web.json_response({"error": "tags array is required"}, status=400)
|
||
|
||
# Validate default params
|
||
required = ["checkpoint", "width", "height", "sampler_name", "scheduler", "steps", "cfg", "seed", "positive_template", "negative"]
|
||
for key in required:
|
||
if key not in params:
|
||
return web.json_response({"error": f"Missing required param: {key}"}, status=400)
|
||
|
||
task_ids = []
|
||
for tag in tags:
|
||
t_uuid = tag.get("t_uuid")
|
||
tag_text = tag.get("text", "")
|
||
if not t_uuid:
|
||
continue
|
||
try:
|
||
# Build per-tag params: replace {text} in positive_template with tag text
|
||
positive = params["positive_template"].replace("{text}", tag_text)
|
||
tag_params = {
|
||
"checkpoint": params["checkpoint"],
|
||
"width": params["width"],
|
||
"height": params["height"],
|
||
"sampler_name": params["sampler_name"],
|
||
"scheduler": params.get("scheduler", "normal"),
|
||
"steps": params["steps"],
|
||
"cfg": params["cfg"],
|
||
"seed": params["seed"],
|
||
"positive": positive,
|
||
"negative": params["negative"],
|
||
}
|
||
task_id = await enqueue_generation(t_uuid, tag_params)
|
||
task_ids.append({"t_uuid": t_uuid, "task_id": task_id})
|
||
except RuntimeError:
|
||
# Skip tags that already have a generation in progress
|
||
pass
|
||
|
||
return web.json_response({"data": {"task_ids": task_ids}})
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "tag_image/cancel_all")
|
||
async def _cancel_all_tag_image_tasks(request):
|
||
try:
|
||
count = await cancel_all_tasks()
|
||
return web.json_response({"data": {"cancelled": count}})
|
||
except Exception as e:
|
||
print(f"Error cancelling tasks: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "tag_image/status")
|
||
async def _get_tag_image_status_by_task(request):
|
||
data = await request.json()
|
||
task_id = data.get("task_id")
|
||
|
||
if not task_id:
|
||
return web.json_response({"error": "task_id is required"}, status=400)
|
||
|
||
task = get_task_status(task_id)
|
||
if not task:
|
||
return web.json_response({"error": "task not found"}, status=404)
|
||
|
||
return web.json_response({"data": {
|
||
"task_id": task["task_id"],
|
||
"status": task["status"],
|
||
"error_msg": task.get("error_msg"),
|
||
"image_path": task.get("image_path"),
|
||
}})
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "tag_image/status/{t_uuid}")
|
||
async def _get_tag_image_status_by_uuid(request):
|
||
t_uuid = request.match_info.get("t_uuid")
|
||
|
||
if not t_uuid:
|
||
return web.json_response({"error": "t_uuid is required"}, status=400)
|
||
|
||
# Check DB first
|
||
db_info = await get_tag_image_info(t_uuid)
|
||
status = db_info["image_status"] if db_info else None
|
||
image_path = db_info["image_path"] if db_info else None
|
||
|
||
# Check if there's a running task
|
||
task = get_task_status_by_tuuid(t_uuid)
|
||
task_id = task["task_id"] if task else None
|
||
if task and task["status"] in ("pending", "generating"):
|
||
status = task["status"]
|
||
|
||
return web.json_response({"data": {
|
||
"status": status,
|
||
"task_id": task_id,
|
||
"image_url": f"/weilin/prompt_ui/api/tag_image/{t_uuid}" if status == "ready" else None,
|
||
"thumb_url": f"/weilin/prompt_ui/api/tag_thumb/{t_uuid}" if status == "ready" else None,
|
||
"error_msg": task.get("error_msg") if task else None,
|
||
}})
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "tag_image/{t_uuid}")
|
||
async def _serve_tag_image(request):
|
||
t_uuid = request.match_info.get("t_uuid")
|
||
img_dir = image_dir()
|
||
# Try common extensions
|
||
for ext in (".png", ".jpg", ".jpeg", ".webp"):
|
||
path = os.path.join(img_dir, f"{t_uuid}{ext}")
|
||
if os.path.isfile(path):
|
||
return web.FileResponse(path)
|
||
return web.Response(status=404, text="Image not found")
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "tag_thumb/{t_uuid}")
|
||
async def _serve_tag_thumb(request):
|
||
t_uuid = request.match_info.get("t_uuid")
|
||
t_dir = thumb_dir()
|
||
path = os.path.join(t_dir, f"{t_uuid}.webp")
|
||
if os.path.isfile(path):
|
||
return web.FileResponse(path)
|
||
return web.Response(status=404, text="Thumbnail not found")
|
||
|
||
|
||
# ============================================ 标签图片管理 End ============================================
|
||
|
||
|
||
# ============================================ 标签数据管理 ============================================
|
||
|
||
|
||
@PromptServer.instance.routes.get(baseUrl + "labels/get")
|
||
async def _get_tag_labels(request):
|
||
"""
|
||
获取标签数据
|
||
GET /weilin/prompt_ui/api/labels/get
|
||
"""
|
||
try:
|
||
# 获取项目根目录下的 tag_labels.json
|
||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||
json_path = os.path.join(current_dir, "../../tag_labels.json")
|
||
|
||
print(f"[WeiLin] 读取标签数据从: {json_path}")
|
||
|
||
# 如果文件不存在，返回默认空数据
|
||
if not os.path.exists(json_path):
|
||
print("[WeiLin] 文件不存在，返回默认数据")
|
||
default_data = {
|
||
"items": [],
|
||
"settings": {
|
||
"sortMode": "time",
|
||
"sortTimeDesc": True,
|
||
"sortNameAsc": True,
|
||
"selectedId": None,
|
||
},
|
||
}
|
||
return web.json_response({"code": 200, "data": default_data})
|
||
|
||
# 读取 JSON 文件
|
||
with open(json_path, encoding="utf-8") as f:
|
||
data = json.load(f)
|
||
|
||
print(f"[WeiLin] 读取到 {len(data.get('items', []))} 个标签")
|
||
|
||
return web.json_response({"code": 200, "data": data})
|
||
except Exception as e:
|
||
print(f"[WeiLin] Error getting tag labels: {e}")
|
||
import traceback
|
||
|
||
traceback.print_exc()
|
||
return web.Response(status=500, text=str(e))
|
||
|
||
|
||
@PromptServer.instance.routes.post(baseUrl + "labels/save")
|
||
async def _save_tag_labels(request):
|
||
"""
|
||
保存标签数据
|
||
POST /weilin/prompt_ui/api/labels/save
|
||
Body: {
|
||
"items": [...],
|
||
"settings": {...}
|
||
}
|
||
"""
|
||
try:
|
||
data = await request.json()
|
||
|
||
# 获取项目根目录下的 tag_labels.json
|
||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||
json_path = os.path.join(current_dir, "../../tag_labels.json")
|
||
|
||
print(f"[WeiLin] 保存标签数据到: {json_path}")
|
||
print(f"[WeiLin] 标签数量: {len(data.get('items', []))}")
|
||
|
||
# 写入 JSON 文件
|
||
with open(json_path, "w", encoding="utf-8") as f:
|
||
json.dump(data, f, ensure_ascii=False, indent=2)
|
||
|
||
print("[WeiLin] 标签数据保存成功")
|
||
|
||
return web.json_response(
|
||
{"code": 200, "message": "保存成功", "file_path": json_path}
|
||
)
|
||
except Exception as e:
|
||
print(f"[WeiLin] Error saving tag labels: {e}")
|
||
import traceback
|
||
|
||
traceback.print_exc()
|
||
return web.Response(status=500, text=str(e))
|
||
|
||
|
||
# ============================================ 标签数据管理 End ============================================
|
||
|
||
print("======== WeiLin插件服务已启动 ========")
|
||
print("======== WeiLin Server Init ========")
|
||
|
||
|
||
def go_run_node_auto_random_tag(name):
|
||
try:
|
||
return go_radom_template(name)
|
||
except Exception as e:
|
||
print(f"Error: {e}")
|
||
return ""
|
||
|
||
# Start the tag image generation worker
|
||
import asyncio as _asyncio
|
||
try:
|
||
_loop = _asyncio.get_event_loop()
|
||
if _loop.is_running():
|
||
_asyncio.ensure_future(start_worker())
|
||
else:
|
||
_loop.run_until_complete(start_worker())
|
||
except Exception:
|
||
pass
|