Fixed RPG
This commit is contained in:
+6
-1
@@ -103,6 +103,8 @@ async def _migrate_personas_columns(db):
|
|||||||
await db.execute("ALTER TABLE personas ADD COLUMN lorebook_json TEXT DEFAULT '[]'")
|
await db.execute("ALTER TABLE personas ADD COLUMN lorebook_json TEXT DEFAULT '[]'")
|
||||||
if "avatar_path" not in cols:
|
if "avatar_path" not in cols:
|
||||||
await db.execute("ALTER TABLE personas ADD COLUMN avatar_path TEXT DEFAULT ''")
|
await db.execute("ALTER TABLE personas ADD COLUMN avatar_path TEXT DEFAULT ''")
|
||||||
|
if "alternate_greetings_json" not in cols:
|
||||||
|
await db.execute("ALTER TABLE personas ADD COLUMN alternate_greetings_json TEXT DEFAULT '[]'")
|
||||||
|
|
||||||
|
|
||||||
async def _migrate_sessions_columns(db):
|
async def _migrate_sessions_columns(db):
|
||||||
@@ -126,7 +128,8 @@ async def _migrate_sessions_columns(db):
|
|||||||
await db.execute("ALTER TABLE sessions ADD COLUMN genre TEXT DEFAULT 'adventure'")
|
await db.execute("ALTER TABLE sessions ADD COLUMN genre TEXT DEFAULT 'adventure'")
|
||||||
if "rpg_settings_json" not in cols:
|
if "rpg_settings_json" not in cols:
|
||||||
await db.execute("ALTER TABLE sessions ADD COLUMN rpg_settings_json TEXT DEFAULT '{}'")
|
await db.execute("ALTER TABLE sessions ADD COLUMN rpg_settings_json TEXT DEFAULT '{}'")
|
||||||
|
if "outfit_json" not in cols:
|
||||||
|
await db.execute("ALTER TABLE sessions ADD COLUMN outfit_json TEXT DEFAULT '[]'")
|
||||||
|
|
||||||
async def _migrate_rpg_quests(db):
|
async def _migrate_rpg_quests(db):
|
||||||
await db.executescript("""
|
await db.executescript("""
|
||||||
@@ -165,3 +168,5 @@ async def _migrate_characters_columns(db):
|
|||||||
cols = {row[1] for row in await cur.fetchall()}
|
cols = {row[1] for row in await cur.fetchall()}
|
||||||
if "avatar_path" not in cols:
|
if "avatar_path" not in cols:
|
||||||
await db.execute("ALTER TABLE characters ADD COLUMN avatar_path TEXT DEFAULT ''")
|
await db.execute("ALTER TABLE characters ADD COLUMN avatar_path TEXT DEFAULT ''")
|
||||||
|
if "alternate_greetings_json" not in cols:
|
||||||
|
await db.execute("ALTER TABLE characters ADD COLUMN alternate_greetings_json TEXT DEFAULT '[]'")
|
||||||
|
|||||||
@@ -0,0 +1,6 @@
|
|||||||
|
import sqlite3
|
||||||
|
db = sqlite3.connect("data/chat.db")
|
||||||
|
cur = db.execute("DELETE FROM messages WHERE role='assistant' AND trim(coalesce(content,''))=''")
|
||||||
|
print("Deleted:", cur.rowcount, "empty assistant messages")
|
||||||
|
db.commit()
|
||||||
|
db.close()
|
||||||
@@ -7,6 +7,7 @@ class ChatRequest(BaseModel):
|
|||||||
persona_id: Optional[str] = "default"
|
persona_id: Optional[str] = "default"
|
||||||
is_narrator_choice: bool = False
|
is_narrator_choice: bool = False
|
||||||
skip_user_add: bool = False
|
skip_user_add: bool = False
|
||||||
|
first_mes_override: Optional[str] = None
|
||||||
|
|
||||||
|
|
||||||
class MessageEditRequest(BaseModel):
|
class MessageEditRequest(BaseModel):
|
||||||
|
|||||||
+48
-5
@@ -2,7 +2,14 @@ from fastapi import APIRouter, File, Form, HTTPException, UploadFile
|
|||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
from typing import Optional
|
from typing import Optional
|
||||||
|
|
||||||
from services.character_card import list_characters, get_character, import_card_file, update_character, update_appearance_tags
|
from services.character_card import (
|
||||||
|
list_characters,
|
||||||
|
get_character,
|
||||||
|
import_card_file,
|
||||||
|
preview_card_file,
|
||||||
|
update_character,
|
||||||
|
update_appearance_tags,
|
||||||
|
)
|
||||||
|
|
||||||
router = APIRouter(prefix="/characters", tags=["characters"])
|
router = APIRouter(prefix="/characters", tags=["characters"])
|
||||||
|
|
||||||
@@ -17,6 +24,7 @@ class CardPatch(BaseModel):
|
|||||||
appearance_tags: Optional[str] = None
|
appearance_tags: Optional[str] = None
|
||||||
lora_name: Optional[str] = None
|
lora_name: Optional[str] = None
|
||||||
lora_weight: Optional[float] = None
|
lora_weight: Optional[float] = None
|
||||||
|
alternate_greetings_json: Optional[str] = None
|
||||||
|
|
||||||
|
|
||||||
@router.get("/")
|
@router.get("/")
|
||||||
@@ -32,6 +40,15 @@ async def get_one(card_id: str):
|
|||||||
return card
|
return card
|
||||||
|
|
||||||
|
|
||||||
|
@router.post("/preview")
|
||||||
|
async def preview_card(file: UploadFile = File(...)):
|
||||||
|
content = await file.read()
|
||||||
|
try:
|
||||||
|
return await preview_card_file(content, file.filename or "card.json")
|
||||||
|
except Exception as e:
|
||||||
|
raise HTTPException(status_code=400, detail=str(e))
|
||||||
|
|
||||||
|
|
||||||
@router.patch("/{card_id}")
|
@router.patch("/{card_id}")
|
||||||
async def patch_card(card_id: str, body: CardPatch):
|
async def patch_card(card_id: str, body: CardPatch):
|
||||||
card = await get_character(card_id)
|
card = await get_character(card_id)
|
||||||
@@ -39,20 +56,26 @@ async def patch_card(card_id: str, body: CardPatch):
|
|||||||
raise HTTPException(status_code=404, detail="Карточка не найдена")
|
raise HTTPException(status_code=404, detail="Карточка не найдена")
|
||||||
fields = {k: v for k, v in body.model_dump().items() if v is not None}
|
fields = {k: v for k, v in body.model_dump().items() if v is not None}
|
||||||
await update_character(card_id, fields)
|
await update_character(card_id, fields)
|
||||||
# sync appearance_tags and lora to persona
|
|
||||||
from services.personas import update_persona_appearance
|
from services.personas import update_persona_appearance
|
||||||
if "appearance_tags" in fields:
|
if "appearance_tags" in fields:
|
||||||
await update_persona_appearance(f"card_{card_id}", fields["appearance_tags"])
|
await update_persona_appearance(f"card_{card_id}", fields["appearance_tags"])
|
||||||
if {"lora_name", "lora_weight"} & fields.keys():
|
if {"lora_name", "lora_weight"} & fields.keys():
|
||||||
from services.personas import update_persona_lora
|
from services.personas import update_persona_lora
|
||||||
await update_persona_lora(f"card_{card_id}", fields.get("lora_name"), fields.get("lora_weight"))
|
await update_persona_lora(f"card_{card_id}", fields.get("lora_name"), fields.get("lora_weight"))
|
||||||
# rebuild system prompt if character fields changed
|
|
||||||
char_fields = {"name", "description", "personality", "scenario", "first_mes", "mes_example"}
|
char_fields = {"name", "description", "personality", "scenario", "first_mes", "mes_example"}
|
||||||
if char_fields & fields.keys():
|
if char_fields & fields.keys():
|
||||||
updated = await get_character(card_id)
|
updated = await get_character(card_id)
|
||||||
from services.character_card import build_system_prompt
|
from services.character_card import build_system_prompt
|
||||||
from services.personas import update_persona_prompt
|
from services.personas import update_persona_prompt
|
||||||
await update_persona_prompt(f"card_{card_id}", build_system_prompt(updated))
|
await update_persona_prompt(f"card_{card_id}", build_system_prompt(updated))
|
||||||
|
if "first_mes" in fields or "alternate_greetings_json" in fields:
|
||||||
|
from services.personas import patch_persona
|
||||||
|
sync = {}
|
||||||
|
if "first_mes" in fields:
|
||||||
|
sync["first_mes"] = fields["first_mes"]
|
||||||
|
if "alternate_greetings_json" in fields:
|
||||||
|
sync["alternate_greetings_json"] = fields["alternate_greetings_json"]
|
||||||
|
await patch_persona(f"card_{card_id}", sync)
|
||||||
return await get_character(card_id)
|
return await get_character(card_id)
|
||||||
|
|
||||||
|
|
||||||
@@ -67,7 +90,6 @@ async def upload_avatar(card_id: str, file: UploadFile = File(...)):
|
|||||||
from services.character_card import _save_avatar_bytes
|
from services.character_card import _save_avatar_bytes
|
||||||
rel = _save_avatar_bytes(content, f"card_{card_id}")
|
rel = _save_avatar_bytes(content, f"card_{card_id}")
|
||||||
await update_character(card_id, {"avatar_path": rel})
|
await update_character(card_id, {"avatar_path": rel})
|
||||||
# sync persona
|
|
||||||
from services.personas import patch_persona
|
from services.personas import patch_persona
|
||||||
await patch_persona(f"card_{card_id}", {"avatar_path": rel})
|
await patch_persona(f"card_{card_id}", {"avatar_path": rel})
|
||||||
return {"avatar_path": f"/static/{rel}"}
|
return {"avatar_path": f"/static/{rel}"}
|
||||||
@@ -78,14 +100,35 @@ async def import_card(
|
|||||||
file: UploadFile = File(...),
|
file: UploadFile = File(...),
|
||||||
lora_name: str = Form(""),
|
lora_name: str = Form(""),
|
||||||
lora_weight: float = Form(0.8),
|
lora_weight: float = Form(0.8),
|
||||||
|
card_id: str = Form(""),
|
||||||
|
name: str = Form(""),
|
||||||
|
description: str = Form(""),
|
||||||
|
personality: str = Form(""),
|
||||||
|
scenario: str = Form(""),
|
||||||
|
first_mes: str = Form(""),
|
||||||
|
mes_example: str = Form(""),
|
||||||
|
appearance_tags: str = Form(""),
|
||||||
|
alternate_greetings_json: str = Form("[]"),
|
||||||
):
|
):
|
||||||
content = await file.read()
|
content = await file.read()
|
||||||
|
overrides = {
|
||||||
|
"name": name or None,
|
||||||
|
"description": description or None,
|
||||||
|
"personality": personality or None,
|
||||||
|
"scenario": scenario or None,
|
||||||
|
"first_mes": first_mes or None,
|
||||||
|
"mes_example": mes_example or None,
|
||||||
|
"appearance_tags": appearance_tags or None,
|
||||||
|
"alternate_greetings_json": alternate_greetings_json or "[]",
|
||||||
|
}
|
||||||
try:
|
try:
|
||||||
card = await import_card_file(
|
card = await import_card_file(
|
||||||
content,
|
content,
|
||||||
file.filename or "card.json",
|
file.filename or "card.json",
|
||||||
lora_name=lora_name,
|
lora_name=lora_name,
|
||||||
lora_weight=lora_weight,
|
lora_weight=lora_weight,
|
||||||
|
overrides=overrides,
|
||||||
|
card_id=card_id.strip() or None,
|
||||||
)
|
)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise HTTPException(status_code=400, detail=str(e))
|
raise HTTPException(status_code=400, detail=str(e))
|
||||||
@@ -94,6 +137,7 @@ async def import_card(
|
|||||||
"card_id": card["card_id"],
|
"card_id": card["card_id"],
|
||||||
"persona_id": f"card_{card['card_id']}",
|
"persona_id": f"card_{card['card_id']}",
|
||||||
"name": card["name"],
|
"name": card["name"],
|
||||||
|
"alternate_greetings": card.get("alternate_greetings", []),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -104,4 +148,3 @@ async def remove_card(card_id: str):
|
|||||||
if not await delete_persona(f"card_{card_id}"):
|
if not await delete_persona(f"card_{card_id}"):
|
||||||
raise HTTPException(status_code=404, detail="Карточка не найдена")
|
raise HTTPException(status_code=404, detail="Карточка не найдена")
|
||||||
return {"status": "deleted", "card_id": card_id}
|
return {"status": "deleted", "card_id": card_id}
|
||||||
|
|
||||||
|
|||||||
+78
-100
@@ -1,4 +1,5 @@
|
|||||||
import json
|
import json
|
||||||
|
import logging
|
||||||
import os
|
import os
|
||||||
import random
|
import random
|
||||||
|
|
||||||
@@ -26,6 +27,8 @@ from services.memory import (
|
|||||||
update_session_affinity,
|
update_session_affinity,
|
||||||
update_session_genre,
|
update_session_genre,
|
||||||
update_session_rpg_settings,
|
update_session_rpg_settings,
|
||||||
|
update_session_outfit,
|
||||||
|
update_session_plot_arc,
|
||||||
upsert_quest,
|
upsert_quest,
|
||||||
get_quests,
|
get_quests,
|
||||||
add_action_resolution,
|
add_action_resolution,
|
||||||
@@ -35,84 +38,57 @@ from services.memory import (
|
|||||||
delete_message,
|
delete_message,
|
||||||
)
|
)
|
||||||
from services.personas import get_persona
|
from services.personas import get_persona
|
||||||
from services.sd_prompt import (
|
from services.sd_prompt import generate_sd_prompt, strip_image_prompt_tag, extract_image_prompt_tag
|
||||||
generate_sd_prompt,
|
|
||||||
strip_image_prompt_tag,
|
|
||||||
extract_image_prompt_tag,
|
|
||||||
)
|
|
||||||
from services.lorebook import get_lorebook_context
|
from services.lorebook import get_lorebook_context
|
||||||
from services.character_card import get_character
|
from services.character_card import get_character
|
||||||
from services import sdbackend as sd_service
|
from services import sdbackend as sd_service
|
||||||
from services.rpg_facts import extract_facts, merge_facts, facts_to_prompt
|
from services.rpg_facts import extract_facts, merge_facts, facts_to_prompt
|
||||||
from services.rpg_plot import generate_plot_arc, should_advance_arc, pop_matching_beats
|
from services.rpg_plot import generate_plot_arc, should_advance_arc, pop_matching_beats, advance_phase
|
||||||
from services.rpg_narrator import narrator_pre, narrator_post
|
from services.rpg_narrator import narrator_pre, narrator_post
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
router = APIRouter(prefix="/chat", tags=["chat"])
|
router = APIRouter(prefix="/chat", tags=["chat"])
|
||||||
|
|
||||||
DEFAULT_PROMPT = "Ты — полезный AI ассистент. Отвечай чётко и по делу."
|
DEFAULT_PROMPT = "Ты — полезный AI ассистент. Отвечай чётко и по делу."
|
||||||
SD_AUTO_GENERATE = os.getenv("SD_AUTO_GENERATE", "false").lower() in ("1", "true", "yes")
|
SD_AUTO_GENERATE = os.getenv("SD_AUTO_GENERATE", "false").lower() in ("1", "true", "yes")
|
||||||
|
|
||||||
def affinity_prompt_block(affinity: int) -> str:
|
DEFAULT_RPG_SETTINGS = {"dice": True, "narrator": True, "quests": True, "affinity": True, "choices": True}
|
||||||
if affinity >= 10:
|
|
||||||
tone = "very warm, trusting, affectionate"
|
|
||||||
elif affinity >= 5:
|
|
||||||
tone = "friendly and open"
|
|
||||||
elif affinity >= 1:
|
|
||||||
tone = "slightly positive"
|
|
||||||
elif affinity <= -5:
|
|
||||||
tone = "hostile or deeply distrustful"
|
|
||||||
elif affinity <= -1:
|
|
||||||
tone = "cold and wary"
|
|
||||||
else:
|
|
||||||
tone = "neutral"
|
|
||||||
return (
|
|
||||||
f"\n\n--- Relationship ---\n"
|
|
||||||
f"Affinity toward player: {affinity} ({tone}). "
|
|
||||||
f"Reflect this in your attitude and word choice.\n---"
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
DEFAULT_RPG_SETTINGS = {
|
|
||||||
"dice": True,
|
|
||||||
"narrator": True,
|
|
||||||
"quests": True,
|
|
||||||
"affinity": True,
|
|
||||||
"choices": True,
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
def get_rpg_settings(session: dict) -> dict:
|
def get_rpg_settings(session: dict) -> dict:
|
||||||
try:
|
try:
|
||||||
s = json.loads(session.get("rpg_settings_json") or "{}")
|
return {**DEFAULT_RPG_SETTINGS, **json.loads(session.get("rpg_settings_json") or "{}")}
|
||||||
return {**DEFAULT_RPG_SETTINGS, **s}
|
|
||||||
except Exception:
|
except Exception:
|
||||||
return DEFAULT_RPG_SETTINGS
|
return DEFAULT_RPG_SETTINGS
|
||||||
|
|
||||||
|
|
||||||
|
def affinity_prompt_block(affinity: int) -> str:
|
||||||
|
if affinity >= 10: tone = "very warm, trusting, affectionate"
|
||||||
|
elif affinity >= 5: tone = "friendly and open"
|
||||||
|
elif affinity >= 1: tone = "slightly positive"
|
||||||
|
elif affinity <= -5: tone = "hostile or deeply distrustful"
|
||||||
|
elif affinity <= -1: tone = "cold and wary"
|
||||||
|
else: tone = "neutral"
|
||||||
|
return f"\n\n--- Relationship ---\nAffinity toward player: {affinity} ({tone}). Reflect this in your attitude and word choice.\n---"
|
||||||
|
|
||||||
|
|
||||||
async def get_system_prompt(persona_id: str, history: list, user_message: str = "") -> str:
|
async def get_system_prompt(persona_id: str, history: list, user_message: str = "") -> str:
|
||||||
persona = await get_persona(persona_id)
|
persona = await get_persona(persona_id)
|
||||||
if not persona:
|
if not persona:
|
||||||
return DEFAULT_PROMPT
|
return DEFAULT_PROMPT
|
||||||
|
|
||||||
prompt = persona["prompt"]
|
prompt = persona["prompt"]
|
||||||
|
|
||||||
if persona.get("lorebook_json"):
|
|
||||||
recent = [m for m in history if m["role"] in ("user", "assistant")][-5:]
|
recent = [m for m in history if m["role"] in ("user", "assistant")][-5:]
|
||||||
context = recent + [{"role": "user", "content": user_message}]
|
context = recent + [{"role": "user", "content": user_message}]
|
||||||
|
if persona.get("lorebook_json"):
|
||||||
lore = get_lorebook_context(persona.get("lorebook_json", "[]"), context)
|
lore = get_lorebook_context(persona.get("lorebook_json", "[]"), context)
|
||||||
if lore:
|
if lore:
|
||||||
prompt = prompt + "\n\n" + lore
|
prompt += "\n\n" + lore
|
||||||
|
|
||||||
if persona_id.startswith("card_"):
|
if persona_id.startswith("card_"):
|
||||||
card_id = persona_id[5:]
|
card = await get_character(persona_id[5:])
|
||||||
card = await get_character(card_id)
|
|
||||||
if card:
|
if card:
|
||||||
recent = [m for m in history if m["role"] in ("user", "assistant")][-5:]
|
|
||||||
context = recent + [{"role": "user", "content": user_message}]
|
|
||||||
lore = get_lorebook_context(card.get("lorebook_json", "[]"), context)
|
lore = get_lorebook_context(card.get("lorebook_json", "[]"), context)
|
||||||
if lore:
|
if lore:
|
||||||
prompt = prompt + "\n\n" + lore
|
prompt += "\n\n" + lore
|
||||||
|
|
||||||
return prompt
|
return prompt
|
||||||
|
|
||||||
|
|
||||||
@@ -126,12 +102,18 @@ async def get_system_blob(session_id: str):
|
|||||||
history = await get_history(session_id)
|
history = await get_history(session_id)
|
||||||
system_msg = next((m for m in history if m.get("role") == "system"), None)
|
system_msg = next((m for m in history if m.get("role") == "system"), None)
|
||||||
session = await get_session(session_id)
|
session = await get_session(session_id)
|
||||||
|
quests = await get_quests(session_id)
|
||||||
return {
|
return {
|
||||||
"system_prompt": system_msg.get("content") if system_msg else "",
|
"system_prompt": system_msg.get("content") if system_msg else "",
|
||||||
"facts_json": session.get("facts_json") if session else "[]",
|
|
||||||
"status_quo": session.get("status_quo") if session else "",
|
"status_quo": session.get("status_quo") if session else "",
|
||||||
|
"facts_json": session.get("facts_json") if session else "[]",
|
||||||
"plot_arc_json": session.get("plot_arc_json") if session else "{}",
|
"plot_arc_json": session.get("plot_arc_json") if session else "{}",
|
||||||
|
"outfit_json": session.get("outfit_json") if session else "[]",
|
||||||
|
"affinity": session.get("affinity", 0) if session else 0,
|
||||||
|
"genre": session.get("genre", "") if session else "",
|
||||||
|
"rpg_settings_json": session.get("rpg_settings_json") if session else "{}",
|
||||||
"rpg_enabled": bool(session.get("rpg_enabled")) if session else False,
|
"rpg_enabled": bool(session.get("rpg_enabled")) if session else False,
|
||||||
|
"quests": quests,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -147,6 +129,10 @@ async def init_chat(request: ChatRequest):
|
|||||||
await add_message(request.session_id, "system", system_prompt)
|
await add_message(request.session_id, "system", system_prompt)
|
||||||
|
|
||||||
first_mes = None
|
first_mes = None
|
||||||
|
if request.first_mes_override and request.first_mes_override.strip():
|
||||||
|
first_mes = request.first_mes_override.strip()
|
||||||
|
await add_message(request.session_id, "assistant", first_mes)
|
||||||
|
else:
|
||||||
persona = await get_persona(persona_id)
|
persona = await get_persona(persona_id)
|
||||||
if persona and persona.get("first_mes"):
|
if persona and persona.get("first_mes"):
|
||||||
first_mes = persona["first_mes"]
|
first_mes = persona["first_mes"]
|
||||||
@@ -196,9 +182,9 @@ async def rpg_bootstrap(req: RpgBootstrapRequest):
|
|||||||
|
|
||||||
# Seed quests from beats
|
# Seed quests from beats
|
||||||
for beat in arc.get("beats", []):
|
for beat in arc.get("beats", []):
|
||||||
injection = beat.get("injection", "").strip()
|
title = (beat.get("title") or beat.get("injection", "")).strip()
|
||||||
if injection:
|
if title:
|
||||||
await upsert_quest(req.session_id, injection[:120])
|
await upsert_quest(req.session_id, title[:120])
|
||||||
|
|
||||||
quests = await get_quests(req.session_id)
|
quests = await get_quests(req.session_id)
|
||||||
return {"plot_arc": arc, "quests": quests}
|
return {"plot_arc": arc, "quests": quests}
|
||||||
@@ -339,11 +325,21 @@ async def chat_stream(request: ChatRequest):
|
|||||||
|
|
||||||
async def generate():
|
async def generate():
|
||||||
nonlocal arc
|
nonlocal arc
|
||||||
|
|
||||||
|
# Send narrator BEFORE streaming so it appears above the reply
|
||||||
|
if narrator_msg:
|
||||||
|
yield f"data: {json.dumps({'narrator': narrator_msg})}\n\n"
|
||||||
|
|
||||||
|
try:
|
||||||
async for chunk in stream_message(
|
async for chunk in stream_message(
|
||||||
[{"role": m["role"], "content": m["content"]} for m in messages]
|
[{"role": m["role"], "content": m["content"]} for m in messages]
|
||||||
):
|
):
|
||||||
full_reply.append(chunk)
|
full_reply.append(chunk)
|
||||||
yield f"data: {json.dumps({'chunk': chunk})}\n\n"
|
yield f"data: {json.dumps({'chunk': chunk})}\n\n"
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("stream_message failed: %s", e)
|
||||||
|
yield f"data: {json.dumps({'error': str(e)})}\n\n"
|
||||||
|
return
|
||||||
|
|
||||||
complete = "".join(full_reply)
|
complete = "".join(full_reply)
|
||||||
display_text = strip_image_prompt_tag(complete)
|
display_text = strip_image_prompt_tag(complete)
|
||||||
@@ -351,17 +347,14 @@ async def chat_stream(request: ChatRequest):
|
|||||||
hist_with_reply = await get_history(request.session_id) + [
|
hist_with_reply = await get_history(request.session_id) + [
|
||||||
{"role": "assistant", "content": display_text}
|
{"role": "assistant", "content": display_text}
|
||||||
]
|
]
|
||||||
sd_result = await generate_sd_prompt(hist_with_reply, persona_id)
|
sd_result = await generate_sd_prompt(
|
||||||
prompt_str = sd_result[0] if sd_result else None
|
hist_with_reply, persona_id,
|
||||||
if not prompt_str:
|
outfit_json=session.get("outfit_json", "[]") if session else "[]"
|
||||||
prompt_str = extract_image_prompt_tag(complete)
|
|
||||||
|
|
||||||
await add_message(
|
|
||||||
request.session_id,
|
|
||||||
"assistant",
|
|
||||||
display_text or complete,
|
|
||||||
image_prompt=prompt_str,
|
|
||||||
)
|
)
|
||||||
|
prompt_str = (sd_result[0] if sd_result and sd_result[0] else None) or extract_image_prompt_tag(complete)
|
||||||
|
|
||||||
|
if (display_text or complete).strip():
|
||||||
|
await add_message(request.session_id, "assistant", display_text or complete, image_prompt=prompt_str)
|
||||||
|
|
||||||
choices = []
|
choices = []
|
||||||
debug_blocks = []
|
debug_blocks = []
|
||||||
@@ -370,38 +363,36 @@ async def chat_stream(request: ChatRequest):
|
|||||||
if session and session.get("rpg_enabled"):
|
if session and session.get("rpg_enabled"):
|
||||||
if not arc:
|
if not arc:
|
||||||
persona = await get_persona(persona_id) or {}
|
persona = await get_persona(persona_id) or {}
|
||||||
genre = (session.get("genre") or "adventure")
|
|
||||||
arc = await generate_plot_arc(
|
arc = await generate_plot_arc(
|
||||||
persona.get("name", persona_id),
|
persona.get("name", persona_id),
|
||||||
persona.get("description", ""),
|
persona.get("description", ""),
|
||||||
persona.get("scenario", ""),
|
persona.get("scenario", ""),
|
||||||
persona.get("first_mes", ""),
|
persona.get("first_mes", ""),
|
||||||
facts_block=facts_to_prompt(session.get("facts_json", "[]")),
|
facts_block=facts_to_prompt(session.get("facts_json", "[]")),
|
||||||
genre=genre,
|
genre=session.get("genre") or "adventure",
|
||||||
)
|
)
|
||||||
if arc:
|
if arc:
|
||||||
from services.memory import update_session_plot_arc
|
|
||||||
await update_session_plot_arc(request.session_id, json.dumps(arc, ensure_ascii=False))
|
await update_session_plot_arc(request.session_id, json.dumps(arc, ensure_ascii=False))
|
||||||
debug_blocks.append({"type": "plot_arc", "text": json.dumps(arc, ensure_ascii=False, indent=2)})
|
debug_blocks.append({"type": "plot_arc", "text": json.dumps(arc, ensure_ascii=False, indent=2)})
|
||||||
if rpg_settings.get("quests", True):
|
if rpg_settings.get("quests", True):
|
||||||
for beat in arc.get("beats", []):
|
for beat in arc.get("beats", []):
|
||||||
inj = beat.get("injection", "").strip()
|
t = (beat.get("title") or beat.get("injection", "")).strip()
|
||||||
if inj:
|
if t:
|
||||||
await upsert_quest(request.session_id, inj[:120])
|
await upsert_quest(request.session_id, t[:120])
|
||||||
|
|
||||||
trig = should_advance_arc(request.message)
|
trig = should_advance_arc(request.message)
|
||||||
if trig and arc:
|
if trig and arc:
|
||||||
arc, beats = pop_matching_beats(arc, trig, max_beats=1)
|
arc, beats = pop_matching_beats(arc, trig, max_beats=1)
|
||||||
if beats:
|
if beats:
|
||||||
from services.memory import update_session_plot_arc
|
|
||||||
await update_session_plot_arc(request.session_id, json.dumps(arc, ensure_ascii=False))
|
await update_session_plot_arc(request.session_id, json.dumps(arc, ensure_ascii=False))
|
||||||
inj = beats[0].get("injection", "")
|
inj = beats[0].get("injection", "")
|
||||||
if inj:
|
if inj:
|
||||||
debug_blocks.append({"type": "narrator_injection", "text": inj})
|
debug_blocks.append({"type": "narrator_injection", "text": inj})
|
||||||
if rpg_settings.get("choices", True):
|
if rpg_settings.get("choices", True):
|
||||||
beat_choices = beats[0].get("choices") or []
|
choices += beats[0].get("choices") or []
|
||||||
if beat_choices:
|
if advance_phase(arc):
|
||||||
choices = choices + beat_choices
|
await update_session_plot_arc(request.session_id, json.dumps(arc, ensure_ascii=False))
|
||||||
|
debug_blocks.append({"type": "phase_advance", "text": arc["phase"]})
|
||||||
|
|
||||||
ctx = [m for m in (await get_history(request.session_id)) if m["role"] in ("user", "assistant")][-10:]
|
ctx = [m for m in (await get_history(request.session_id)) if m["role"] in ("user", "assistant")][-10:]
|
||||||
new_facts = await extract_facts(ctx)
|
new_facts = await extract_facts(ctx)
|
||||||
@@ -409,55 +400,53 @@ async def chat_stream(request: ChatRequest):
|
|||||||
merged = merge_facts(session.get("facts_json", "[]"), new_facts)
|
merged = merge_facts(session.get("facts_json", "[]"), new_facts)
|
||||||
await update_session_facts(request.session_id, merged)
|
await update_session_facts(request.session_id, merged)
|
||||||
session["facts_json"] = merged
|
session["facts_json"] = merged
|
||||||
debug_blocks.append({"type": "facts", "text": facts_to_prompt(merged)})
|
|
||||||
|
|
||||||
persona = await get_persona(persona_id) or {}
|
persona = await get_persona(persona_id) or {}
|
||||||
ctx_txt = "\n".join(
|
ctx_txt = "\n".join(f"{m['role']}: {m['content']}" for m in ctx[-8:] if m.get("role") in ("user", "assistant"))
|
||||||
f"{m['role']}: {m['content']}" for m in ctx[-8:]
|
|
||||||
if m.get("role") in ("user", "assistant")
|
|
||||||
)
|
|
||||||
post = await narrator_post(
|
post = await narrator_post(
|
||||||
persona.get("name", persona_id),
|
persona.get("name", persona_id),
|
||||||
ctx_txt,
|
ctx_txt,
|
||||||
json.dumps(arc, ensure_ascii=False) if arc else "",
|
json.dumps(arc, ensure_ascii=False) if arc else "",
|
||||||
facts_to_prompt(session.get("facts_json", "[]")),
|
facts_to_prompt(session.get("facts_json", "[]")),
|
||||||
)
|
)
|
||||||
|
|
||||||
sq = (post.get("status_quo_update") or "").strip()
|
sq = (post.get("status_quo_update") or "").strip()
|
||||||
if sq:
|
if sq:
|
||||||
await update_session_status_quo(request.session_id, sq)
|
await update_session_status_quo(request.session_id, sq)
|
||||||
session["status_quo"] = sq
|
debug_blocks.append({"type": "status_quo", "text": sq})
|
||||||
debug_blocks.append({"type": "status_quo", "text": f"--- Status quo ---\n{sq}\n---"})
|
|
||||||
|
|
||||||
if rpg_settings.get("choices", True):
|
if rpg_settings.get("choices", True):
|
||||||
extra_choices = post.get("choices") or []
|
choices += post.get("choices") or []
|
||||||
if extra_choices:
|
|
||||||
choices = choices + extra_choices
|
|
||||||
|
|
||||||
if rpg_settings.get("affinity", True):
|
if rpg_settings.get("affinity", True):
|
||||||
delta = int(post.get("affinity_delta") or 0)
|
delta = int(post.get("affinity_delta") or 0)
|
||||||
if delta:
|
if delta:
|
||||||
await update_session_affinity(request.session_id, delta)
|
await update_session_affinity(request.session_id, delta)
|
||||||
|
|
||||||
|
outfit_update = post.get("outfit_update")
|
||||||
|
if isinstance(outfit_update, list) and outfit_update:
|
||||||
|
outfit_str = json.dumps(outfit_update, ensure_ascii=False)
|
||||||
|
await update_session_outfit(request.session_id, outfit_str)
|
||||||
|
session["outfit_json"] = outfit_str
|
||||||
|
|
||||||
if rpg_settings.get("quests", True):
|
if rpg_settings.get("quests", True):
|
||||||
for qu in (post.get("quest_updates") or []):
|
for qu in (post.get("quest_updates") or []):
|
||||||
title = (qu.get("title") or "").strip()
|
t = (qu.get("title") or "").strip()
|
||||||
status = qu.get("status", "active")
|
if t:
|
||||||
if title:
|
await upsert_quest(request.session_id, t[:120], qu.get("status", "active"))
|
||||||
await upsert_quest(request.session_id, title[:120], status)
|
|
||||||
quests_updated = await get_quests(request.session_id)
|
quests_updated = await get_quests(request.session_id)
|
||||||
|
|
||||||
count = await get_message_count(request.session_id)
|
count = await get_message_count(request.session_id)
|
||||||
if count == 2 and not request.skip_user_add:
|
if count == 2 and not request.skip_user_add:
|
||||||
persona = await get_persona(persona_id) or {}
|
persona = await get_persona(persona_id) or {}
|
||||||
persona_name = persona.get("name", persona_id)
|
|
||||||
preview = request.message[:40] + ("…" if len(request.message) > 40 else "")
|
preview = request.message[:40] + ("…" if len(request.message) > 40 else "")
|
||||||
current = (session or {}).get("title") or "Новый чат"
|
if (session or {}).get("title", "Новый чат") in ("", "Новый чат"):
|
||||||
if current in ("", "Новый чат"):
|
await update_session_title(request.session_id, f"{persona.get('name', persona_id)} — {preview}")
|
||||||
await update_session_title(request.session_id, f"{persona_name} — {preview}")
|
|
||||||
|
|
||||||
image_path = None
|
image_path = None
|
||||||
image_error = None
|
image_error = None
|
||||||
if prompt_str and SD_AUTO_GENERATE:
|
if prompt_str and SD_AUTO_GENERATE:
|
||||||
|
yield f"data: {json.dumps({'image_generating': True, 'image_prompt': prompt_str})}\n\n"
|
||||||
rel, err = await sd_service.generate_from_full_prompt(prompt_str)
|
rel, err = await sd_service.generate_from_full_prompt(prompt_str)
|
||||||
if rel:
|
if rel:
|
||||||
image_path = rel
|
image_path = rel
|
||||||
@@ -467,21 +456,10 @@ async def chat_stream(request: ChatRequest):
|
|||||||
else:
|
else:
|
||||||
image_error = err
|
image_error = err
|
||||||
|
|
||||||
# Fetch current affinity for UI
|
|
||||||
updated_session = await get_session(request.session_id)
|
updated_session = await get_session(request.session_id)
|
||||||
affinity = updated_session.get("affinity", 0) if updated_session else 0
|
affinity = updated_session.get("affinity", 0) if updated_session else 0
|
||||||
|
|
||||||
yield f"data: {json.dumps({
|
yield f"data: {json.dumps({'done': True, 'image_prompt': prompt_str, 'image_path': f'/static/{image_path}' if image_path else None, 'image_error': image_error, 'choices': choices, 'debug': debug_blocks, 'affinity': affinity, 'quests': quests_updated})}\n\n"
|
||||||
'done': True,
|
|
||||||
'image_prompt': prompt_str,
|
|
||||||
'image_path': f'/static/{image_path}' if image_path else None,
|
|
||||||
'image_error': image_error,
|
|
||||||
'choices': choices,
|
|
||||||
'debug': debug_blocks,
|
|
||||||
'narrator': narrator_msg,
|
|
||||||
'affinity': affinity,
|
|
||||||
'quests': quests_updated,
|
|
||||||
})}\n\n"
|
|
||||||
|
|
||||||
return StreamingResponse(
|
return StreamingResponse(
|
||||||
generate(),
|
generate(),
|
||||||
|
|||||||
+4
-11
@@ -1,11 +1,11 @@
|
|||||||
from fastapi import APIRouter, HTTPException
|
from fastapi import APIRouter, HTTPException
|
||||||
from services.memory import (
|
from services.memory import (
|
||||||
get_all_sessions,
|
get_all_sessions,
|
||||||
|
get_session,
|
||||||
get_or_create_session,
|
get_or_create_session,
|
||||||
delete_session,
|
delete_session,
|
||||||
update_session_title,
|
update_session_title,
|
||||||
update_session_persona,
|
update_session_persona,
|
||||||
get_history,
|
|
||||||
get_message_count,
|
get_message_count,
|
||||||
update_session_rpg,
|
update_session_rpg,
|
||||||
update_session_facts,
|
update_session_facts,
|
||||||
@@ -16,7 +16,6 @@ from services.memory import (
|
|||||||
get_quests,
|
get_quests,
|
||||||
get_last_message_preview,
|
get_last_message_preview,
|
||||||
fork_session,
|
fork_session,
|
||||||
get_session,
|
|
||||||
)
|
)
|
||||||
from models.schemas import ForkSessionRequest
|
from models.schemas import ForkSessionRequest
|
||||||
|
|
||||||
@@ -30,11 +29,7 @@ async def list_sessions():
|
|||||||
for s in sessions:
|
for s in sessions:
|
||||||
count = await get_message_count(s["session_id"])
|
count = await get_message_count(s["session_id"])
|
||||||
preview = await get_last_message_preview(s["session_id"])
|
preview = await get_last_message_preview(s["session_id"])
|
||||||
result.append({
|
result.append({**s, "message_count": count, "last_message_preview": preview})
|
||||||
**s,
|
|
||||||
"message_count": count,
|
|
||||||
"last_message_preview": preview,
|
|
||||||
})
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
|
||||||
@@ -44,9 +39,8 @@ async def list_quests(session_id: str):
|
|||||||
|
|
||||||
|
|
||||||
@router.get("/{session_id}")
|
@router.get("/{session_id}")
|
||||||
async def get_session(session_id: str):
|
async def get_session_route(session_id: str):
|
||||||
sessions = await get_all_sessions()
|
s = await get_session(session_id)
|
||||||
s = next((x for x in sessions if x["session_id"] == session_id), None)
|
|
||||||
if not s:
|
if not s:
|
||||||
raise HTTPException(status_code=404, detail="Сессия не найдена")
|
raise HTTPException(status_code=404, detail="Сессия не найдена")
|
||||||
return s
|
return s
|
||||||
@@ -86,4 +80,3 @@ async def fork_session_route(session_id: str, req: ForkSessionRequest):
|
|||||||
async def remove_session(session_id: str):
|
async def remove_session(session_id: str):
|
||||||
await delete_session(session_id)
|
await delete_session(session_id)
|
||||||
return {"status": "deleted", "session_id": session_id}
|
return {"status": "deleted", "session_id": session_id}
|
||||||
|
|
||||||
|
|||||||
+122
-31
@@ -7,7 +7,19 @@ import aiosqlite
|
|||||||
from database.db import DB_PATH
|
from database.db import DB_PATH
|
||||||
|
|
||||||
|
|
||||||
def parse_card_v2(data: dict) -> dict:
|
def _normalize_alternate_greetings(inner: dict) -> list[str]:
|
||||||
|
raw = inner.get("alternate_greetings") or []
|
||||||
|
if not isinstance(raw, list):
|
||||||
|
return []
|
||||||
|
out = []
|
||||||
|
for item in raw:
|
||||||
|
text = str(item).strip()
|
||||||
|
if text and text not in out:
|
||||||
|
out.append(text)
|
||||||
|
return out
|
||||||
|
|
||||||
|
|
||||||
|
def parse_card_v2(data: dict, card_id: str | None = None) -> dict:
|
||||||
inner = data.get("data", data)
|
inner = data.get("data", data)
|
||||||
if isinstance(inner, str):
|
if isinstance(inner, str):
|
||||||
inner = json.loads(inner)
|
inner = json.loads(inner)
|
||||||
@@ -17,12 +29,15 @@ def parse_card_v2(data: dict) -> dict:
|
|||||||
if isinstance(entries, dict):
|
if isinstance(entries, dict):
|
||||||
entries = list(entries.values())
|
entries = list(entries.values())
|
||||||
|
|
||||||
return {
|
alternates = _normalize_alternate_greetings(inner)
|
||||||
"card_id": (
|
cid = card_id or (
|
||||||
inner.get("name", "imported").lower().replace(" ", "_")[:48]
|
inner.get("name", "imported").lower().replace(" ", "_")[:48]
|
||||||
+ "_"
|
+ "_"
|
||||||
+ uuid.uuid4().hex[:8]
|
+ uuid.uuid4().hex[:8]
|
||||||
),
|
)
|
||||||
|
|
||||||
|
return {
|
||||||
|
"card_id": cid,
|
||||||
"name": inner.get("name", "Character"),
|
"name": inner.get("name", "Character"),
|
||||||
"description": inner.get("description", ""),
|
"description": inner.get("description", ""),
|
||||||
"personality": inner.get("personality", ""),
|
"personality": inner.get("personality", ""),
|
||||||
@@ -31,10 +46,22 @@ def parse_card_v2(data: dict) -> dict:
|
|||||||
"mes_example": inner.get("mes_example", ""),
|
"mes_example": inner.get("mes_example", ""),
|
||||||
"appearance_tags": _extract_appearance(inner),
|
"appearance_tags": _extract_appearance(inner),
|
||||||
"lorebook_json": json.dumps(entries, ensure_ascii=False),
|
"lorebook_json": json.dumps(entries, ensure_ascii=False),
|
||||||
|
"alternate_greetings": alternates,
|
||||||
|
"alternate_greetings_json": json.dumps(alternates, ensure_ascii=False),
|
||||||
"raw_json": json.dumps(data if "data" in data else {"data": inner}, ensure_ascii=False),
|
"raw_json": json.dumps(data if "data" in data else {"data": inner}, ensure_ascii=False),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def parse_card_bytes(content: bytes, filename: str) -> dict:
|
||||||
|
if filename.lower().endswith(".png"):
|
||||||
|
card = parse_png_card(content)
|
||||||
|
if not card:
|
||||||
|
raise ValueError("PNG does not contain character card metadata")
|
||||||
|
card["_png_bytes"] = content
|
||||||
|
return card
|
||||||
|
return parse_card_v2(json.loads(content.decode("utf-8")))
|
||||||
|
|
||||||
|
|
||||||
def _extract_appearance(inner: dict) -> str:
|
def _extract_appearance(inner: dict) -> str:
|
||||||
"""Extract booru-style appearance tags from character fields."""
|
"""Extract booru-style appearance tags from character fields."""
|
||||||
import re
|
import re
|
||||||
@@ -107,12 +134,18 @@ def build_system_prompt(card: dict) -> str:
|
|||||||
|
|
||||||
async def save_character(card: dict, lora_name: str = "", lora_weight: float = 0.8) -> dict:
|
async def save_character(card: dict, lora_name: str = "", lora_weight: float = 0.8) -> dict:
|
||||||
card_id = card["card_id"]
|
card_id = card["card_id"]
|
||||||
|
alt_json = card.get("alternate_greetings_json")
|
||||||
|
if alt_json is None:
|
||||||
|
alts = card.get("alternate_greetings") or []
|
||||||
|
alt_json = json.dumps(alts, ensure_ascii=False) if isinstance(alts, list) else "[]"
|
||||||
|
|
||||||
async with aiosqlite.connect(DB_PATH) as db:
|
async with aiosqlite.connect(DB_PATH) as db:
|
||||||
await db.execute(
|
await db.execute(
|
||||||
"""INSERT OR REPLACE INTO characters
|
"""INSERT OR REPLACE INTO characters
|
||||||
(card_id, name, description, personality, scenario, first_mes,
|
(card_id, name, description, personality, scenario, first_mes,
|
||||||
mes_example, raw_json, lora_name, lora_weight, appearance_tags, lorebook_json, avatar_path)
|
mes_example, raw_json, lora_name, lora_weight, appearance_tags, lorebook_json,
|
||||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""",
|
avatar_path, alternate_greetings_json)
|
||||||
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""",
|
||||||
(
|
(
|
||||||
card_id,
|
card_id,
|
||||||
card["name"],
|
card["name"],
|
||||||
@@ -127,6 +160,7 @@ async def save_character(card: dict, lora_name: str = "", lora_weight: float = 0
|
|||||||
card.get("appearance_tags", ""),
|
card.get("appearance_tags", ""),
|
||||||
card["lorebook_json"],
|
card["lorebook_json"],
|
||||||
card.get("avatar_path", ""),
|
card.get("avatar_path", ""),
|
||||||
|
alt_json,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
await db.commit()
|
await db.commit()
|
||||||
@@ -140,7 +174,7 @@ async def get_character(card_id: str) -> dict | None:
|
|||||||
"SELECT * FROM characters WHERE card_id = ?", (card_id,)
|
"SELECT * FROM characters WHERE card_id = ?", (card_id,)
|
||||||
) as cur:
|
) as cur:
|
||||||
row = await cur.fetchone()
|
row = await cur.fetchone()
|
||||||
return dict(row) if row else None
|
return card_to_api(dict(row)) if row else None
|
||||||
|
|
||||||
|
|
||||||
async def list_characters() -> list:
|
async def list_characters() -> list:
|
||||||
@@ -171,9 +205,31 @@ async def update_appearance_tags(card_id: str, appearance_tags: str):
|
|||||||
await db.commit()
|
await db.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def card_to_api(card: dict) -> dict:
|
||||||
|
alts = card.get("alternate_greetings")
|
||||||
|
if alts is None:
|
||||||
|
try:
|
||||||
|
alts = json.loads(card.get("alternate_greetings_json") or "[]")
|
||||||
|
except Exception:
|
||||||
|
alts = []
|
||||||
|
if not isinstance(alts, list):
|
||||||
|
alts = []
|
||||||
|
return {**card, "alternate_greetings": alts}
|
||||||
|
|
||||||
|
|
||||||
|
async def preview_card_file(content: bytes, filename: str) -> dict:
|
||||||
|
card = parse_card_bytes(content, filename)
|
||||||
|
png_bytes = card.pop("_png_bytes", None)
|
||||||
|
preview = card_to_api(card)
|
||||||
|
preview["is_png"] = bool(png_bytes)
|
||||||
|
preview["alternate_count"] = len(preview.get("alternate_greetings") or [])
|
||||||
|
return preview
|
||||||
|
|
||||||
|
|
||||||
async def update_character(card_id: str, fields: dict) -> bool:
|
async def update_character(card_id: str, fields: dict) -> bool:
|
||||||
allowed = {"name", "description", "personality", "scenario", "first_mes",
|
allowed = {"name", "description", "personality", "scenario", "first_mes",
|
||||||
"mes_example", "appearance_tags", "lora_name", "lora_weight", "avatar_path"}
|
"mes_example", "appearance_tags", "lora_name", "lora_weight", "avatar_path",
|
||||||
|
"alternate_greetings_json"}
|
||||||
updates = {k: v for k, v in fields.items() if k in allowed}
|
updates = {k: v for k, v in fields.items() if k in allowed}
|
||||||
if not updates:
|
if not updates:
|
||||||
return False
|
return False
|
||||||
@@ -187,37 +243,72 @@ async def update_character(card_id: str, fields: dict) -> bool:
|
|||||||
return cur.rowcount > 0
|
return cur.rowcount > 0
|
||||||
|
|
||||||
|
|
||||||
async def import_card_file(content: bytes, filename: str, lora_name: str = "", lora_weight: float = 0.8) -> dict:
|
async def import_card_file(
|
||||||
if filename.lower().endswith(".png"):
|
content: bytes,
|
||||||
card = parse_png_card(content)
|
filename: str,
|
||||||
if not card:
|
lora_name: str = "",
|
||||||
raise ValueError("PNG does not contain character card metadata")
|
lora_weight: float = 0.8,
|
||||||
# Use the PNG itself as avatar
|
overrides: dict | None = None,
|
||||||
avatar_rel = _save_avatar_bytes(content, f"card_{card['card_id']}")
|
card_id: str | None = None,
|
||||||
|
) -> dict:
|
||||||
|
card = parse_card_bytes(content, filename)
|
||||||
|
png_bytes = card.pop("_png_bytes", None)
|
||||||
|
|
||||||
|
if card_id:
|
||||||
|
card["card_id"] = card_id
|
||||||
|
|
||||||
|
if overrides:
|
||||||
|
for key in (
|
||||||
|
"name", "description", "personality", "scenario", "first_mes",
|
||||||
|
"mes_example", "appearance_tags", "lorebook_json",
|
||||||
|
):
|
||||||
|
if key in overrides and overrides[key] is not None:
|
||||||
|
card[key] = overrides[key]
|
||||||
|
if overrides.get("alternate_greetings_json") is not None:
|
||||||
|
card["alternate_greetings_json"] = overrides["alternate_greetings_json"]
|
||||||
|
elif overrides.get("alternate_greetings") is not None:
|
||||||
|
alts = overrides["alternate_greetings"]
|
||||||
|
if isinstance(alts, str):
|
||||||
|
try:
|
||||||
|
alts = json.loads(alts)
|
||||||
|
except Exception:
|
||||||
|
alts = []
|
||||||
|
card["alternate_greetings"] = alts
|
||||||
|
card["alternate_greetings_json"] = json.dumps(alts, ensure_ascii=False)
|
||||||
|
|
||||||
|
if png_bytes:
|
||||||
|
avatar_rel = _save_avatar_bytes(png_bytes, f"card_{card['card_id']}")
|
||||||
card["avatar_path"] = avatar_rel
|
card["avatar_path"] = avatar_rel
|
||||||
else:
|
|
||||||
card = parse_card_v2(json.loads(content.decode("utf-8")))
|
|
||||||
|
|
||||||
saved = await save_character(card, lora_name=lora_name, lora_weight=lora_weight)
|
saved = await save_character(card, lora_name=lora_name, lora_weight=lora_weight)
|
||||||
|
|
||||||
persona_id = f"card_{saved['card_id']}"
|
persona_id = f"card_{saved['card_id']}"
|
||||||
from services.personas import create_persona, get_persona
|
from services.personas import create_persona, get_persona, patch_persona
|
||||||
|
|
||||||
existing = await get_persona(persona_id)
|
existing = await get_persona(persona_id)
|
||||||
|
persona_fields = {
|
||||||
|
"name": saved["name"],
|
||||||
|
"emoji": "🎭",
|
||||||
|
"description": (saved["description"] or "")[:80] or "Character card",
|
||||||
|
"prompt": build_system_prompt(saved),
|
||||||
|
"sd_enabled": True,
|
||||||
|
"lora_name": lora_name,
|
||||||
|
"lora_weight": lora_weight,
|
||||||
|
"appearance_tags": saved.get("appearance_tags", ""),
|
||||||
|
"avatar_path": saved.get("avatar_path", ""),
|
||||||
|
"personality": saved.get("personality", ""),
|
||||||
|
"scenario": saved.get("scenario", ""),
|
||||||
|
"first_mes": saved.get("first_mes", ""),
|
||||||
|
"mes_example": saved.get("mes_example", ""),
|
||||||
|
"lorebook_json": saved.get("lorebook_json", "[]"),
|
||||||
|
"alternate_greetings_json": saved.get("alternate_greetings_json", "[]"),
|
||||||
|
}
|
||||||
if not existing:
|
if not existing:
|
||||||
await create_persona(
|
await create_persona(persona_id=persona_id, **persona_fields)
|
||||||
persona_id=persona_id,
|
else:
|
||||||
name=saved["name"],
|
await patch_persona(persona_id, persona_fields)
|
||||||
emoji="🎭",
|
|
||||||
description=saved["description"][:80] or "Character card",
|
return card_to_api(saved)
|
||||||
prompt=build_system_prompt(saved),
|
|
||||||
sd_enabled=True,
|
|
||||||
lora_name=lora_name,
|
|
||||||
lora_weight=lora_weight,
|
|
||||||
appearance_tags=saved.get("appearance_tags", ""),
|
|
||||||
avatar_path=saved.get("avatar_path", ""),
|
|
||||||
)
|
|
||||||
return saved
|
|
||||||
|
|
||||||
|
|
||||||
def _save_avatar_bytes(png_bytes: bytes, prefix: str) -> str:
|
def _save_avatar_bytes(png_bytes: bytes, prefix: str) -> str:
|
||||||
|
|||||||
+51
-44
@@ -1,12 +1,18 @@
|
|||||||
import httpx
|
import httpx
|
||||||
|
import json
|
||||||
|
import logging
|
||||||
import os
|
import os
|
||||||
from dotenv import load_dotenv
|
from dotenv import load_dotenv
|
||||||
|
|
||||||
load_dotenv()
|
load_dotenv()
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
OPENROUTER_KEY = os.getenv("ROUTER_KEY")
|
OPENROUTER_KEY = os.getenv("ROUTER_KEY")
|
||||||
OPENROUTER_URL = "https://openrouter.ai/api/v1/chat/completions"
|
OPENROUTER_URL = "https://openrouter.ai/api/v1/chat/completions"
|
||||||
MODEL = "google/gemini-2.5-flash"
|
|
||||||
|
CHAT_MODEL = os.getenv("CHAT_MODEL", "mistralai/mistral-nemo")
|
||||||
|
SYSTEM_MODEL = os.getenv("SYSTEM_MODEL", "google/gemini-2.5-flash")
|
||||||
|
|
||||||
HEADERS = {
|
HEADERS = {
|
||||||
"Authorization": f"Bearer {OPENROUTER_KEY}",
|
"Authorization": f"Bearer {OPENROUTER_KEY}",
|
||||||
@@ -14,66 +20,67 @@ HEADERS = {
|
|||||||
"HTTP-Referer": "http://localhost:8000",
|
"HTTP-Referer": "http://localhost:8000",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def _clean(messages: list) -> list:
|
||||||
|
"""Filter out messages with empty content."""
|
||||||
|
return [m for m in messages if (m.get("content") or "").strip()]
|
||||||
|
|
||||||
|
|
||||||
|
async def _post(model: str, messages: list, extra: dict | None = None) -> str:
|
||||||
|
payload = {"model": model, "messages": _clean(messages), **(extra or {})}
|
||||||
|
async with httpx.AsyncClient(timeout=90) as client:
|
||||||
|
r = await client.post(OPENROUTER_URL, headers=HEADERS, json=payload)
|
||||||
|
r.raise_for_status()
|
||||||
|
return r.json()["choices"][0]["message"]["content"]
|
||||||
|
|
||||||
|
|
||||||
async def send_message(messages: list) -> str:
|
async def send_message(messages: list) -> str:
|
||||||
"""Обычный запрос — используем для внутренних нужд"""
|
"""System model — narrator, facts, SD prompt."""
|
||||||
payload = {
|
return await _post(SYSTEM_MODEL, messages)
|
||||||
"model": MODEL,
|
|
||||||
"messages": messages,
|
|
||||||
}
|
|
||||||
async with httpx.AsyncClient(timeout=60) as client:
|
|
||||||
response = await client.post(
|
|
||||||
OPENROUTER_URL,
|
|
||||||
headers=HEADERS,
|
|
||||||
json=payload
|
|
||||||
)
|
|
||||||
response.raise_for_status()
|
|
||||||
data = response.json()
|
|
||||||
return data["choices"][0]["message"]["content"]
|
|
||||||
|
|
||||||
|
|
||||||
async def send_message_with_model(messages: list, model: str) -> str:
|
async def send_message_with_model(messages: list, model: str) -> str:
|
||||||
payload = {
|
"""Explicit model — plot arc, narrator override."""
|
||||||
"model": model,
|
return await _post(model, messages)
|
||||||
"messages": messages,
|
|
||||||
}
|
|
||||||
async with httpx.AsyncClient(timeout=90) as client:
|
|
||||||
response = await client.post(
|
|
||||||
OPENROUTER_URL,
|
|
||||||
headers=HEADERS,
|
|
||||||
json=payload
|
|
||||||
)
|
|
||||||
response.raise_for_status()
|
|
||||||
data = response.json()
|
|
||||||
return data["choices"][0]["message"]["content"]
|
|
||||||
|
|
||||||
|
|
||||||
async def stream_message(messages: list):
|
async def stream_message(messages: list):
|
||||||
"""Стриминг — отдаём чанки по мере получения"""
|
"""Chat model stream — roleplay dialogue."""
|
||||||
payload = {
|
payload = {
|
||||||
"model": MODEL,
|
"model": CHAT_MODEL,
|
||||||
"messages": messages,
|
"messages": _clean(messages),
|
||||||
"stream": True,
|
"stream": True,
|
||||||
}
|
}
|
||||||
async with httpx.AsyncClient(timeout=60) as client:
|
timeout = httpx.Timeout(connect=10, read=120, write=10, pool=5)
|
||||||
async with client.stream(
|
chunk_count = 0
|
||||||
"POST",
|
async with httpx.AsyncClient(timeout=timeout) as client:
|
||||||
OPENROUTER_URL,
|
try:
|
||||||
headers=HEADERS,
|
async with client.stream("POST", OPENROUTER_URL, headers=HEADERS, json=payload) as response:
|
||||||
json=payload
|
|
||||||
) as response:
|
|
||||||
response.raise_for_status()
|
response.raise_for_status()
|
||||||
async for line in response.aiter_lines():
|
buf = ""
|
||||||
|
async for raw in response.aiter_bytes():
|
||||||
|
text = raw.decode("utf-8", errors="replace")
|
||||||
|
if not buf and chunk_count == 0:
|
||||||
|
logger.info("stream first bytes: %.200s", text)
|
||||||
|
buf += text
|
||||||
|
while "\n" in buf:
|
||||||
|
line, buf = buf.split("\n", 1)
|
||||||
|
line = line.rstrip("\r")
|
||||||
if not line.startswith("data: "):
|
if not line.startswith("data: "):
|
||||||
continue
|
continue
|
||||||
data = line[6:] # убираем "data: "
|
data = line[6:]
|
||||||
if data == "[DONE]":
|
if data == "[DONE]":
|
||||||
break
|
return
|
||||||
try:
|
try:
|
||||||
import json
|
|
||||||
chunk = json.loads(data)
|
chunk = json.loads(data)
|
||||||
delta = chunk["choices"][0]["delta"]
|
content = chunk["choices"][0]["delta"].get("content", "")
|
||||||
content = delta.get("content", "")
|
|
||||||
if content:
|
if content:
|
||||||
|
chunk_count += 1
|
||||||
yield content
|
yield content
|
||||||
except Exception:
|
except Exception:
|
||||||
continue
|
continue
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("stream_message error after %d chunks: %s", chunk_count, e)
|
||||||
|
raise
|
||||||
|
finally:
|
||||||
|
logger.info("stream_message finished: %d chunks", chunk_count)
|
||||||
|
|||||||
@@ -380,6 +380,15 @@ async def update_session_rpg_settings(session_id: str, settings_json: str):
|
|||||||
await db.commit()
|
await db.commit()
|
||||||
|
|
||||||
|
|
||||||
|
async def update_session_outfit(session_id: str, outfit_json: str):
|
||||||
|
async with aiosqlite.connect(DB_PATH) as db:
|
||||||
|
await db.execute(
|
||||||
|
"UPDATE sessions SET outfit_json = ?, updated_at = CURRENT_TIMESTAMP WHERE session_id = ?",
|
||||||
|
(outfit_json, session_id),
|
||||||
|
)
|
||||||
|
await db.commit()
|
||||||
|
|
||||||
|
|
||||||
async def upsert_quest(session_id: str, title: str, status: str = "active"):
|
async def upsert_quest(session_id: str, title: str, status: str = "active"):
|
||||||
async with aiosqlite.connect(DB_PATH) as db:
|
async with aiosqlite.connect(DB_PATH) as db:
|
||||||
async with db.execute(
|
async with db.execute(
|
||||||
|
|||||||
@@ -69,6 +69,7 @@ def _row_to_persona(row: dict) -> dict:
|
|||||||
"mes_example": row.get("mes_example", "") or "",
|
"mes_example": row.get("mes_example", "") or "",
|
||||||
"lorebook_json": row.get("lorebook_json", "[]") or "[]",
|
"lorebook_json": row.get("lorebook_json", "[]") or "[]",
|
||||||
"avatar_path": row.get("avatar_path", "") or "",
|
"avatar_path": row.get("avatar_path", "") or "",
|
||||||
|
"alternate_greetings_json": row.get("alternate_greetings_json", "[]") or "[]",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -122,6 +123,7 @@ async def create_persona(
|
|||||||
mes_example: str = "",
|
mes_example: str = "",
|
||||||
lorebook_json: str = "[]",
|
lorebook_json: str = "[]",
|
||||||
avatar_path: str = "",
|
avatar_path: str = "",
|
||||||
|
alternate_greetings_json: str = "[]",
|
||||||
) -> dict:
|
) -> dict:
|
||||||
final_prompt = prompt.strip() or build_persona_prompt(
|
final_prompt = prompt.strip() or build_persona_prompt(
|
||||||
{
|
{
|
||||||
@@ -137,12 +139,14 @@ async def create_persona(
|
|||||||
"""INSERT INTO personas
|
"""INSERT INTO personas
|
||||||
(persona_id, name, emoji, description, prompt, custom,
|
(persona_id, name, emoji, description, prompt, custom,
|
||||||
sd_enabled, lora_name, lora_weight, appearance_tags,
|
sd_enabled, lora_name, lora_weight, appearance_tags,
|
||||||
personality, scenario, first_mes, mes_example, lorebook_json, avatar_path)
|
personality, scenario, first_mes, mes_example, lorebook_json, avatar_path,
|
||||||
VALUES (?, ?, ?, ?, ?, 1, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""",
|
alternate_greetings_json)
|
||||||
|
VALUES (?, ?, ?, ?, ?, 1, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""",
|
||||||
(
|
(
|
||||||
persona_id, name, emoji, description, final_prompt,
|
persona_id, name, emoji, description, final_prompt,
|
||||||
1 if sd_enabled else 0, lora_name, lora_weight, appearance_tags,
|
1 if sd_enabled else 0, lora_name, lora_weight, appearance_tags,
|
||||||
personality, scenario, first_mes, mes_example, lorebook_json, avatar_path,
|
personality, scenario, first_mes, mes_example, lorebook_json, avatar_path,
|
||||||
|
alternate_greetings_json,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
await db.commit()
|
await db.commit()
|
||||||
@@ -162,6 +166,7 @@ async def create_persona(
|
|||||||
"mes_example": mes_example,
|
"mes_example": mes_example,
|
||||||
"lorebook_json": lorebook_json,
|
"lorebook_json": lorebook_json,
|
||||||
"avatar_path": avatar_path,
|
"avatar_path": avatar_path,
|
||||||
|
"alternate_greetings_json": alternate_greetings_json,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -227,6 +232,7 @@ async def patch_persona(persona_id: str, fields: dict) -> bool:
|
|||||||
"mes_example",
|
"mes_example",
|
||||||
"lorebook_json",
|
"lorebook_json",
|
||||||
"avatar_path",
|
"avatar_path",
|
||||||
|
"alternate_greetings_json",
|
||||||
}
|
}
|
||||||
updates = {k: v for k, v in fields.items() if k in allowed}
|
updates = {k: v for k, v in fields.items() if k in allowed}
|
||||||
if not updates:
|
if not updates:
|
||||||
|
|||||||
@@ -35,12 +35,14 @@ Return ONLY valid JSON (no markdown):
|
|||||||
"facts": ["durable facts only"],
|
"facts": ["durable facts only"],
|
||||||
"choices": [{"id":"a","label":"..."}, ...],
|
"choices": [{"id":"a","label":"..."}, ...],
|
||||||
"affinity_delta": 0,
|
"affinity_delta": 0,
|
||||||
"quest_updates": [{"title": "quest title", "status": "active|done|failed"}]
|
"quest_updates": [{"title": "quest title", "status": "active|done|failed"}],
|
||||||
|
"outfit_update": ["danbooru_tag", "danbooru_tag"]
|
||||||
}
|
}
|
||||||
Rules:
|
Rules:
|
||||||
- affinity_delta: integer -2..+2. Positive if character warmed up to player, negative if pushed away. 0 if neutral.
|
- affinity_delta: integer -2..+2. Positive if character warmed up to player, negative if pushed away. 0 if neutral.
|
||||||
- quest_updates: only include if a quest was clearly started, completed, or failed. Empty array otherwise.
|
- quest_updates: only include if a quest was clearly started, completed, or failed. Empty array otherwise.
|
||||||
- choices: 0-4 options for what the player can do next."""
|
- choices: 0-4 options for what the player can do next.
|
||||||
|
- outfit_update: ONLY include if the character's clothing visibly changed (put on, took off, changed outfit). Use exact danbooru-style underscore_tags (e.g. ["white_dress", "red_ribbon", "barefoot"]). Empty array if no change."""
|
||||||
|
|
||||||
|
|
||||||
async def narrator_pre(
|
async def narrator_pre(
|
||||||
|
|||||||
+19
-1
@@ -27,7 +27,7 @@ Return ONLY valid JSON (no markdown):
|
|||||||
"cast": [{"name":"NPC name","role":"helper|antagonist|bystander","motivation":"..."}],
|
"cast": [{"name":"NPC name","role":"helper|antagonist|bystander","motivation":"..."}],
|
||||||
"secrets": ["hidden truths not revealed yet"],
|
"secrets": ["hidden truths not revealed yet"],
|
||||||
"beats": [
|
"beats": [
|
||||||
{"id":"b1","trigger":"event_driven:rest|event_driven:travel|event_driven:help_request|event_driven:after_fail|event_driven:after_success",
|
{"id":"b1","title":"short quest title (3-6 words)","trigger":"event_driven:rest|event_driven:travel|event_driven:help_request|event_driven:after_fail|event_driven:after_success",
|
||||||
"injection":"1-3 sentences to introduce the beat WITHOUT breaking current scene",
|
"injection":"1-3 sentences to introduce the beat WITHOUT breaking current scene",
|
||||||
"choices":[{"id":"a","label":"..."},{"id":"b","label":"..."}]}
|
"choices":[{"id":"a","label":"..."},{"id":"b","label":"..."}]}
|
||||||
],
|
],
|
||||||
@@ -90,6 +90,24 @@ def should_advance_arc(user_text: str) -> str | None:
|
|||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
PHASE_ORDER = ["opening", "hook", "complication", "reveal", "climax", "aftermath"]
|
||||||
|
|
||||||
|
|
||||||
|
def advance_phase(arc: dict) -> bool:
|
||||||
|
"""Advance arc to next phase if beats are exhausted. Returns True if phase changed."""
|
||||||
|
current = arc.get("phase", "opening")
|
||||||
|
if arc.get("beats"):
|
||||||
|
return False
|
||||||
|
try:
|
||||||
|
idx = PHASE_ORDER.index(current)
|
||||||
|
except ValueError:
|
||||||
|
return False
|
||||||
|
if idx + 1 >= len(PHASE_ORDER):
|
||||||
|
return False
|
||||||
|
arc["phase"] = PHASE_ORDER[idx + 1]
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
def pop_matching_beats(arc: dict, trigger: str, max_beats: int = 1) -> tuple[dict, list[dict]]:
|
def pop_matching_beats(arc: dict, trigger: str, max_beats: int = 1) -> tuple[dict, list[dict]]:
|
||||||
beats = arc.get("beats", [])
|
beats = arc.get("beats", [])
|
||||||
if not isinstance(beats, list):
|
if not isinstance(beats, list):
|
||||||
|
|||||||
+64
-33
@@ -1,20 +1,24 @@
|
|||||||
import json
|
import json
|
||||||
|
import logging
|
||||||
import os
|
import os
|
||||||
import re
|
import re
|
||||||
from services.llm import send_message
|
|
||||||
|
from services.llm import send_message, send_message_with_model
|
||||||
from services.personas import get_persona
|
from services.personas import get_persona
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
PROMPT_BUILDER_SYSTEM = """You are a Stable Diffusion prompt engineer for anime illustration models.
|
PROMPT_BUILDER_SYSTEM = """You are a Stable Diffusion prompt engineer for anime illustration models.
|
||||||
Given a roleplay chat excerpt and character appearance hints, output ONLY valid JSON (no markdown):
|
Given a roleplay chat excerpt, output ONLY valid JSON (no markdown):
|
||||||
{
|
{
|
||||||
"should_generate": true,
|
"should_generate": true,
|
||||||
"shot_type": "first_person_pov" | "landscape" | "third_person",
|
"shot_type": "first_person_pov" | "landscape" | "third_person",
|
||||||
"appearance_tags": "booru-style tags for character appearance extracted from hints, e.g. 'white hair, wolf ears, wolf tail, yellow eyes'",
|
"action_tags": "booru-style tags for pose/action/expression, e.g. 'sitting, smiling, holding_cup'",
|
||||||
"action_tags": "booru-style tags for pose/action, e.g. 'sitting, smiling, looking at viewer'",
|
"environment_tags": "booru-style tags for location/lighting/time, e.g. 'indoors, kitchen, sunlight, daytime'"
|
||||||
"environment_tags": "booru-style tags for location/lighting, e.g. 'indoors, kitchen, sunlight'"
|
|
||||||
}
|
}
|
||||||
Rules:
|
Rules:
|
||||||
- ONLY use real danbooru/e621 tags. Multi-word concepts MUST be written as single tags: 'white hair' not 'white, hair'. 'wolf ears' not 'wolf, ears'.
|
- ONLY use real danbooru/e621 tags. Multi-word concepts MUST be underscore_joined: 'fox_ears' not 'fox ears'.
|
||||||
|
- Do NOT include appearance/character tags — those are provided separately.
|
||||||
- Do NOT include quality tags, model names, style words, 'pov', or category/metadata words.
|
- Do NOT include quality tags, model names, style words, 'pov', or category/metadata words.
|
||||||
- Do NOT invent tags. If unsure — omit.
|
- Do NOT invent tags. If unsure — omit.
|
||||||
- Keep each field to 3-6 tags."""
|
- Keep each field to 3-6 tags."""
|
||||||
@@ -35,19 +39,38 @@ def strip_image_prompt_tag(text: str) -> str:
|
|||||||
return re.sub(r"\[IMAGE_PROMPT:.*?\]", "", text, flags=re.DOTALL).strip()
|
return re.sub(r"\[IMAGE_PROMPT:.*?\]", "", text, flags=re.DOTALL).strip()
|
||||||
|
|
||||||
|
|
||||||
PONY_CHECKPOINTS = {"ponyDiffusionV6XL_v6StartWithThisOne.safetensors"}
|
|
||||||
SD_CHECKPOINT = os.getenv("SD_CHECKPOINT", "")
|
SD_CHECKPOINT = os.getenv("SD_CHECKPOINT", "")
|
||||||
PONY_NEGATIVE = "score_1, score_2, score_3, score_4, worst quality, low quality, blurry, bad anatomy, watermark, text, censored"
|
SD_UNET = os.getenv("SD_UNET", "")
|
||||||
|
SD_PROMPT_MODEL = os.getenv("SD_PROMPT_MODEL", "").strip()
|
||||||
|
|
||||||
|
PONY_CHECKPOINTS = {"ponyDiffusionV6XL_v6StartWithThisOne.safetensors"}
|
||||||
|
PONY_NEGATIVE = "score_1, score_2, score_3, score_4, worst quality, low quality, blurry, bad anatomy, watermark, text, censored"
|
||||||
|
ANIMA_NEGATIVE = "worst quality, low quality, score_1, score_2, score_3, blurry, jpeg artifacts, sepia"
|
||||||
|
|
||||||
|
|
||||||
|
def _is_pony() -> bool:
|
||||||
|
return SD_CHECKPOINT in PONY_CHECKPOINTS
|
||||||
|
|
||||||
|
|
||||||
|
def _is_anima() -> bool:
|
||||||
|
return bool(SD_UNET) and not SD_CHECKPOINT
|
||||||
|
|
||||||
|
|
||||||
|
def build_positive_prompt(scene: dict, persona: dict | None, outfit_tags: str = "") -> str:
|
||||||
|
if _is_pony():
|
||||||
|
quality = "score_9, score_8_up, score_7_up, source_anime, highres"
|
||||||
|
elif _is_anima():
|
||||||
|
quality = "masterpiece, best quality, score_7, anime"
|
||||||
|
else:
|
||||||
|
quality = "masterpiece, best quality, highres"
|
||||||
|
|
||||||
def build_positive_prompt(scene: dict, persona: dict | None) -> str:
|
|
||||||
is_pony = SD_CHECKPOINT in PONY_CHECKPOINTS
|
|
||||||
quality = "score_9, score_8_up, score_7_up, source_anime, highres" if is_pony else "masterpiece, best quality, highres"
|
|
||||||
parts = [quality]
|
parts = [quality]
|
||||||
|
|
||||||
# prefer LLM-extracted appearance over raw persona tags
|
appearance = (persona or {}).get("appearance_tags", "")
|
||||||
appearance = scene.get("appearance_tags") or (persona or {}).get("appearance_tags", "")
|
|
||||||
if appearance:
|
if appearance:
|
||||||
parts.append(appearance)
|
parts.append(appearance)
|
||||||
|
if outfit_tags:
|
||||||
|
parts.append(outfit_tags)
|
||||||
|
|
||||||
if scene.get("shot_type") == "landscape":
|
if scene.get("shot_type") == "landscape":
|
||||||
parts.append(scene.get("environment_tags", ""))
|
parts.append(scene.get("environment_tags", ""))
|
||||||
@@ -75,9 +98,12 @@ def build_positive_prompt(scene: dict, persona: dict | None) -> str:
|
|||||||
async def generate_sd_prompt(
|
async def generate_sd_prompt(
|
||||||
messages: list,
|
messages: list,
|
||||||
persona_id: str,
|
persona_id: str,
|
||||||
|
outfit_json: str = "[]",
|
||||||
) -> tuple[str | None, str | None]:
|
) -> tuple[str | None, str | None]:
|
||||||
persona = await get_persona(persona_id)
|
persona = await get_persona(persona_id)
|
||||||
if not persona or not persona.get("sd_enabled"):
|
# Generate only if persona has appearance tags
|
||||||
|
if not persona or not (persona.get("appearance_tags") or "").strip():
|
||||||
|
logger.debug("sd_prompt skip: persona=%s no appearance_tags", persona_id)
|
||||||
return None, None
|
return None, None
|
||||||
|
|
||||||
recent = [m for m in messages if m["role"] in ("user", "assistant")][-6:]
|
recent = [m for m in messages if m["role"] in ("user", "assistant")][-6:]
|
||||||
@@ -86,40 +112,45 @@ async def generate_sd_prompt(
|
|||||||
|
|
||||||
excerpt = "\n".join(f"{m['role']}: {strip_image_prompt_tag(m['content'])}" for m in recent)
|
excerpt = "\n".join(f"{m['role']}: {strip_image_prompt_tag(m['content'])}" for m in recent)
|
||||||
|
|
||||||
appearance = persona.get("appearance_tags", "")
|
|
||||||
# For card personas, also include description for better visual context
|
|
||||||
if persona_id.startswith("card_"):
|
|
||||||
from services.character_card import get_character
|
|
||||||
card = await get_character(persona_id[5:])
|
|
||||||
if card and card.get("description"):
|
|
||||||
appearance = f"{appearance}\nCharacter description: {card['description'][:400]}"
|
|
||||||
|
|
||||||
builder_messages = [
|
builder_messages = [
|
||||||
{"role": "system", "content": PROMPT_BUILDER_SYSTEM},
|
{"role": "system", "content": PROMPT_BUILDER_SYSTEM},
|
||||||
{
|
{"role": "user", "content": f"Chat:\n{excerpt}"},
|
||||||
"role": "user",
|
|
||||||
"content": f"Persona appearance hints: {appearance}\n\nChat:\n{excerpt}",
|
|
||||||
},
|
|
||||||
]
|
]
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
if SD_PROMPT_MODEL:
|
||||||
|
raw = await send_message_with_model(builder_messages, SD_PROMPT_MODEL)
|
||||||
|
else:
|
||||||
raw = await send_message(builder_messages)
|
raw = await send_message(builder_messages)
|
||||||
raw = raw.strip()
|
raw = raw.strip()
|
||||||
if raw.startswith("```"):
|
if raw.startswith("```"):
|
||||||
raw = re.sub(r"^```\w*\n?", "", raw)
|
raw = re.sub(r"^```\w*\n?", "", raw)
|
||||||
raw = re.sub(r"\n?```$", "", raw)
|
raw = re.sub(r"\n?```$", "", raw)
|
||||||
scene = json.loads(raw)
|
scene = json.loads(raw)
|
||||||
except (json.JSONDecodeError, Exception):
|
if not isinstance(scene, dict):
|
||||||
|
logger.warning("sd_prompt: LLM returned non-dict: %.100s", raw)
|
||||||
|
return None, None
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning("sd_prompt failed: %s raw=%.200s", e, locals().get("raw", ""))
|
||||||
return None, None
|
return None, None
|
||||||
|
|
||||||
|
try:
|
||||||
|
outfit_list = json.loads(outfit_json or "[]")
|
||||||
|
outfit_tags = ", ".join(outfit_list) if isinstance(outfit_list, list) else ""
|
||||||
|
except Exception:
|
||||||
|
outfit_tags = ""
|
||||||
|
|
||||||
|
positive = build_positive_prompt(scene, persona, outfit_tags)
|
||||||
|
|
||||||
|
if _is_pony():
|
||||||
|
negative = PONY_NEGATIVE
|
||||||
|
elif _is_anima():
|
||||||
|
negative = ANIMA_NEGATIVE
|
||||||
|
else:
|
||||||
|
negative = "low quality, blurry, bad anatomy, watermark, text"
|
||||||
|
|
||||||
positive = build_positive_prompt(scene, persona)
|
|
||||||
is_pony = SD_CHECKPOINT in PONY_CHECKPOINTS
|
|
||||||
negative = PONY_NEGATIVE if is_pony else "low quality, blurry, bad anatomy, watermark, text"
|
|
||||||
if scene.get("shot_type") == "first_person_pov":
|
if scene.get("shot_type") == "first_person_pov":
|
||||||
negative += ", third person, over the shoulder"
|
negative += ", third person, over the shoulder"
|
||||||
|
|
||||||
full = positive
|
full = positive + f"\n\nNegative prompt: {negative}"
|
||||||
if negative:
|
|
||||||
full += f"\n\nNegative prompt: {negative}"
|
|
||||||
return full, negative
|
return full, negative
|
||||||
|
|||||||
+50
-16
@@ -16,13 +16,26 @@ SD_STEPS = int(os.getenv("SD_STEPS", "28"))
|
|||||||
SD_CFG = float(os.getenv("SD_CFG", "7"))
|
SD_CFG = float(os.getenv("SD_CFG", "7"))
|
||||||
SD_SAMPLER = os.getenv("SD_SAMPLER", "euler")
|
SD_SAMPLER = os.getenv("SD_SAMPLER", "euler")
|
||||||
SD_SCHEDULER = os.getenv("SD_SCHEDULER", "normal")
|
SD_SCHEDULER = os.getenv("SD_SCHEDULER", "normal")
|
||||||
SD_CHECKPOINT = os.getenv("SD_CHECKPOINT", "NetaYumev35_pretrained_all_in_one.safetensors")
|
SD_CHECKPOINT = os.getenv("SD_CHECKPOINT", "")
|
||||||
SD_DEFAULT_NEGATIVE = os.getenv(
|
SD_DEFAULT_NEGATIVE = os.getenv(
|
||||||
"SD_DEFAULT_NEGATIVE",
|
"SD_DEFAULT_NEGATIVE",
|
||||||
"low quality, worst quality, blurry, bad anatomy, watermark, text",
|
"low quality, worst quality, blurry, bad anatomy, watermark, text",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Anima split-model settings
|
||||||
|
SD_UNET = os.getenv("SD_UNET", "anima-preview3-base.safetensors")
|
||||||
|
SD_CLIP = os.getenv("SD_CLIP", "qwen_3_06b_base.safetensors")
|
||||||
|
SD_VAE = os.getenv("SD_VAE", "qwen_image_vae.safetensors")
|
||||||
|
|
||||||
IMAGES_DIR = Path(os.getenv("IMAGES_DIR", "static/images"))
|
IMAGES_DIR = Path(os.getenv("IMAGES_DIR", "static/images"))
|
||||||
|
|
||||||
|
ANIMA_CHECKPOINTS = {"anima-preview3-base.safetensors"}
|
||||||
|
PONY_CHECKPOINTS = {"ponyDiffusionV6XL_v6StartWithThisOne.safetensors"}
|
||||||
|
|
||||||
|
|
||||||
|
def _use_anima() -> bool:
|
||||||
|
return bool(SD_UNET) and not SD_CHECKPOINT
|
||||||
|
|
||||||
|
|
||||||
def split_prompt_and_negative(full_prompt: str) -> tuple[str, str]:
|
def split_prompt_and_negative(full_prompt: str) -> tuple[str, str]:
|
||||||
if "\n\nNegative prompt:" in full_prompt:
|
if "\n\nNegative prompt:" in full_prompt:
|
||||||
@@ -32,7 +45,30 @@ def split_prompt_and_negative(full_prompt: str) -> tuple[str, str]:
|
|||||||
|
|
||||||
|
|
||||||
def _build_workflow(positive: str, negative: str) -> dict:
|
def _build_workflow(positive: str, negative: str) -> dict:
|
||||||
"""Minimal KSampler workflow for ComfyUI API."""
|
seed = int(uuid.uuid4().int % 2**32)
|
||||||
|
if _use_anima():
|
||||||
|
return {
|
||||||
|
"44": {"class_type": "UNETLoader", "inputs": {"unet_name": SD_UNET, "weight_dtype": "default"}},
|
||||||
|
"45": {"class_type": "CLIPLoader", "inputs": {"clip_name": SD_CLIP, "type": "stable_diffusion", "device": "default"}},
|
||||||
|
"15": {"class_type": "VAELoader", "inputs": {"vae_name": SD_VAE}},
|
||||||
|
"28": {"class_type": "EmptyLatentImage", "inputs": {"width": 1024, "height": 1024, "batch_size": 1}},
|
||||||
|
"11": {"class_type": "CLIPTextEncode", "inputs": {"text": positive, "clip": ["45", 0]}},
|
||||||
|
"12": {"class_type": "CLIPTextEncode", "inputs": {"text": negative, "clip": ["45", 0]}},
|
||||||
|
"19": {
|
||||||
|
"class_type": "KSampler",
|
||||||
|
"inputs": {
|
||||||
|
"model": ["44", 0], "positive": ["11", 0], "negative": ["12", 0],
|
||||||
|
"latent_image": ["28", 0], "seed": seed,
|
||||||
|
"steps": SD_STEPS, "cfg": SD_CFG,
|
||||||
|
"sampler_name": os.getenv("SD_SAMPLER", "er_sde"),
|
||||||
|
"scheduler": os.getenv("SD_SCHEDULER", "simple"),
|
||||||
|
"denoise": 1.0,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"8": {"class_type": "VAEDecode", "inputs": {"samples": ["19", 0], "vae": ["15", 0]}},
|
||||||
|
"9": {"class_type": "SaveImage", "inputs": {"filename_prefix": "chatbot", "images": ["8", 0]}},
|
||||||
|
}
|
||||||
|
# Standard checkpoint workflow (Pony / SDXL)
|
||||||
return {
|
return {
|
||||||
"4": {"class_type": "CheckpointLoaderSimple", "inputs": {"ckpt_name": SD_CHECKPOINT}},
|
"4": {"class_type": "CheckpointLoaderSimple", "inputs": {"ckpt_name": SD_CHECKPOINT}},
|
||||||
"5": {"class_type": "EmptyLatentImage", "inputs": {"width": 832, "height": 1216, "batch_size": 1}},
|
"5": {"class_type": "EmptyLatentImage", "inputs": {"width": 832, "height": 1216, "batch_size": 1}},
|
||||||
@@ -43,15 +79,10 @@ def _build_workflow(positive: str, negative: str) -> dict:
|
|||||||
"10": {
|
"10": {
|
||||||
"class_type": "KSampler",
|
"class_type": "KSampler",
|
||||||
"inputs": {
|
"inputs": {
|
||||||
"model": ["4", 0],
|
"model": ["4", 0], "positive": ["6", 0], "negative": ["7", 0],
|
||||||
"positive": ["6", 0],
|
"latent_image": ["5", 0], "seed": seed,
|
||||||
"negative": ["7", 0],
|
"steps": SD_STEPS, "cfg": SD_CFG,
|
||||||
"latent_image": ["5", 0],
|
"sampler_name": SD_SAMPLER, "scheduler": SD_SCHEDULER,
|
||||||
"seed": int(uuid.uuid4().int % 2**32),
|
|
||||||
"steps": SD_STEPS,
|
|
||||||
"cfg": SD_CFG,
|
|
||||||
"sampler_name": SD_SAMPLER,
|
|
||||||
"scheduler": SD_SCHEDULER,
|
|
||||||
"denoise": 1.0,
|
"denoise": 1.0,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -74,7 +105,6 @@ async def txt2img(prompt: str, negative_prompt: str | None = None) -> tuple[byte
|
|||||||
|
|
||||||
logger.info("ComfyUI request → %s prompt: %.120s", SD_BASE_URL, prompt)
|
logger.info("ComfyUI request → %s prompt: %.120s", SD_BASE_URL, prompt)
|
||||||
async with httpx.AsyncClient(timeout=300) as client:
|
async with httpx.AsyncClient(timeout=300) as client:
|
||||||
# queue the prompt
|
|
||||||
resp = await client.post(
|
resp = await client.post(
|
||||||
f"{SD_BASE_URL}/prompt",
|
f"{SD_BASE_URL}/prompt",
|
||||||
json={"prompt": workflow, "client_id": client_id},
|
json={"prompt": workflow, "client_id": client_id},
|
||||||
@@ -83,14 +113,17 @@ async def txt2img(prompt: str, negative_prompt: str | None = None) -> tuple[byte
|
|||||||
prompt_id = resp.json()["prompt_id"]
|
prompt_id = resp.json()["prompt_id"]
|
||||||
logger.info("ComfyUI queued prompt_id=%s", prompt_id)
|
logger.info("ComfyUI queued prompt_id=%s", prompt_id)
|
||||||
|
|
||||||
# poll until done
|
|
||||||
for _ in range(300):
|
for _ in range(300):
|
||||||
await asyncio.sleep(1)
|
await asyncio.sleep(1)
|
||||||
hist = await client.get(f"{SD_BASE_URL}/history/{prompt_id}")
|
hist = await client.get(f"{SD_BASE_URL}/history/{prompt_id}")
|
||||||
data = hist.json()
|
data = hist.json()
|
||||||
if prompt_id in data:
|
if prompt_id in data:
|
||||||
outputs = data[prompt_id]["outputs"]
|
entry = data[prompt_id]
|
||||||
# find first image output
|
# Log any errors from ComfyUI
|
||||||
|
if entry.get("status", {}).get("status_str") == "error":
|
||||||
|
msgs = entry.get("status", {}).get("messages", [])
|
||||||
|
logger.error("ComfyUI workflow error: %s", msgs)
|
||||||
|
outputs = entry.get("outputs", {})
|
||||||
for node_output in outputs.values():
|
for node_output in outputs.values():
|
||||||
if "images" in node_output:
|
if "images" in node_output:
|
||||||
img_info = node_output["images"][0]
|
img_info = node_output["images"][0]
|
||||||
@@ -100,12 +133,13 @@ async def txt2img(prompt: str, negative_prompt: str | None = None) -> tuple[byte
|
|||||||
)
|
)
|
||||||
img_resp.raise_for_status()
|
img_resp.raise_for_status()
|
||||||
image_bytes = img_resp.content
|
image_bytes = img_resp.content
|
||||||
|
|
||||||
IMAGES_DIR.mkdir(parents=True, exist_ok=True)
|
IMAGES_DIR.mkdir(parents=True, exist_ok=True)
|
||||||
filename = f"{uuid.uuid4().hex}.png"
|
filename = f"{uuid.uuid4().hex}.png"
|
||||||
(IMAGES_DIR / filename).write_bytes(image_bytes)
|
(IMAGES_DIR / filename).write_bytes(image_bytes)
|
||||||
logger.info("ComfyUI done → saved %s", filename)
|
logger.info("ComfyUI done → saved %s", filename)
|
||||||
return image_bytes, f"images/{filename}"
|
return image_bytes, f"images/{filename}"
|
||||||
|
logger.error("ComfyUI no image output. status=%s outputs_keys=%s",
|
||||||
|
entry.get("status"), list(outputs.keys()))
|
||||||
break
|
break
|
||||||
|
|
||||||
raise RuntimeError("ComfyUI generation timed out or produced no output")
|
raise RuntimeError("ComfyUI generation timed out or produced no output")
|
||||||
|
|||||||
Binary file not shown.
|
After Width: | Height: | Size: 77 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 1.4 MiB |
Binary file not shown.
|
After Width: | Height: | Size: 1.4 MiB |
+34
-1
@@ -136,11 +136,13 @@ header h1 { font-size: 1.1rem; color: #e94560; }
|
|||||||
.system-blob-header {
|
.system-blob-header {
|
||||||
display: flex;
|
display: flex;
|
||||||
align-items: center;
|
align-items: center;
|
||||||
|
gap: 6px;
|
||||||
justify-content: space-between;
|
justify-content: space-between;
|
||||||
color: #888;
|
color: #888;
|
||||||
font-size: 0.8rem;
|
font-size: 0.8rem;
|
||||||
margin-bottom: 6px;
|
margin-bottom: 6px;
|
||||||
}
|
}
|
||||||
|
.system-blob-header span { flex: 1; }
|
||||||
.system-blob-header button {
|
.system-blob-header button {
|
||||||
background: transparent;
|
background: transparent;
|
||||||
border: 1px solid #0f3460;
|
border: 1px solid #0f3460;
|
||||||
@@ -150,6 +152,10 @@ header h1 { font-size: 1.1rem; color: #e94560; }
|
|||||||
cursor: pointer;
|
cursor: pointer;
|
||||||
}
|
}
|
||||||
.system-blob-header button:hover { border-color: #e94560; color: #e94560; }
|
.system-blob-header button:hover { border-color: #e94560; color: #e94560; }
|
||||||
|
#systemBlobRefresh { font-size: 1rem; padding: 2px 8px; }
|
||||||
|
#systemBlobRefresh.spinning { animation: spin 0.6s linear infinite; }
|
||||||
|
@keyframes spin { to { transform: rotate(360deg); } }
|
||||||
|
.blob-changed { background: rgba(255, 200, 50, 0.15); border-radius: 3px; transition: background 2s ease; }
|
||||||
.system-blob-content {
|
.system-blob-content {
|
||||||
white-space: pre-wrap;
|
white-space: pre-wrap;
|
||||||
word-break: break-word;
|
word-break: break-word;
|
||||||
@@ -278,6 +284,31 @@ header h1 { font-size: 1.1rem; color: #e94560; }
|
|||||||
.translate-btn:disabled { opacity: 0.5; cursor: default; }
|
.translate-btn:disabled { opacity: 0.5; cursor: default; }
|
||||||
|
|
||||||
.chat-image { margin-top: 8px; max-width: 100%; border-radius: 8px; border: 1px solid #0f3460; }
|
.chat-image { margin-top: 8px; max-width: 100%; border-radius: 8px; border: 1px solid #0f3460; }
|
||||||
|
|
||||||
|
.image-generating {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 10px;
|
||||||
|
margin-top: 8px;
|
||||||
|
padding: 12px 14px;
|
||||||
|
border-radius: 8px;
|
||||||
|
border: 1px dashed #533483;
|
||||||
|
background: rgba(15, 52, 96, 0.45);
|
||||||
|
color: #bbb;
|
||||||
|
font-size: 0.9em;
|
||||||
|
}
|
||||||
|
.image-generating-spinner {
|
||||||
|
width: 18px;
|
||||||
|
height: 18px;
|
||||||
|
border: 2px solid #0f3460;
|
||||||
|
border-top-color: #e94560;
|
||||||
|
border-radius: 50%;
|
||||||
|
animation: image-spin 0.75s linear infinite;
|
||||||
|
flex-shrink: 0;
|
||||||
|
}
|
||||||
|
@keyframes image-spin { to { transform: rotate(360deg); } }
|
||||||
|
|
||||||
|
.gen-image-btn:disabled { opacity: 0.6; cursor: wait; }
|
||||||
.image-error { margin-top: 6px; font-size: 0.75rem; color: #888; }
|
.image-error { margin-top: 6px; font-size: 0.75rem; color: #888; }
|
||||||
|
|
||||||
.choice-row {
|
.choice-row {
|
||||||
@@ -407,11 +438,13 @@ textarea:focus { border-color: #e94560; }
|
|||||||
.wizard-nav-btn:disabled { opacity: 0.45; cursor: not-allowed; }
|
.wizard-nav-btn:disabled { opacity: 0.45; cursor: not-allowed; }
|
||||||
.modal h2 { font-size: 1.1rem; color: #e94560; }
|
.modal h2 { font-size: 1.1rem; color: #e94560; }
|
||||||
.modal label { display: flex; flex-direction: column; gap: 4px; font-size: 0.8rem; color: #888; }
|
.modal label { display: flex; flex-direction: column; gap: 4px; font-size: 0.8rem; color: #888; }
|
||||||
.modal input, .modal textarea {
|
.modal input, .modal textarea, .modal select {
|
||||||
background: #1a1a2e; border: 1px solid #0f3460;
|
background: #1a1a2e; border: 1px solid #0f3460;
|
||||||
border-radius: 8px; color: #e0e0e0;
|
border-radius: 8px; color: #e0e0e0;
|
||||||
padding: 8px 10px; outline: none; font-family: inherit;
|
padding: 8px 10px; outline: none; font-family: inherit;
|
||||||
}
|
}
|
||||||
|
.modal select { cursor: pointer; }
|
||||||
|
.modal select[size] { min-height: 80px; }
|
||||||
.modal-buttons { display: flex; gap: 8px; justify-content: flex-end; }
|
.modal-buttons { display: flex; gap: 8px; justify-content: flex-end; }
|
||||||
.modal-wizard-footer { justify-content: space-between; align-items: center; }
|
.modal-wizard-footer { justify-content: space-between; align-items: center; }
|
||||||
.modal-buttons button { padding: 8px 18px; border-radius: 8px; border: none; cursor: pointer; }
|
.modal-buttons button { padding: 8px 18px; border-radius: 8px; border: none; cursor: pointer; }
|
||||||
|
|||||||
+51
-6
@@ -35,6 +35,7 @@
|
|||||||
<div class="system-blob" id="systemBlob">
|
<div class="system-blob" id="systemBlob">
|
||||||
<div class="system-blob-header">
|
<div class="system-blob-header">
|
||||||
<span>System</span>
|
<span>System</span>
|
||||||
|
<button type="button" id="systemBlobRefresh" title="Обновить">↻</button>
|
||||||
<button type="button" id="systemBlobToggle">Скрыть</button>
|
<button type="button" id="systemBlobToggle">Скрыть</button>
|
||||||
</div>
|
</div>
|
||||||
<pre class="system-blob-content" id="systemBlobContent">—</pre>
|
<pre class="system-blob-content" id="systemBlobContent">—</pre>
|
||||||
@@ -126,20 +127,53 @@
|
|||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div class="modal-overlay" id="cardModalOverlay">
|
<div class="modal-overlay" id="cardModalOverlay">
|
||||||
<div class="modal">
|
<div class="modal modal-wizard" style="max-width:520px">
|
||||||
<h2>📥 Импорт карточки (chub.io / V2)</h2>
|
<div class="modal-wizard-header">
|
||||||
<label>Файл JSON или PNG
|
<h2>📥 Импорт карточки</h2>
|
||||||
|
<div class="wizard-steps">
|
||||||
|
<span class="wizard-step-dot active" data-step="1">1</span>
|
||||||
|
<span class="wizard-step-line"></span>
|
||||||
|
<span class="wizard-step-dot" data-step="2">2</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="modal-wizard-body">
|
||||||
|
<div class="wizard-page active" data-step="1">
|
||||||
|
<p class="wizard-page-title">Файл</p>
|
||||||
|
<label>JSON или PNG (chub.io / V2)
|
||||||
<input type="file" id="cardFile" accept=".json,.png">
|
<input type="file" id="cardFile" accept=".json,.png">
|
||||||
</label>
|
</label>
|
||||||
|
<p class="wizard-hint" id="cardPreviewHint"></p>
|
||||||
<label>LoRA
|
<label>LoRA
|
||||||
<input type="text" id="cardLora" placeholder="CharacterLoRA">
|
<input type="text" id="cardLora" placeholder="CharacterLoRA">
|
||||||
</label>
|
</label>
|
||||||
<label>Вес LoRA
|
<label>Вес LoRA
|
||||||
<input type="number" id="cardLoraWeight" value="0.8" min="0" max="2" step="0.1">
|
<input type="number" id="cardLoraWeight" value="0.8" min="0" max="2" step="0.1">
|
||||||
</label>
|
</label>
|
||||||
<div class="modal-buttons">
|
</div>
|
||||||
|
<div class="wizard-page" data-step="2">
|
||||||
|
<p class="wizard-page-title">Проверь и отредактируй</p>
|
||||||
|
<label>Имя <input type="text" id="impCardName"></label>
|
||||||
|
<label>Описание <textarea id="impCardDescription" rows="3"></textarea></label>
|
||||||
|
<label>Личность <textarea id="impCardPersonality" rows="2"></textarea></label>
|
||||||
|
<label>Сценарий <textarea id="impCardScenario" rows="2"></textarea></label>
|
||||||
|
<label>Первое сообщение
|
||||||
|
<select id="impCardGreetingSelect"></select>
|
||||||
|
</label>
|
||||||
|
<label>Текст первого сообщения
|
||||||
|
<textarea id="impCardFirstMes" rows="4"></textarea>
|
||||||
|
</label>
|
||||||
|
<p class="wizard-hint hidden" id="impCardAltHint"></p>
|
||||||
|
<label>Пример диалога <textarea id="impCardMesExample" rows="2"></textarea></label>
|
||||||
|
<label>Теги внешности (SD) <input type="text" id="impCardAppearance"></label>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="modal-buttons modal-wizard-footer">
|
||||||
<button id="cardModalCancel" type="button">Отмена</button>
|
<button id="cardModalCancel" type="button">Отмена</button>
|
||||||
<button id="cardModalImport" type="button">Импорт</button>
|
<div class="wizard-nav">
|
||||||
|
<button id="cardModalPrev" type="button" class="wizard-nav-btn hidden">← Назад</button>
|
||||||
|
<button id="cardModalNext" type="button" class="wizard-nav-btn">Далее →</button>
|
||||||
|
<button id="cardModalImport" type="button" class="hidden" style="background:#e94560;color:white">Импорт</button>
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
@@ -156,6 +190,9 @@
|
|||||||
<label>Личность <textarea id="editPersonality" rows="3"></textarea></label>
|
<label>Личность <textarea id="editPersonality" rows="3"></textarea></label>
|
||||||
<label>Сценарий <textarea id="editScenario" rows="3"></textarea></label>
|
<label>Сценарий <textarea id="editScenario" rows="3"></textarea></label>
|
||||||
<label>Первое сообщение <textarea id="editFirstMes" rows="3"></textarea></label>
|
<label>Первое сообщение <textarea id="editFirstMes" rows="3"></textarea></label>
|
||||||
|
<label class="hidden" id="editCardAltBlock">Альтернативные приветствия (из карточки)
|
||||||
|
<select id="editCardGreetingSelect" size="4"></select>
|
||||||
|
</label>
|
||||||
<label>Пример диалога <textarea id="editMesExample" rows="3"></textarea></label>
|
<label>Пример диалога <textarea id="editMesExample" rows="3"></textarea></label>
|
||||||
<label>Теги внешности (SD) <input type="text" id="editAppearance" placeholder="silver hair, yellow eyes, wolf ears, black cloak"></label>
|
<label>Теги внешности (SD) <input type="text" id="editAppearance" placeholder="silver hair, yellow eyes, wolf ears, black cloak"></label>
|
||||||
<label>LoRA <input type="text" id="editLora" placeholder="CharacterLoRA"></label>
|
<label>LoRA <input type="text" id="editLora" placeholder="CharacterLoRA"></label>
|
||||||
@@ -226,6 +263,14 @@
|
|||||||
<label>Название чата
|
<label>Название чата
|
||||||
<input type="text" id="newChatTitle" placeholder="Оставь пустым — сгенерируем автоматически">
|
<input type="text" id="newChatTitle" placeholder="Оставь пустым — сгенерируем автоматически">
|
||||||
</label>
|
</label>
|
||||||
|
<div id="newChatGreetingBlock" class="hidden" style="margin-top:12px">
|
||||||
|
<label>Первое сообщение
|
||||||
|
<select id="newChatGreetingSelect"></select>
|
||||||
|
</label>
|
||||||
|
<label>Текст (можно отредактировать)
|
||||||
|
<textarea id="newChatGreetingText" rows="3"></textarea>
|
||||||
|
</label>
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
<div id="newChatRpgStep" class="hidden">
|
<div id="newChatRpgStep" class="hidden">
|
||||||
<p class="wizard-page-title">Жанры и настройки RPG</p>
|
<p class="wizard-page-title">Жанры и настройки RPG</p>
|
||||||
@@ -301,6 +346,6 @@
|
|||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<script type="module" src="/static/js/app.js"></script>
|
<script type="module" src="/static/js/app.js?v=4"></script>
|
||||||
</body>
|
</body>
|
||||||
</html>
|
</html>
|
||||||
|
|||||||
+3
-3
@@ -1,7 +1,7 @@
|
|||||||
import { toggleSidebar, dom } from './state.js';
|
import { toggleSidebar, dom } from './state.js';
|
||||||
import {
|
import { initSessions } from './sessions.js';
|
||||||
initSessions, openNewChatWizard, initNewChatWizard, initChatSettings, openChatSettings,
|
import { openNewChatWizard, initNewChatWizard } from './newChatWizard.js';
|
||||||
} from './sessions.js';
|
import { openChatSettings, initChatSettings } from './chatSettings.js';
|
||||||
import { loadPersonas, initPersonaModals } from './personas.js';
|
import { loadPersonas, initPersonaModals } from './personas.js';
|
||||||
import { sendMessage, clearHistory } from './chat.js';
|
import { sendMessage, clearHistory } from './chat.js';
|
||||||
|
|
||||||
|
|||||||
+109
-138
@@ -1,13 +1,11 @@
|
|||||||
import { sessionId, currentPersona, dom } from './state.js';
|
import { sessionId, currentPersona, dom } from './state.js';
|
||||||
import { parseImagePromptFromContent, copyToClipboard } from './utils.js';
|
import { parseImagePromptFromContent, copyToClipboard } from './utils.js';
|
||||||
|
|
||||||
export async function initChat() {
|
export async function initChat(options = {}) {
|
||||||
if (!sessionId || !currentPersona) return;
|
if (!sessionId || !currentPersona) return;
|
||||||
const res = await fetch('/chat/init', {
|
const payload = { message: '', session_id: sessionId, persona_id: currentPersona };
|
||||||
method: 'POST',
|
if (options.first_mes_override?.trim()) payload.first_mes_override = options.first_mes_override.trim();
|
||||||
headers: { 'Content-Type': 'application/json' },
|
const res = await fetch('/chat/init', { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify(payload) });
|
||||||
body: JSON.stringify({ message: '', session_id: sessionId, persona_id: currentPersona }),
|
|
||||||
});
|
|
||||||
if (!res.ok) return;
|
if (!res.ok) return;
|
||||||
const data = await res.json();
|
const data = await res.json();
|
||||||
if (data.first_mes) addMessage('assistant', data.first_mes);
|
if (data.first_mes) addMessage('assistant', data.first_mes);
|
||||||
@@ -21,7 +19,6 @@ export function updateEmptyState() {
|
|||||||
export function createImagePromptBlock(promptText) {
|
export function createImagePromptBlock(promptText) {
|
||||||
const block = document.createElement('div');
|
const block = document.createElement('div');
|
||||||
block.className = 'image-prompt-block';
|
block.className = 'image-prompt-block';
|
||||||
|
|
||||||
const header = document.createElement('div');
|
const header = document.createElement('div');
|
||||||
header.className = 'image-prompt-header';
|
header.className = 'image-prompt-header';
|
||||||
header.innerHTML = '<span>🎨 SD prompt</span>';
|
header.innerHTML = '<span>🎨 SD prompt</span>';
|
||||||
@@ -37,17 +34,43 @@ export function createImagePromptBlock(promptText) {
|
|||||||
});
|
});
|
||||||
header.appendChild(copyBtn);
|
header.appendChild(copyBtn);
|
||||||
|
|
||||||
const genBtn = document.createElement('button');
|
const regenBtn = document.createElement('button');
|
||||||
genBtn.type = 'button';
|
regenBtn.type = 'button';
|
||||||
genBtn.className = 'gen-image-btn';
|
regenBtn.className = 'copy-prompt-btn';
|
||||||
genBtn.textContent = '🖼 Генерировать';
|
regenBtn.textContent = '🖼 Перегенерировать';
|
||||||
genBtn.addEventListener('click', () => generateImageViaA1111(promptText, block));
|
regenBtn.addEventListener('click', async () => {
|
||||||
header.appendChild(genBtn);
|
const wrapper = block.parentElement;
|
||||||
|
regenBtn.disabled = true;
|
||||||
|
regenBtn.textContent = '⏳…';
|
||||||
|
wrapper?.querySelector('.chat-image')?.remove();
|
||||||
|
wrapper?.querySelector('.image-error')?.remove();
|
||||||
|
showImageGenerating(wrapper);
|
||||||
|
try {
|
||||||
|
const res = await fetch('/images/generate', {
|
||||||
|
method: 'POST',
|
||||||
|
headers: { 'Content-Type': 'application/json' },
|
||||||
|
body: JSON.stringify({ session_id: sessionId, prompt: promptText }),
|
||||||
|
});
|
||||||
|
const data = await res.json();
|
||||||
|
if (!res.ok) throw new Error(data.detail || res.statusText);
|
||||||
|
removeImageGenerating(wrapper);
|
||||||
|
appendChatImage(wrapper, data.image_path);
|
||||||
|
} catch (e) {
|
||||||
|
removeImageGenerating(wrapper);
|
||||||
|
const err = document.createElement('div');
|
||||||
|
err.className = 'image-error';
|
||||||
|
err.textContent = '🖼 ' + e.message;
|
||||||
|
wrapper?.appendChild(err);
|
||||||
|
} finally {
|
||||||
|
regenBtn.disabled = false;
|
||||||
|
regenBtn.textContent = '🖼 Перегенерировать';
|
||||||
|
}
|
||||||
|
});
|
||||||
|
header.appendChild(regenBtn);
|
||||||
|
|
||||||
const textEl = document.createElement('span');
|
const textEl = document.createElement('span');
|
||||||
textEl.className = 'prompt-text';
|
textEl.className = 'prompt-text';
|
||||||
textEl.textContent = promptText;
|
textEl.textContent = promptText;
|
||||||
|
|
||||||
block.appendChild(header);
|
block.appendChild(header);
|
||||||
block.appendChild(textEl);
|
block.appendChild(textEl);
|
||||||
return block;
|
return block;
|
||||||
@@ -60,37 +83,38 @@ const OUTCOME_CLASS = {
|
|||||||
'critical success': 'outcome-crit-success',
|
'critical success': 'outcome-crit-success',
|
||||||
};
|
};
|
||||||
|
|
||||||
function renderNarratorMessage(narrator) {
|
function buildNarratorEl(narrator) {
|
||||||
// narrator = { roll, outcome, text }
|
|
||||||
const wrapper = document.createElement('div');
|
const wrapper = document.createElement('div');
|
||||||
wrapper.className = 'message narrator';
|
wrapper.className = 'message narrator';
|
||||||
|
|
||||||
const label = document.createElement('div');
|
const label = document.createElement('div');
|
||||||
label.className = 'label';
|
label.className = 'label';
|
||||||
label.textContent = '📖 Рассказчик';
|
label.textContent = '📖 Рассказчик';
|
||||||
wrapper.appendChild(label);
|
wrapper.appendChild(label);
|
||||||
|
|
||||||
const bubble = document.createElement('div');
|
const bubble = document.createElement('div');
|
||||||
bubble.className = 'bubble';
|
bubble.className = 'bubble';
|
||||||
|
if (narrator.roll != null) {
|
||||||
const diceBlock = document.createElement('div');
|
const diceBlock = document.createElement('div');
|
||||||
diceBlock.className = `dice-block ${OUTCOME_CLASS[narrator.outcome] || ''}`;
|
diceBlock.className = `dice-block ${OUTCOME_CLASS[narrator.outcome] || ''}`;
|
||||||
diceBlock.innerHTML = `<span class="dice-icon">🎲</span><span class="dice-roll">${narrator.roll}</span><span class="dice-outcome">${narrator.outcome}</span>`;
|
diceBlock.innerHTML = `<span class="dice-icon">🎲</span><span class="dice-roll">${narrator.roll}</span><span class="dice-outcome">${narrator.outcome}</span>`;
|
||||||
bubble.appendChild(diceBlock);
|
bubble.appendChild(diceBlock);
|
||||||
|
}
|
||||||
const textEl = document.createElement('div');
|
const textEl = document.createElement('div');
|
||||||
textEl.className = 'narrator-text';
|
textEl.className = 'narrator-text';
|
||||||
textEl.textContent = narrator.text;
|
textEl.textContent = narrator.text;
|
||||||
bubble.appendChild(textEl);
|
bubble.appendChild(textEl);
|
||||||
|
|
||||||
wrapper.appendChild(bubble);
|
wrapper.appendChild(bubble);
|
||||||
dom.messagesEl.appendChild(wrapper);
|
|
||||||
dom.messagesEl.scrollTop = dom.messagesEl.scrollHeight;
|
|
||||||
return wrapper;
|
return wrapper;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function renderNarratorMessage(narrator) {
|
||||||
|
const el = buildNarratorEl(narrator);
|
||||||
|
dom.messagesEl.appendChild(el);
|
||||||
|
dom.messagesEl.scrollTop = dom.messagesEl.scrollHeight;
|
||||||
|
return el;
|
||||||
|
}
|
||||||
|
|
||||||
function renderChoices(wrapper, choices) {
|
function renderChoices(wrapper, choices) {
|
||||||
if (!choices || !choices.length) return;
|
if (!choices?.length) return;
|
||||||
const row = document.createElement('div');
|
const row = document.createElement('div');
|
||||||
row.className = 'choice-row';
|
row.className = 'choice-row';
|
||||||
for (const c of choices) {
|
for (const c of choices) {
|
||||||
@@ -98,20 +122,17 @@ function renderChoices(wrapper, choices) {
|
|||||||
btn.type = 'button';
|
btn.type = 'button';
|
||||||
btn.className = 'choice-btn';
|
btn.className = 'choice-btn';
|
||||||
btn.textContent = c.label;
|
btn.textContent = c.label;
|
||||||
btn.addEventListener('click', () => {
|
btn.addEventListener('click', () => sendMessage(c.label, true));
|
||||||
sendMessage(c.label, true);
|
|
||||||
});
|
|
||||||
row.appendChild(btn);
|
row.appendChild(btn);
|
||||||
}
|
}
|
||||||
wrapper.appendChild(row);
|
wrapper.appendChild(row);
|
||||||
}
|
}
|
||||||
|
|
||||||
function renderDebugBlocks(wrapper, blocks) {
|
function renderDebugBlocks(wrapper, blocks) {
|
||||||
if (!blocks || !blocks.length) return;
|
if (!blocks?.length) return;
|
||||||
for (const b of blocks) {
|
for (const b of blocks) {
|
||||||
if (!b?.text) continue;
|
if (!b?.text) continue;
|
||||||
if (b.type === 'narrator_injection') {
|
if (b.type === 'narrator_injection') {
|
||||||
// Show beat injections as narrator bubbles (no dice)
|
|
||||||
const w = document.createElement('div');
|
const w = document.createElement('div');
|
||||||
w.className = 'message narrator';
|
w.className = 'message narrator';
|
||||||
const lbl = document.createElement('div');
|
const lbl = document.createElement('div');
|
||||||
@@ -124,7 +145,6 @@ function renderDebugBlocks(wrapper, blocks) {
|
|||||||
w.appendChild(bub);
|
w.appendChild(bub);
|
||||||
dom.messagesEl.appendChild(w);
|
dom.messagesEl.appendChild(w);
|
||||||
}
|
}
|
||||||
// facts/status_quo/plot_arc — silently skip (debug only, not shown to user)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -149,31 +169,6 @@ export function updateAffinityDisplay(affinity) {
|
|||||||
el.className = `affinity-display ${affinity > 5 ? 'affinity-high' : affinity < -3 ? 'affinity-low' : ''}`;
|
el.className = `affinity-display ${affinity > 5 ? 'affinity-high' : affinity < -3 ? 'affinity-low' : ''}`;
|
||||||
}
|
}
|
||||||
|
|
||||||
async function generateImageViaA1111(promptText, block) {
|
|
||||||
block.parentElement.querySelector('.chat-image')?.remove();
|
|
||||||
block.parentElement.querySelector('.image-error')?.remove();
|
|
||||||
|
|
||||||
try {
|
|
||||||
const res = await fetch('/images/generate', {
|
|
||||||
method: 'POST',
|
|
||||||
headers: { 'Content-Type': 'application/json' },
|
|
||||||
body: JSON.stringify({ session_id: sessionId, prompt: promptText }),
|
|
||||||
});
|
|
||||||
const data = await res.json();
|
|
||||||
if (!res.ok) throw new Error(data.detail || res.statusText);
|
|
||||||
|
|
||||||
const img = document.createElement('img');
|
|
||||||
img.className = 'chat-image';
|
|
||||||
img.src = data.image_path;
|
|
||||||
block.parentElement.appendChild(img);
|
|
||||||
} catch (e) {
|
|
||||||
const err = document.createElement('div');
|
|
||||||
err.className = 'image-error';
|
|
||||||
err.textContent = '🖼 ' + e.message;
|
|
||||||
block.parentElement.appendChild(err);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
export function appendChatImage(wrapper, imagePath) {
|
export function appendChatImage(wrapper, imagePath) {
|
||||||
if (!imagePath) return;
|
if (!imagePath) return;
|
||||||
const img = document.createElement('img');
|
const img = document.createElement('img');
|
||||||
@@ -182,20 +177,31 @@ export function appendChatImage(wrapper, imagePath) {
|
|||||||
wrapper.appendChild(img);
|
wrapper.appendChild(img);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export function showImageGenerating(wrapper) {
|
||||||
|
if (!wrapper || wrapper.querySelector('.image-generating')) return;
|
||||||
|
const el = document.createElement('div');
|
||||||
|
el.className = 'image-generating';
|
||||||
|
el.setAttribute('role', 'status');
|
||||||
|
el.innerHTML = '<span class="image-generating-spinner" aria-hidden="true"></span><span class="image-generating-text">Генерация изображения в ComfyUI…</span>';
|
||||||
|
wrapper.appendChild(el);
|
||||||
|
dom.messagesEl.scrollTop = dom.messagesEl.scrollHeight;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function removeImageGenerating(wrapper) {
|
||||||
|
wrapper?.querySelector('.image-generating')?.remove();
|
||||||
|
}
|
||||||
|
|
||||||
function attachMessageActions(wrapper, messageId, role) {
|
function attachMessageActions(wrapper, messageId, role) {
|
||||||
if (!messageId) return;
|
if (!messageId) return;
|
||||||
wrapper.dataset.messageId = String(messageId);
|
wrapper.dataset.messageId = String(messageId);
|
||||||
|
|
||||||
const actions = document.createElement('div');
|
const actions = document.createElement('div');
|
||||||
actions.className = 'message-actions';
|
actions.className = 'message-actions';
|
||||||
|
|
||||||
const editBtn = document.createElement('button');
|
const editBtn = document.createElement('button');
|
||||||
editBtn.type = 'button';
|
editBtn.type = 'button';
|
||||||
editBtn.textContent = '✏️';
|
editBtn.textContent = '✏️';
|
||||||
editBtn.title = 'Редактировать';
|
editBtn.title = 'Редактировать';
|
||||||
editBtn.addEventListener('click', () => startEditMessage(wrapper, messageId));
|
editBtn.addEventListener('click', () => startEditMessage(wrapper, messageId));
|
||||||
actions.appendChild(editBtn);
|
actions.appendChild(editBtn);
|
||||||
|
|
||||||
if (role === 'assistant') {
|
if (role === 'assistant') {
|
||||||
const regenBtn = document.createElement('button');
|
const regenBtn = document.createElement('button');
|
||||||
regenBtn.type = 'button';
|
regenBtn.type = 'button';
|
||||||
@@ -204,14 +210,12 @@ function attachMessageActions(wrapper, messageId, role) {
|
|||||||
regenBtn.addEventListener('click', () => regenerateMessage(messageId, wrapper));
|
regenBtn.addEventListener('click', () => regenerateMessage(messageId, wrapper));
|
||||||
actions.appendChild(regenBtn);
|
actions.appendChild(regenBtn);
|
||||||
}
|
}
|
||||||
|
|
||||||
const branchBtn = document.createElement('button');
|
const branchBtn = document.createElement('button');
|
||||||
branchBtn.type = 'button';
|
branchBtn.type = 'button';
|
||||||
branchBtn.textContent = '🌿';
|
branchBtn.textContent = '🌿';
|
||||||
branchBtn.title = 'Ветка отсюда';
|
branchBtn.title = 'Ветка отсюда';
|
||||||
branchBtn.addEventListener('click', () => forkFromMessage(messageId));
|
branchBtn.addEventListener('click', () => forkFromMessage(messageId));
|
||||||
actions.appendChild(branchBtn);
|
actions.appendChild(branchBtn);
|
||||||
|
|
||||||
wrapper.appendChild(actions);
|
wrapper.appendChild(actions);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -224,7 +228,6 @@ async function startEditMessage(wrapper, messageId) {
|
|||||||
ta.value = original;
|
ta.value = original;
|
||||||
bubble.replaceWith(ta);
|
bubble.replaceWith(ta);
|
||||||
wrapper.querySelector('.message-actions')?.remove();
|
wrapper.querySelector('.message-actions')?.remove();
|
||||||
|
|
||||||
const saveRow = document.createElement('div');
|
const saveRow = document.createElement('div');
|
||||||
saveRow.className = 'message-actions';
|
saveRow.className = 'message-actions';
|
||||||
const saveBtn = document.createElement('button');
|
const saveBtn = document.createElement('button');
|
||||||
@@ -234,17 +237,10 @@ async function startEditMessage(wrapper, messageId) {
|
|||||||
saveRow.appendChild(saveBtn);
|
saveRow.appendChild(saveBtn);
|
||||||
saveRow.appendChild(cancelBtn);
|
saveRow.appendChild(cancelBtn);
|
||||||
wrapper.appendChild(saveRow);
|
wrapper.appendChild(saveRow);
|
||||||
|
|
||||||
const truncate = role => confirm(
|
|
||||||
role === 'user'
|
|
||||||
? 'Удалить все сообщения после этого? (рекомендуется)'
|
|
||||||
: 'Удалить все сообщения после этого?',
|
|
||||||
);
|
|
||||||
|
|
||||||
cancelBtn.addEventListener('click', () => reloadChatFromServer(sessionId));
|
cancelBtn.addEventListener('click', () => reloadChatFromServer(sessionId));
|
||||||
saveBtn.addEventListener('click', async () => {
|
saveBtn.addEventListener('click', async () => {
|
||||||
const role = wrapper.classList.contains('user') ? 'user' : 'assistant';
|
const role = wrapper.classList.contains('user') ? 'user' : 'assistant';
|
||||||
const doTruncate = truncate(role);
|
const doTruncate = confirm(role === 'user' ? 'Удалить все сообщения после этого? (рекомендуется)' : 'Удалить все сообщения после этого?');
|
||||||
const res = await fetch(`/chat/messages/${messageId}`, {
|
const res = await fetch(`/chat/messages/${messageId}`, {
|
||||||
method: 'PATCH',
|
method: 'PATCH',
|
||||||
headers: { 'Content-Type': 'application/json' },
|
headers: { 'Content-Type': 'application/json' },
|
||||||
@@ -266,11 +262,7 @@ async function regenerateMessage(messageId, wrapper) {
|
|||||||
const res = await fetch('/chat/regenerate', {
|
const res = await fetch('/chat/regenerate', {
|
||||||
method: 'POST',
|
method: 'POST',
|
||||||
headers: { 'Content-Type': 'application/json' },
|
headers: { 'Content-Type': 'application/json' },
|
||||||
body: JSON.stringify({
|
body: JSON.stringify({ session_id: sessionId, persona_id: currentPersona, message_id: messageId }),
|
||||||
session_id: sessionId,
|
|
||||||
persona_id: currentPersona,
|
|
||||||
message_id: messageId,
|
|
||||||
}),
|
|
||||||
});
|
});
|
||||||
if (!res.ok) throw new Error('Ошибка: ' + res.status);
|
if (!res.ok) throw new Error('Ошибка: ' + res.status);
|
||||||
removeTyping();
|
removeTyping();
|
||||||
@@ -315,6 +307,8 @@ export async function reloadChatFromServer(id) {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const IMAGE_PROMPT_RE = /\[IMAGE_PROMPT:.*?\]/gs;
|
||||||
|
|
||||||
async function consumeStream(res) {
|
async function consumeStream(res) {
|
||||||
const reader = res.body.getReader();
|
const reader = res.body.getReader();
|
||||||
const decoder = new TextDecoder();
|
const decoder = new TextDecoder();
|
||||||
@@ -324,71 +318,80 @@ async function consumeStream(res) {
|
|||||||
while (true) {
|
while (true) {
|
||||||
const { done, value } = await reader.read();
|
const { done, value } = await reader.read();
|
||||||
if (done) break;
|
if (done) break;
|
||||||
|
|
||||||
buffer += decoder.decode(value, { stream: true });
|
buffer += decoder.decode(value, { stream: true });
|
||||||
const lines = buffer.split('\n');
|
const lines = buffer.split('\n');
|
||||||
buffer = lines.pop();
|
buffer = lines.pop();
|
||||||
|
|
||||||
for (const line of lines) {
|
for (const line of lines) {
|
||||||
if (!line.startsWith('data: ')) continue;
|
if (!line.startsWith('data: ')) continue;
|
||||||
try {
|
let data;
|
||||||
const data = JSON.parse(line.slice(6));
|
try { data = JSON.parse(line.slice(6)); } catch { continue; }
|
||||||
|
|
||||||
|
// Narrator arrives BEFORE chunks — render immediately
|
||||||
|
if (data.narrator) {
|
||||||
|
renderNarratorMessage(data.narrator);
|
||||||
|
}
|
||||||
|
|
||||||
if (data.chunk !== undefined) {
|
if (data.chunk !== undefined) {
|
||||||
if (!bubble) {
|
if (!bubble) {
|
||||||
bubble = addMessage('assistant', '');
|
bubble = addMessage('assistant', '');
|
||||||
bubble.classList.add('typing-active');
|
bubble.classList.add('typing-active');
|
||||||
}
|
}
|
||||||
bubble.textContent += data.chunk;
|
bubble.textContent += data.chunk;
|
||||||
bubble.textContent = bubble.textContent.replace(/\[IMAGE_PROMPT:.*?\]/gs, '').trim();
|
|
||||||
dom.messagesEl.scrollTop = dom.messagesEl.scrollHeight;
|
dom.messagesEl.scrollTop = dom.messagesEl.scrollHeight;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (data.image_generating && bubble) {
|
||||||
|
bubble.classList.remove('typing-active');
|
||||||
|
const wrapper = bubble.parentElement;
|
||||||
|
if (data.image_prompt && !wrapper.querySelector('.image-prompt-block')) {
|
||||||
|
wrapper.appendChild(createImagePromptBlock(data.image_prompt));
|
||||||
|
}
|
||||||
|
showImageGenerating(wrapper);
|
||||||
|
dom.messagesEl.scrollTop = dom.messagesEl.scrollHeight;
|
||||||
|
}
|
||||||
|
|
||||||
if (data.done) {
|
if (data.done) {
|
||||||
|
const wrapper = bubble?.parentElement;
|
||||||
|
removeImageGenerating(wrapper);
|
||||||
bubble?.classList.remove('typing-active');
|
bubble?.classList.remove('typing-active');
|
||||||
if (data.narrator && !bubble) {
|
|
||||||
renderNarratorMessage(data.narrator);
|
// Strip IMAGE_PROMPT tag from final text
|
||||||
} else if (data.narrator && bubble) {
|
if (bubble) {
|
||||||
const assistantWrapper = bubble.parentElement;
|
bubble.textContent = bubble.textContent.replace(IMAGE_PROMPT_RE, '').trim();
|
||||||
dom.messagesEl.insertBefore(buildNarratorWrapper(data.narrator), assistantWrapper);
|
|
||||||
}
|
}
|
||||||
if (data.image_prompt && bubble) {
|
|
||||||
bubble.parentElement.appendChild(createImagePromptBlock(data.image_prompt));
|
if (data.image_prompt && wrapper && !wrapper.querySelector('.image-prompt-block')) {
|
||||||
|
wrapper.appendChild(createImagePromptBlock(data.image_prompt));
|
||||||
}
|
}
|
||||||
if (data.image_path && bubble) {
|
if (data.image_path && wrapper) {
|
||||||
appendChatImage(bubble.parentElement, data.image_path);
|
console.log('[image] appending', data.image_path, 'to', wrapper);
|
||||||
|
appendChatImage(wrapper, data.image_path);
|
||||||
|
} else {
|
||||||
|
console.log('[image] skip: image_path=', data.image_path, 'wrapper=', wrapper);
|
||||||
}
|
}
|
||||||
if (data.image_error && bubble) {
|
if (data.image_error && wrapper) {
|
||||||
const err = document.createElement('div');
|
const err = document.createElement('div');
|
||||||
err.className = 'image-error';
|
err.className = 'image-error';
|
||||||
err.textContent = '🖼 ' + data.image_error;
|
err.textContent = '🖼 ' + data.image_error;
|
||||||
bubble.parentElement.appendChild(err);
|
wrapper.appendChild(err);
|
||||||
}
|
}
|
||||||
if (data.choices && bubble) {
|
if (data.choices?.length && bubble) renderChoices(bubble.parentElement, data.choices);
|
||||||
renderChoices(bubble.parentElement, data.choices);
|
if (data.debug) renderDebugBlocks(bubble?.parentElement || dom.messagesEl, data.debug);
|
||||||
}
|
if (data.affinity !== undefined) updateAffinityDisplay(data.affinity);
|
||||||
if (data.debug) {
|
if (data.quests?.length) updateQuestPanel(data.quests);
|
||||||
renderDebugBlocks(bubble?.parentElement || dom.messagesEl, data.debug);
|
|
||||||
}
|
|
||||||
if (data.affinity !== undefined) {
|
|
||||||
updateAffinityDisplay(data.affinity);
|
|
||||||
}
|
|
||||||
if (data.quests?.length) {
|
|
||||||
updateQuestPanel(data.quests);
|
|
||||||
}
|
|
||||||
await reloadChatFromServer(sessionId);
|
|
||||||
const { loadSessions } = await import('./sessions.js');
|
const { loadSessions } = await import('./sessions.js');
|
||||||
loadSessions();
|
loadSessions();
|
||||||
}
|
}
|
||||||
} catch { /* skip */ }
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
export function addMessage(role, content = '', imagePrompt = null, imagePath = null, messageId = null) {
|
export function addMessage(role, content = '', imagePrompt = null, imagePath = null, messageId = null) {
|
||||||
updateEmptyState();
|
updateEmptyState();
|
||||||
|
|
||||||
const wrapper = document.createElement('div');
|
const wrapper = document.createElement('div');
|
||||||
wrapper.className = `message ${role}`;
|
wrapper.className = `message ${role}`;
|
||||||
|
|
||||||
const label = document.createElement('div');
|
const label = document.createElement('div');
|
||||||
label.className = 'label';
|
label.className = 'label';
|
||||||
label.textContent = role === 'user' ? 'Вы' : 'AI';
|
label.textContent = role === 'user' ? 'Вы' : 'AI';
|
||||||
@@ -445,9 +448,7 @@ export function addMessage(role, content = '', imagePrompt = null, imagePath = n
|
|||||||
|
|
||||||
if (prompt) wrapper.appendChild(createImagePromptBlock(prompt));
|
if (prompt) wrapper.appendChild(createImagePromptBlock(prompt));
|
||||||
if (imagePath) appendChatImage(wrapper, imagePath);
|
if (imagePath) appendChatImage(wrapper, imagePath);
|
||||||
|
|
||||||
attachMessageActions(wrapper, messageId, role);
|
attachMessageActions(wrapper, messageId, role);
|
||||||
|
|
||||||
dom.messagesEl.appendChild(wrapper);
|
dom.messagesEl.appendChild(wrapper);
|
||||||
dom.messagesEl.scrollTop = dom.messagesEl.scrollHeight;
|
dom.messagesEl.scrollTop = dom.messagesEl.scrollHeight;
|
||||||
return bubble;
|
return bubble;
|
||||||
@@ -477,27 +478,18 @@ export function clearMessages() {
|
|||||||
export async function sendMessage(text, isNarratorChoice = false) {
|
export async function sendMessage(text, isNarratorChoice = false) {
|
||||||
if (typeof text !== 'string') text = dom.inputEl.value.trim();
|
if (typeof text !== 'string') text = dom.inputEl.value.trim();
|
||||||
if (!text || !sessionId) return;
|
if (!text || !sessionId) return;
|
||||||
|
|
||||||
dom.inputEl.value = '';
|
dom.inputEl.value = '';
|
||||||
dom.inputEl.style.height = 'auto';
|
dom.inputEl.style.height = 'auto';
|
||||||
dom.sendBtn.disabled = true;
|
dom.sendBtn.disabled = true;
|
||||||
|
|
||||||
addMessage('user', isNarratorChoice ? `[${text}]` : text);
|
addMessage('user', isNarratorChoice ? `[${text}]` : text);
|
||||||
showTyping();
|
showTyping();
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const res = await fetch('/chat/stream', {
|
const res = await fetch('/chat/stream', {
|
||||||
method: 'POST',
|
method: 'POST',
|
||||||
headers: { 'Content-Type': 'application/json' },
|
headers: { 'Content-Type': 'application/json' },
|
||||||
body: JSON.stringify({
|
body: JSON.stringify({ message: text, session_id: sessionId, persona_id: currentPersona, is_narrator_choice: isNarratorChoice }),
|
||||||
message: text,
|
|
||||||
session_id: sessionId,
|
|
||||||
persona_id: currentPersona,
|
|
||||||
is_narrator_choice: isNarratorChoice,
|
|
||||||
}),
|
|
||||||
});
|
});
|
||||||
if (!res.ok) throw new Error('Ошибка сервера: ' + res.status);
|
if (!res.ok) throw new Error('Ошибка сервера: ' + res.status);
|
||||||
|
|
||||||
removeTyping();
|
removeTyping();
|
||||||
await consumeStream(res);
|
await consumeStream(res);
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
@@ -509,27 +501,6 @@ export async function sendMessage(text, isNarratorChoice = false) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
function buildNarratorWrapper(narrator) {
|
|
||||||
const wrapper = document.createElement('div');
|
|
||||||
wrapper.className = 'message narrator';
|
|
||||||
const label = document.createElement('div');
|
|
||||||
label.className = 'label';
|
|
||||||
label.textContent = '📖 Рассказчик';
|
|
||||||
wrapper.appendChild(label);
|
|
||||||
const bubble = document.createElement('div');
|
|
||||||
bubble.className = 'bubble';
|
|
||||||
const diceBlock = document.createElement('div');
|
|
||||||
diceBlock.className = `dice-block ${OUTCOME_CLASS[narrator.outcome] || ''}`;
|
|
||||||
diceBlock.innerHTML = `<span class="dice-icon">🎲</span><span class="dice-roll">${narrator.roll}</span><span class="dice-outcome">${narrator.outcome}</span>`;
|
|
||||||
bubble.appendChild(diceBlock);
|
|
||||||
const textEl = document.createElement('div');
|
|
||||||
textEl.className = 'narrator-text';
|
|
||||||
textEl.textContent = narrator.text;
|
|
||||||
bubble.appendChild(textEl);
|
|
||||||
wrapper.appendChild(bubble);
|
|
||||||
return wrapper;
|
|
||||||
}
|
|
||||||
|
|
||||||
export async function clearHistory() {
|
export async function clearHistory() {
|
||||||
if (!sessionId) return;
|
if (!sessionId) return;
|
||||||
await fetch(`/chat/${sessionId}`, { method: 'DELETE' });
|
await fetch(`/chat/${sessionId}`, { method: 'DELETE' });
|
||||||
|
|||||||
@@ -0,0 +1,154 @@
|
|||||||
|
import { sessionId, currentPersona, dom } from './state.js';
|
||||||
|
import { GENRE_LABELS, bindGenreGrid, resetGenreGrid } from './utils.js';
|
||||||
|
|
||||||
|
const chatSettingsGenres = new Set();
|
||||||
|
|
||||||
|
function updateChatSettingsGenresLabel() {
|
||||||
|
const el = document.getElementById('chatSettingsGenresLabel');
|
||||||
|
const labels = [...chatSettingsGenres].map(g => GENRE_LABELS[g] || g);
|
||||||
|
if (!el) return;
|
||||||
|
if (labels.length) {
|
||||||
|
el.textContent = `Выбрано: ${labels.join(' + ')}`;
|
||||||
|
el.classList.remove('hidden');
|
||||||
|
} else {
|
||||||
|
el.classList.add('hidden');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function loadRpgSettingsToDom(prefix, settings) {
|
||||||
|
document.getElementById(`${prefix}SettingDice`).checked = settings.dice !== false;
|
||||||
|
document.getElementById(`${prefix}SettingNarrator`).checked = settings.narrator !== false;
|
||||||
|
document.getElementById(`${prefix}SettingQuests`).checked = settings.quests !== false;
|
||||||
|
document.getElementById(`${prefix}SettingAffinity`).checked = settings.affinity !== false;
|
||||||
|
document.getElementById(`${prefix}SettingChoices`).checked = settings.choices !== false;
|
||||||
|
}
|
||||||
|
|
||||||
|
function readRpgSettingsFromDom(prefix) {
|
||||||
|
return {
|
||||||
|
dice: document.getElementById(`${prefix}SettingDice`)?.checked ?? true,
|
||||||
|
narrator: document.getElementById(`${prefix}SettingNarrator`)?.checked ?? true,
|
||||||
|
quests: document.getElementById(`${prefix}SettingQuests`)?.checked ?? true,
|
||||||
|
affinity: document.getElementById(`${prefix}SettingAffinity`)?.checked ?? true,
|
||||||
|
choices: document.getElementById(`${prefix}SettingChoices`)?.checked ?? true,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
async function bootstrapRpg(sid, personaId, genreValue, settings) {
|
||||||
|
const { updateQuestPanel, addMessage } = await import('./chat.js');
|
||||||
|
await fetch(`/sessions/${sid}`, {
|
||||||
|
method: 'PATCH',
|
||||||
|
headers: { 'Content-Type': 'application/json' },
|
||||||
|
body: JSON.stringify({
|
||||||
|
rpg_enabled: true,
|
||||||
|
genre: genreValue,
|
||||||
|
rpg_settings_json: JSON.stringify(settings),
|
||||||
|
}),
|
||||||
|
});
|
||||||
|
const res = await fetch('/chat/rpg/bootstrap', {
|
||||||
|
method: 'POST',
|
||||||
|
headers: { 'Content-Type': 'application/json' },
|
||||||
|
body: JSON.stringify({ session_id: sid, persona_id: personaId, genre: genreValue }),
|
||||||
|
});
|
||||||
|
if (res.ok) {
|
||||||
|
const data = await res.json();
|
||||||
|
if (data.quests) updateQuestPanel(data.quests);
|
||||||
|
if (data.plot_arc) {
|
||||||
|
const title = data.plot_arc.title || '';
|
||||||
|
const hint = data.plot_arc.next_beat_hint || '';
|
||||||
|
if (title || hint) addMessage('assistant', `📖 ${title}${hint ? '\n' + hint : ''}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function openChatSettings() {
|
||||||
|
if (!sessionId) return;
|
||||||
|
const res = await fetch(`/sessions/${sessionId}`);
|
||||||
|
if (!res.ok) return;
|
||||||
|
const s = await res.json();
|
||||||
|
|
||||||
|
document.getElementById('chatSettingsTitle').value = s.title || '';
|
||||||
|
const rpgOn = !!s.rpg_enabled;
|
||||||
|
document.getElementById('chatSettingsRpg').checked = rpgOn;
|
||||||
|
document.getElementById('chatSettingsRpgBlock').classList.toggle('hidden', !rpgOn);
|
||||||
|
|
||||||
|
chatSettingsGenres.clear();
|
||||||
|
(s.genre || 'adventure').split(',').forEach(g => {
|
||||||
|
const t = g.trim();
|
||||||
|
if (t) chatSettingsGenres.add(t);
|
||||||
|
});
|
||||||
|
resetGenreGrid(document.getElementById('chatSettingsGenreGrid'), chatSettingsGenres);
|
||||||
|
document.getElementById('chatSettingsGenreGrid')?.querySelectorAll('.genre-btn').forEach(btn => {
|
||||||
|
if (chatSettingsGenres.has(btn.dataset.genre)) btn.classList.add('selected');
|
||||||
|
});
|
||||||
|
updateChatSettingsGenresLabel();
|
||||||
|
|
||||||
|
let settings = {};
|
||||||
|
try { settings = JSON.parse(s.rpg_settings_json || '{}'); } catch { /* ignore */ }
|
||||||
|
loadRpgSettingsToDom('cs', settings);
|
||||||
|
|
||||||
|
let phase = '';
|
||||||
|
try {
|
||||||
|
const arc = JSON.parse(s.plot_arc_json || '{}');
|
||||||
|
phase = arc.phase || '';
|
||||||
|
} catch { /* ignore */ }
|
||||||
|
document.getElementById('chatSettingsMeta').innerHTML = [
|
||||||
|
`Симпатия: ${s.affinity ?? 0}`,
|
||||||
|
s.genre ? `Жанр: ${(s.genre || '').split(',').map(g => GENRE_LABELS[g.trim()] || g).join(' + ')}` : '',
|
||||||
|
phase ? `Фаза арки: ${phase}` : '',
|
||||||
|
].filter(Boolean).join('<br>');
|
||||||
|
|
||||||
|
document.getElementById('chatSettingsModal').classList.add('open');
|
||||||
|
}
|
||||||
|
|
||||||
|
export function initChatSettings() {
|
||||||
|
bindGenreGrid(
|
||||||
|
document.getElementById('chatSettingsGenreGrid'),
|
||||||
|
chatSettingsGenres,
|
||||||
|
updateChatSettingsGenresLabel,
|
||||||
|
);
|
||||||
|
|
||||||
|
document.getElementById('chatSettingsRpg')?.addEventListener('change', (e) => {
|
||||||
|
document.getElementById('chatSettingsRpgBlock').classList.toggle('hidden', !e.target.checked);
|
||||||
|
});
|
||||||
|
|
||||||
|
document.getElementById('chatSettingsCancel')?.addEventListener('click', () => {
|
||||||
|
document.getElementById('chatSettingsModal').classList.remove('open');
|
||||||
|
});
|
||||||
|
|
||||||
|
document.getElementById('chatSettingsSave')?.addEventListener('click', async () => {
|
||||||
|
if (!sessionId) return;
|
||||||
|
const { loadSessions, applySessionUi } = await import('./sessions.js');
|
||||||
|
|
||||||
|
const title = document.getElementById('chatSettingsTitle').value.trim();
|
||||||
|
const rpgOn = document.getElementById('chatSettingsRpg').checked;
|
||||||
|
const genreValue = [...chatSettingsGenres].join(',') || 'adventure';
|
||||||
|
const settings = readRpgSettingsFromDom('cs');
|
||||||
|
|
||||||
|
await fetch(`/sessions/${sessionId}`, {
|
||||||
|
method: 'PATCH',
|
||||||
|
headers: { 'Content-Type': 'application/json' },
|
||||||
|
body: JSON.stringify({
|
||||||
|
title: title || undefined,
|
||||||
|
rpg_enabled: rpgOn,
|
||||||
|
genre: genreValue,
|
||||||
|
rpg_settings_json: JSON.stringify(settings),
|
||||||
|
}),
|
||||||
|
});
|
||||||
|
|
||||||
|
if (rpgOn) {
|
||||||
|
const sessionRes = await fetch(`/sessions/${sessionId}`);
|
||||||
|
const s = sessionRes.ok ? await sessionRes.json() : {};
|
||||||
|
let arc = {};
|
||||||
|
try { arc = JSON.parse(s.plot_arc_json || '{}'); } catch { /* ignore */ }
|
||||||
|
if (!arc || !Object.keys(arc).length) {
|
||||||
|
await bootstrapRpg(sessionId, currentPersona, genreValue, settings);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
document.getElementById('chatSettingsModal').classList.remove('open');
|
||||||
|
const updated = await (await fetch(`/sessions/${sessionId}`)).json();
|
||||||
|
applySessionUi(updated);
|
||||||
|
dom.headerTitle.textContent = updated.title || 'Новый чат';
|
||||||
|
await loadSessions();
|
||||||
|
});
|
||||||
|
}
|
||||||
@@ -0,0 +1,257 @@
|
|||||||
|
import { setSessionId, setCurrentPersona, currentPersona, dom } from './state.js';
|
||||||
|
import {
|
||||||
|
initWizard,
|
||||||
|
GENRE_LABELS,
|
||||||
|
bindGenreGrid,
|
||||||
|
resetGenreGrid,
|
||||||
|
fillGreetingSelect,
|
||||||
|
getSelectedGreeting,
|
||||||
|
} from './utils.js';
|
||||||
|
import { personaIndex, highlightPersona } from './personas.js';
|
||||||
|
|
||||||
|
let newChatPersonaId = currentPersona;
|
||||||
|
let newChatGreetingCtx = null;
|
||||||
|
const newChatGenres = new Set();
|
||||||
|
const newChatModalEl = document.getElementById('newChatModal');
|
||||||
|
let newChatWizard;
|
||||||
|
|
||||||
|
async function resolveGreetingContext(personaId) {
|
||||||
|
const p = personaIndex.get(personaId);
|
||||||
|
let firstMes = p?.first_mes || '';
|
||||||
|
let alternates = p?.alternate_greetings || [];
|
||||||
|
if (personaId.startsWith('card_') && (!alternates.length || !firstMes)) {
|
||||||
|
try {
|
||||||
|
const r = await fetch(`/characters/${personaId.slice(5)}`);
|
||||||
|
if (r.ok) {
|
||||||
|
const c = await r.json();
|
||||||
|
firstMes = c.first_mes || firstMes;
|
||||||
|
alternates = c.alternate_greetings?.length ? c.alternate_greetings : alternates;
|
||||||
|
}
|
||||||
|
} catch {
|
||||||
|
/* ignore */
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return { firstMes, alternates };
|
||||||
|
}
|
||||||
|
|
||||||
|
async function syncNewChatGreetingBlock() {
|
||||||
|
const block = document.getElementById('newChatGreetingBlock');
|
||||||
|
const select = document.getElementById('newChatGreetingSelect');
|
||||||
|
const text = document.getElementById('newChatGreetingText');
|
||||||
|
if (!block || !select || !text) return;
|
||||||
|
|
||||||
|
newChatGreetingCtx = await resolveGreetingContext(newChatPersonaId);
|
||||||
|
const { firstMes, alternates } = newChatGreetingCtx;
|
||||||
|
if (!alternates.length) {
|
||||||
|
block.classList.add('hidden');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
block.classList.remove('hidden');
|
||||||
|
fillGreetingSelect(select, firstMes, alternates);
|
||||||
|
text.value = firstMes;
|
||||||
|
select.onchange = () => {
|
||||||
|
text.value = getSelectedGreeting(select, firstMes, alternates);
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function getNewChatFirstMesOverride() {
|
||||||
|
const block = document.getElementById('newChatGreetingBlock');
|
||||||
|
if (!block || block.classList.contains('hidden') || !newChatGreetingCtx) return null;
|
||||||
|
const edited = document.getElementById('newChatGreetingText')?.value.trim();
|
||||||
|
if (edited) return edited;
|
||||||
|
const select = document.getElementById('newChatGreetingSelect');
|
||||||
|
const { firstMes, alternates } = newChatGreetingCtx;
|
||||||
|
return getSelectedGreeting(select, firstMes, alternates) || null;
|
||||||
|
}
|
||||||
|
|
||||||
|
function isNewChatRpg() {
|
||||||
|
return document.querySelector('input[name="newChatRpg"]:checked')?.value === '1';
|
||||||
|
}
|
||||||
|
|
||||||
|
function syncNewChatStep3() {
|
||||||
|
const plain = document.getElementById('newChatPlainStep');
|
||||||
|
const rpg = document.getElementById('newChatRpgStep');
|
||||||
|
if (isNewChatRpg()) {
|
||||||
|
plain?.classList.add('hidden');
|
||||||
|
rpg?.classList.remove('hidden');
|
||||||
|
} else {
|
||||||
|
plain?.classList.remove('hidden');
|
||||||
|
rpg?.classList.add('hidden');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function fillNewChatPersonaGrid() {
|
||||||
|
const grid = document.getElementById('newChatPersonaGrid');
|
||||||
|
if (!grid) return;
|
||||||
|
grid.innerHTML = '';
|
||||||
|
newChatPersonaId = currentPersona;
|
||||||
|
for (const p of personaIndex.values()) {
|
||||||
|
const card = document.createElement('button');
|
||||||
|
card.type = 'button';
|
||||||
|
card.className = 'persona-pick-card' + (p.persona_id === newChatPersonaId ? ' selected' : '');
|
||||||
|
card.dataset.id = p.persona_id;
|
||||||
|
card.innerHTML = `<span class="emoji">${p.emoji || '🤖'}</span>${p.name}`;
|
||||||
|
card.addEventListener('click', () => {
|
||||||
|
newChatPersonaId = p.persona_id;
|
||||||
|
grid.querySelectorAll('.persona-pick-card').forEach(c => {
|
||||||
|
c.classList.toggle('selected', c.dataset.id === newChatPersonaId);
|
||||||
|
});
|
||||||
|
syncNewChatGreetingBlock();
|
||||||
|
});
|
||||||
|
grid.appendChild(card);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function updateNewChatGenresLabel() {
|
||||||
|
const el = document.getElementById('newChatGenresLabel');
|
||||||
|
const nextBtn = document.getElementById('newChatNext');
|
||||||
|
const labels = [...newChatGenres].map(g => GENRE_LABELS[g] || g);
|
||||||
|
if (el) {
|
||||||
|
if (labels.length) {
|
||||||
|
el.textContent = `Выбрано: ${labels.join(' + ')}`;
|
||||||
|
el.classList.remove('hidden');
|
||||||
|
} else {
|
||||||
|
el.classList.add('hidden');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (nextBtn && isNewChatRpg()) {
|
||||||
|
const wizard = newChatModalEl?.querySelector('.modal-wizard');
|
||||||
|
const onStep3 = wizard?.querySelector('.wizard-page[data-step="3"]')?.classList.contains('active');
|
||||||
|
if (onStep3) nextBtn.disabled = newChatGenres.size === 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function bootstrapRpg(sid, personaId, genreValue, settings) {
|
||||||
|
const { updateQuestPanel, addMessage } = await import('./chat.js');
|
||||||
|
await fetch(`/sessions/${sid}`, {
|
||||||
|
method: 'PATCH',
|
||||||
|
headers: { 'Content-Type': 'application/json' },
|
||||||
|
body: JSON.stringify({
|
||||||
|
rpg_enabled: true,
|
||||||
|
genre: genreValue,
|
||||||
|
rpg_settings_json: JSON.stringify(settings),
|
||||||
|
}),
|
||||||
|
});
|
||||||
|
const res = await fetch('/chat/rpg/bootstrap', {
|
||||||
|
method: 'POST',
|
||||||
|
headers: { 'Content-Type': 'application/json' },
|
||||||
|
body: JSON.stringify({ session_id: sid, persona_id: personaId, genre: genreValue }),
|
||||||
|
});
|
||||||
|
if (res.ok) {
|
||||||
|
const data = await res.json();
|
||||||
|
if (data.quests) updateQuestPanel(data.quests);
|
||||||
|
if (data.plot_arc) {
|
||||||
|
const title = data.plot_arc.title || '';
|
||||||
|
const hint = data.plot_arc.next_beat_hint || '';
|
||||||
|
if (title || hint) addMessage('assistant', `📖 ${title}${hint ? '\n' + hint : ''}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export function openNewChatWizard() {
|
||||||
|
fillNewChatPersonaGrid();
|
||||||
|
resetGenreGrid(document.getElementById('newChatGenreGrid'), newChatGenres);
|
||||||
|
updateNewChatGenresLabel();
|
||||||
|
document.querySelector('input[name="newChatRpg"][value="0"]')?.click();
|
||||||
|
document.getElementById('newChatTitle').value = '';
|
||||||
|
syncNewChatStep3();
|
||||||
|
newChatWizard?.reset();
|
||||||
|
newChatModalEl?.classList.add('open');
|
||||||
|
syncNewChatGreetingBlock();
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function createNewChatFromWizard() {
|
||||||
|
const { clearMessages, initChat, reloadChatFromServer } = await import('./chat.js');
|
||||||
|
const { loadSessions, applySessionUi } = await import('./sessions.js');
|
||||||
|
|
||||||
|
const sid = 'sess_' + Math.random().toString(36).slice(2, 10);
|
||||||
|
setSessionId(sid);
|
||||||
|
setCurrentPersona(newChatPersonaId);
|
||||||
|
clearMessages();
|
||||||
|
|
||||||
|
const customTitle = document.getElementById('newChatTitle')?.value.trim();
|
||||||
|
const rpg = isNewChatRpg();
|
||||||
|
|
||||||
|
newChatModalEl?.classList.remove('open');
|
||||||
|
newChatWizard?.reset();
|
||||||
|
|
||||||
|
try {
|
||||||
|
await fetch(`/sessions/${sid}`, {
|
||||||
|
method: 'PATCH',
|
||||||
|
headers: { 'Content-Type': 'application/json' },
|
||||||
|
body: JSON.stringify({ persona_id: newChatPersonaId, rpg_enabled: rpg }),
|
||||||
|
});
|
||||||
|
|
||||||
|
if (customTitle) {
|
||||||
|
await fetch(`/sessions/${sid}`, {
|
||||||
|
method: 'PATCH',
|
||||||
|
headers: { 'Content-Type': 'application/json' },
|
||||||
|
body: JSON.stringify({ title: customTitle }),
|
||||||
|
});
|
||||||
|
dom.headerTitle.textContent = customTitle;
|
||||||
|
} else {
|
||||||
|
const pName = personaIndex.get(newChatPersonaId)?.name || newChatPersonaId;
|
||||||
|
dom.headerTitle.textContent = rpg ? `${pName} — RPG` : `${pName} — новый чат`;
|
||||||
|
}
|
||||||
|
|
||||||
|
highlightPersona(newChatPersonaId);
|
||||||
|
const greetingOverride = getNewChatFirstMesOverride();
|
||||||
|
await initChat(greetingOverride ? { first_mes_override: greetingOverride } : {});
|
||||||
|
|
||||||
|
if (rpg) {
|
||||||
|
const genreValue = [...newChatGenres].join(',') || 'adventure';
|
||||||
|
const settings = {
|
||||||
|
dice: document.getElementById('ncSettingDice')?.checked ?? true,
|
||||||
|
narrator: document.getElementById('ncSettingNarrator')?.checked ?? true,
|
||||||
|
quests: document.getElementById('ncSettingQuests')?.checked ?? true,
|
||||||
|
affinity: document.getElementById('ncSettingAffinity')?.checked ?? true,
|
||||||
|
choices: document.getElementById('ncSettingChoices')?.checked ?? true,
|
||||||
|
};
|
||||||
|
await bootstrapRpg(sid, newChatPersonaId, genreValue, settings);
|
||||||
|
}
|
||||||
|
|
||||||
|
await reloadChatFromServer(sid);
|
||||||
|
const sessionRes = await fetch(`/sessions/${sid}`);
|
||||||
|
if (sessionRes.ok) applySessionUi(await sessionRes.json());
|
||||||
|
await loadSessions();
|
||||||
|
} catch (e) {
|
||||||
|
console.error('createNewChat error:', e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export function initNewChatWizard() {
|
||||||
|
if (!newChatModalEl) return;
|
||||||
|
|
||||||
|
newChatWizard = initWizard(newChatModalEl.querySelector('.modal-wizard'), {
|
||||||
|
totalSteps: 3,
|
||||||
|
onStepChange(step) {
|
||||||
|
syncNewChatStep3();
|
||||||
|
if (step === 3 && !isNewChatRpg()) syncNewChatGreetingBlock();
|
||||||
|
const nextBtn = document.getElementById('newChatNext');
|
||||||
|
if (step === 3 && isNewChatRpg()) {
|
||||||
|
nextBtn.disabled = newChatGenres.size === 0;
|
||||||
|
} else {
|
||||||
|
nextBtn.disabled = false;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
validateStep(step) {
|
||||||
|
if (step === 1 && !newChatPersonaId) {
|
||||||
|
alert('Выбери персонажа');
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (step === 3 && isNewChatRpg() && newChatGenres.size === 0) {
|
||||||
|
alert('Выбери хотя бы один жанр');
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
bindGenreGrid(document.getElementById('newChatGenreGrid'), newChatGenres, updateNewChatGenresLabel);
|
||||||
|
|
||||||
|
document.getElementById('newChatCancel')?.addEventListener('click', () => {
|
||||||
|
newChatModalEl.classList.remove('open');
|
||||||
|
newChatWizard.reset();
|
||||||
|
});
|
||||||
|
document.getElementById('newChatCreate')?.addEventListener('click', createNewChatFromWizard);
|
||||||
|
}
|
||||||
+154
-26
@@ -1,10 +1,25 @@
|
|||||||
import { currentPersona, setCurrentPersona, sessionId } from './state.js';
|
import { currentPersona, setCurrentPersona, sessionId } from './state.js';
|
||||||
import { initChat } from './chat.js';
|
import { initChat } from './chat.js';
|
||||||
import { initWizard } from './utils.js';
|
import { initWizard, fillGreetingSelect, getSelectedGreeting } from './utils.js';
|
||||||
|
|
||||||
export let personaIndex = new Map();
|
export let personaIndex = new Map();
|
||||||
|
|
||||||
|
function parseAlternateGreetings(p) {
|
||||||
|
if (Array.isArray(p?.alternate_greetings) && p.alternate_greetings.length) {
|
||||||
|
return p.alternate_greetings;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
const parsed = JSON.parse(p?.alternate_greetings_json || '[]');
|
||||||
|
return Array.isArray(parsed) ? parsed : [];
|
||||||
|
} catch {
|
||||||
|
return [];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
let createWizard;
|
let createWizard;
|
||||||
|
let cardImportWizard;
|
||||||
|
let cardPreview = null;
|
||||||
|
let cardImportFile = null;
|
||||||
|
|
||||||
export function highlightPersona(personaId) {
|
export function highlightPersona(personaId) {
|
||||||
document.querySelectorAll('.persona-card').forEach(c => {
|
document.querySelectorAll('.persona-card').forEach(c => {
|
||||||
@@ -15,7 +30,10 @@ export function highlightPersona(personaId) {
|
|||||||
export async function loadPersonas() {
|
export async function loadPersonas() {
|
||||||
const res = await fetch('/personas/');
|
const res = await fetch('/personas/');
|
||||||
const personas = await res.json();
|
const personas = await res.json();
|
||||||
personaIndex = new Map(personas.map(p => [p.persona_id, p]));
|
personaIndex = new Map(personas.map(p => {
|
||||||
|
const alternate_greetings = parseAlternateGreetings(p);
|
||||||
|
return [p.persona_id, { ...p, alternate_greetings }];
|
||||||
|
}));
|
||||||
const bar = document.getElementById('personaBar');
|
const bar = document.getElementById('personaBar');
|
||||||
bar.innerHTML = '';
|
bar.innerHTML = '';
|
||||||
|
|
||||||
@@ -55,6 +73,19 @@ export async function loadPersonas() {
|
|||||||
document.getElementById('editAppearance').value = data.appearance_tags || '';
|
document.getElementById('editAppearance').value = data.appearance_tags || '';
|
||||||
document.getElementById('editLora').value = data.lora_name || '';
|
document.getElementById('editLora').value = data.lora_name || '';
|
||||||
document.getElementById('editLoraWeight').value = data.lora_weight ?? 0.8;
|
document.getElementById('editLoraWeight').value = data.lora_weight ?? 0.8;
|
||||||
|
const alts = data.alternate_greetings || [];
|
||||||
|
const altBlock = document.getElementById('editCardAltBlock');
|
||||||
|
const altSelect = document.getElementById('editCardGreetingSelect');
|
||||||
|
if (alts.length) {
|
||||||
|
altBlock?.classList.remove('hidden');
|
||||||
|
fillGreetingSelect(altSelect, data.first_mes, alts);
|
||||||
|
altSelect.onchange = () => {
|
||||||
|
document.getElementById('editFirstMes').value =
|
||||||
|
getSelectedGreeting(altSelect, data.first_mes, alts);
|
||||||
|
};
|
||||||
|
} else {
|
||||||
|
altBlock?.classList.add('hidden');
|
||||||
|
}
|
||||||
document.getElementById('cardEditOverlay').classList.add('open');
|
document.getElementById('cardEditOverlay').classList.add('open');
|
||||||
});
|
});
|
||||||
|
|
||||||
@@ -95,7 +126,7 @@ export async function loadPersonas() {
|
|||||||
importBtn.type = 'button';
|
importBtn.type = 'button';
|
||||||
importBtn.className = 'card-import-btn';
|
importBtn.className = 'card-import-btn';
|
||||||
importBtn.innerHTML = '📥<span>Chub</span>';
|
importBtn.innerHTML = '📥<span>Chub</span>';
|
||||||
importBtn.addEventListener('click', () => document.getElementById('cardModalOverlay').classList.add('open'));
|
importBtn.addEventListener('click', () => openCardImportModal());
|
||||||
bar.appendChild(importBtn);
|
bar.appendChild(importBtn);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -112,6 +143,122 @@ export async function selectPersona(personaId) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function fillImpCardForm(preview) {
|
||||||
|
document.getElementById('impCardName').value = preview.name || '';
|
||||||
|
document.getElementById('impCardDescription').value = preview.description || '';
|
||||||
|
document.getElementById('impCardPersonality').value = preview.personality || '';
|
||||||
|
document.getElementById('impCardScenario').value = preview.scenario || '';
|
||||||
|
document.getElementById('impCardMesExample').value = preview.mes_example || '';
|
||||||
|
document.getElementById('impCardAppearance').value = preview.appearance_tags || '';
|
||||||
|
|
||||||
|
const alts = preview.alternate_greetings || [];
|
||||||
|
const selectEl = document.getElementById('impCardGreetingSelect');
|
||||||
|
const firstMesEl = document.getElementById('impCardFirstMes');
|
||||||
|
fillGreetingSelect(selectEl, preview.first_mes, alts);
|
||||||
|
firstMesEl.value = preview.first_mes || '';
|
||||||
|
|
||||||
|
const altHint = document.getElementById('impCardAltHint');
|
||||||
|
if (alts.length) {
|
||||||
|
altHint.textContent = `В карточке ${alts.length} альтернативных приветствий — выбери в списке или отредактируй текст ниже`;
|
||||||
|
altHint.classList.remove('hidden');
|
||||||
|
} else {
|
||||||
|
altHint.classList.add('hidden');
|
||||||
|
}
|
||||||
|
|
||||||
|
selectEl.onchange = () => {
|
||||||
|
firstMesEl.value = getSelectedGreeting(selectEl, preview.first_mes, alts);
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
async function loadCardPreview() {
|
||||||
|
const fileInput = document.getElementById('cardFile');
|
||||||
|
if (!fileInput.files?.length) {
|
||||||
|
alert('Выберите файл карточки (JSON или PNG)');
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
const form = new FormData();
|
||||||
|
form.append('file', fileInput.files[0]);
|
||||||
|
const res = await fetch('/characters/preview', { method: 'POST', body: form });
|
||||||
|
const data = await res.json();
|
||||||
|
if (!res.ok) {
|
||||||
|
alert(data.detail || 'Ошибка чтения карточки');
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
cardPreview = data;
|
||||||
|
cardImportFile = fileInput.files[0];
|
||||||
|
fillImpCardForm(data);
|
||||||
|
const hint = document.getElementById('cardPreviewHint');
|
||||||
|
if (hint) {
|
||||||
|
hint.textContent = `${data.name} · ${data.alternate_count || 0} альт. приветствий`;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
function openCardImportModal() {
|
||||||
|
cardPreview = null;
|
||||||
|
cardImportFile = null;
|
||||||
|
document.getElementById('cardFile').value = '';
|
||||||
|
document.getElementById('cardPreviewHint').textContent = '';
|
||||||
|
document.getElementById('cardLora').value = '';
|
||||||
|
document.getElementById('cardLoraWeight').value = '0.8';
|
||||||
|
cardImportWizard?.reset();
|
||||||
|
document.getElementById('cardModalOverlay').classList.add('open');
|
||||||
|
}
|
||||||
|
|
||||||
|
function closeCardImportModal() {
|
||||||
|
document.getElementById('cardModalOverlay').classList.remove('open');
|
||||||
|
cardImportWizard?.reset();
|
||||||
|
cardPreview = null;
|
||||||
|
cardImportFile = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
function initCardImportWizard() {
|
||||||
|
const modal = document.getElementById('cardModalOverlay')?.querySelector('.modal-wizard');
|
||||||
|
if (!modal) return;
|
||||||
|
|
||||||
|
cardImportWizard = initWizard(modal, {
|
||||||
|
totalSteps: 2,
|
||||||
|
validateStep(step) {
|
||||||
|
if (step !== 1) return true;
|
||||||
|
return loadCardPreview();
|
||||||
|
},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
async function submitCardImport() {
|
||||||
|
if (!cardImportFile || !cardPreview) {
|
||||||
|
alert('Сначала загрузите и проверьте карточку');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const form = new FormData();
|
||||||
|
form.append('file', cardImportFile);
|
||||||
|
form.append('card_id', cardPreview.card_id || '');
|
||||||
|
form.append('lora_name', document.getElementById('cardLora').value.trim());
|
||||||
|
form.append('lora_weight', document.getElementById('cardLoraWeight').value || '0.8');
|
||||||
|
form.append('name', document.getElementById('impCardName').value.trim());
|
||||||
|
form.append('description', document.getElementById('impCardDescription').value.trim());
|
||||||
|
form.append('personality', document.getElementById('impCardPersonality').value.trim());
|
||||||
|
form.append('scenario', document.getElementById('impCardScenario').value.trim());
|
||||||
|
form.append('first_mes', document.getElementById('impCardFirstMes').value.trim());
|
||||||
|
form.append('mes_example', document.getElementById('impCardMesExample').value.trim());
|
||||||
|
form.append('appearance_tags', document.getElementById('impCardAppearance').value.trim());
|
||||||
|
form.append(
|
||||||
|
'alternate_greetings_json',
|
||||||
|
JSON.stringify(cardPreview.alternate_greetings || []),
|
||||||
|
);
|
||||||
|
|
||||||
|
const res = await fetch('/characters/import', { method: 'POST', body: form });
|
||||||
|
const data = await res.json();
|
||||||
|
if (!res.ok) {
|
||||||
|
alert(data.detail || 'Ошибка импорта');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
closeCardImportModal();
|
||||||
|
document.getElementById('cardFile').value = '';
|
||||||
|
await loadPersonas();
|
||||||
|
await selectPersona(data.persona_id);
|
||||||
|
}
|
||||||
|
|
||||||
export function initPersonaModals() {
|
export function initPersonaModals() {
|
||||||
const createModal = document.getElementById('modalOverlay');
|
const createModal = document.getElementById('modalOverlay');
|
||||||
createWizard = initWizard(createModal.querySelector('.modal-wizard'), {
|
createWizard = initWizard(createModal.querySelector('.modal-wizard'), {
|
||||||
@@ -132,8 +279,10 @@ export function initPersonaModals() {
|
|||||||
createModal.classList.remove('open');
|
createModal.classList.remove('open');
|
||||||
createWizard.reset();
|
createWizard.reset();
|
||||||
});
|
});
|
||||||
|
initCardImportWizard();
|
||||||
|
|
||||||
document.getElementById('cardModalCancel').addEventListener('click', () => {
|
document.getElementById('cardModalCancel').addEventListener('click', () => {
|
||||||
document.getElementById('cardModalOverlay').classList.remove('open');
|
closeCardImportModal();
|
||||||
});
|
});
|
||||||
document.getElementById('cardEditCancel').addEventListener('click', () => {
|
document.getElementById('cardEditCancel').addEventListener('click', () => {
|
||||||
document.getElementById('cardEditOverlay').classList.remove('open');
|
document.getElementById('cardEditOverlay').classList.remove('open');
|
||||||
@@ -210,28 +359,7 @@ export function initPersonaModals() {
|
|||||||
await loadPersonas();
|
await loadPersonas();
|
||||||
});
|
});
|
||||||
|
|
||||||
document.getElementById('cardModalImport').addEventListener('click', async () => {
|
document.getElementById('cardModalImport').addEventListener('click', submitCardImport);
|
||||||
const fileInput = document.getElementById('cardFile');
|
|
||||||
if (!fileInput.files?.length) {
|
|
||||||
alert('Выберите файл карточки (JSON или PNG)');
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
const form = new FormData();
|
|
||||||
form.append('file', fileInput.files[0]);
|
|
||||||
form.append('lora_name', document.getElementById('cardLora').value.trim());
|
|
||||||
form.append('lora_weight', document.getElementById('cardLoraWeight').value || '0.8');
|
|
||||||
|
|
||||||
const res = await fetch('/characters/import', { method: 'POST', body: form });
|
|
||||||
const data = await res.json();
|
|
||||||
if (!res.ok) {
|
|
||||||
alert(data.detail || 'Ошибка импорта');
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
document.getElementById('cardModalOverlay').classList.remove('open');
|
|
||||||
fileInput.value = '';
|
|
||||||
await loadPersonas();
|
|
||||||
await selectPersona(data.persona_id);
|
|
||||||
});
|
|
||||||
|
|
||||||
const personaEditSave = document.getElementById('personaEditSave');
|
const personaEditSave = document.getElementById('personaEditSave');
|
||||||
if (personaEditSave) {
|
if (personaEditSave) {
|
||||||
|
|||||||
+70
-329
@@ -1,13 +1,13 @@
|
|||||||
import {
|
import {
|
||||||
sessionId, setSessionId, setCurrentPersona, currentPersona, dom, setRpgEnabled,
|
sessionId, setSessionId, setCurrentPersona, currentPersona, dom, setRpgEnabled,
|
||||||
} from './state.js';
|
} from './state.js';
|
||||||
import {
|
import { updateQuestPanel, updateAffinityDisplay } from './chat.js';
|
||||||
clearMessages, addMessage, initChat, updateQuestPanel, updateAffinityDisplay, reloadChatFromServer,
|
import { highlightPersona, personaIndex } from './personas.js';
|
||||||
} from './chat.js';
|
import { formatSessionDate } from './utils.js';
|
||||||
import { highlightPersona, personaIndex, loadPersonas } from './personas.js';
|
import { openNewChatWizard } from './newChatWizard.js';
|
||||||
import {
|
|
||||||
initWizard, GENRE_LABELS, bindGenreGrid, resetGenreGrid, formatSessionDate,
|
export { openNewChatWizard, initNewChatWizard } from './newChatWizard.js';
|
||||||
} from './utils.js';
|
export { openChatSettings, initChatSettings } from './chatSettings.js';
|
||||||
|
|
||||||
function escapeTitle(t) {
|
function escapeTitle(t) {
|
||||||
const d = document.createElement('div');
|
const d = document.createElement('div');
|
||||||
@@ -15,10 +15,6 @@ function escapeTitle(t) {
|
|||||||
return d.innerHTML;
|
return d.innerHTML;
|
||||||
}
|
}
|
||||||
|
|
||||||
let newChatPersonaId = currentPersona;
|
|
||||||
const newChatGenres = new Set();
|
|
||||||
const chatSettingsGenres = new Set();
|
|
||||||
|
|
||||||
export function applySessionUi(session) {
|
export function applySessionUi(session) {
|
||||||
if (!session) return;
|
if (!session) return;
|
||||||
dom.headerTitle.textContent = session.title || 'Новый чат';
|
dom.headerTitle.textContent = session.title || 'Новый чат';
|
||||||
@@ -106,6 +102,7 @@ export async function loadSessions() {
|
|||||||
|
|
||||||
export async function switchSession(id) {
|
export async function switchSession(id) {
|
||||||
setSessionId(id);
|
setSessionId(id);
|
||||||
|
const { clearMessages } = await import('./chat.js');
|
||||||
clearMessages();
|
clearMessages();
|
||||||
await loadSessions();
|
await loadSessions();
|
||||||
await loadChatHistory(id);
|
await loadChatHistory(id);
|
||||||
@@ -113,337 +110,27 @@ export async function switchSession(id) {
|
|||||||
|
|
||||||
export async function loadChatHistory(id) {
|
export async function loadChatHistory(id) {
|
||||||
const sessionRes = await fetch(`/sessions/${id}`);
|
const sessionRes = await fetch(`/sessions/${id}`);
|
||||||
let session = null;
|
|
||||||
if (sessionRes.ok) {
|
if (sessionRes.ok) {
|
||||||
session = await sessionRes.json();
|
const s = await sessionRes.json();
|
||||||
if (session.persona_id) {
|
if (s.persona_id) {
|
||||||
setCurrentPersona(session.persona_id);
|
setCurrentPersona(s.persona_id);
|
||||||
highlightPersona(session.persona_id);
|
highlightPersona(s.persona_id);
|
||||||
}
|
}
|
||||||
applySessionUi(session);
|
applySessionUi(s);
|
||||||
}
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const blobRes = await fetch(`/chat/system/${id}`);
|
const blobRes = await fetch(`/chat/system/${id}`);
|
||||||
if (blobRes.ok) {
|
if (blobRes.ok) {
|
||||||
const blob = await blobRes.json();
|
_prevBlobSections = {}; // reset on session switch to avoid false highlights
|
||||||
const parts = [];
|
renderSystemBlob(await blobRes.json());
|
||||||
if (blob.system_prompt) parts.push(blob.system_prompt);
|
|
||||||
if (blob.status_quo) parts.push(`--- Status quo ---\n${blob.status_quo}\n---`);
|
|
||||||
if (blob.facts_json) parts.push(`facts_json: ${blob.facts_json}`);
|
|
||||||
if (blob.plot_arc_json) parts.push(`plot_arc_json: ${blob.plot_arc_json}`);
|
|
||||||
dom.systemBlobContent.textContent = parts.filter(Boolean).join('\n\n') || '—';
|
|
||||||
}
|
}
|
||||||
} catch { /* ignore */ }
|
} catch { /* ignore */ }
|
||||||
|
|
||||||
|
const { reloadChatFromServer } = await import('./chat.js');
|
||||||
await reloadChatFromServer(id);
|
await reloadChatFromServer(id);
|
||||||
}
|
}
|
||||||
|
|
||||||
async function bootstrapRpg(sid, personaId, genreValue, settings) {
|
|
||||||
await fetch(`/sessions/${sid}`, {
|
|
||||||
method: 'PATCH',
|
|
||||||
headers: { 'Content-Type': 'application/json' },
|
|
||||||
body: JSON.stringify({
|
|
||||||
rpg_enabled: true,
|
|
||||||
genre: genreValue,
|
|
||||||
rpg_settings_json: JSON.stringify(settings),
|
|
||||||
}),
|
|
||||||
});
|
|
||||||
const res = await fetch('/chat/rpg/bootstrap', {
|
|
||||||
method: 'POST',
|
|
||||||
headers: { 'Content-Type': 'application/json' },
|
|
||||||
body: JSON.stringify({ session_id: sid, persona_id: personaId, genre: genreValue }),
|
|
||||||
});
|
|
||||||
if (res.ok) {
|
|
||||||
const data = await res.json();
|
|
||||||
if (data.quests) updateQuestPanel(data.quests);
|
|
||||||
if (data.plot_arc) {
|
|
||||||
const title = data.plot_arc.title || '';
|
|
||||||
const hint = data.plot_arc.next_beat_hint || '';
|
|
||||||
if (title || hint) addMessage('assistant', `📖 ${title}${hint ? '\n' + hint : ''}`);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function fillNewChatPersonaGrid() {
|
|
||||||
const grid = document.getElementById('newChatPersonaGrid');
|
|
||||||
if (!grid) return;
|
|
||||||
grid.innerHTML = '';
|
|
||||||
newChatPersonaId = currentPersona;
|
|
||||||
for (const p of personaIndex.values()) {
|
|
||||||
const card = document.createElement('button');
|
|
||||||
card.type = 'button';
|
|
||||||
card.className = 'persona-pick-card' + (p.persona_id === newChatPersonaId ? ' selected' : '');
|
|
||||||
card.dataset.id = p.persona_id;
|
|
||||||
card.innerHTML = `<span class="emoji">${p.emoji || '🤖'}</span>${p.name}`;
|
|
||||||
card.addEventListener('click', () => {
|
|
||||||
newChatPersonaId = p.persona_id;
|
|
||||||
grid.querySelectorAll('.persona-pick-card').forEach(c => {
|
|
||||||
c.classList.toggle('selected', c.dataset.id === newChatPersonaId);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
grid.appendChild(card);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function updateNewChatGenresLabel() {
|
|
||||||
const el = document.getElementById('newChatGenresLabel');
|
|
||||||
const nextBtn = document.getElementById('newChatNext');
|
|
||||||
const labels = [...newChatGenres].map(g => GENRE_LABELS[g] || g);
|
|
||||||
if (el) {
|
|
||||||
if (labels.length) {
|
|
||||||
el.textContent = `Выбрано: ${labels.join(' + ')}`;
|
|
||||||
el.classList.remove('hidden');
|
|
||||||
} else {
|
|
||||||
el.classList.add('hidden');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
const isRpg = document.querySelector('input[name="newChatRpg"]:checked')?.value === '1';
|
|
||||||
if (nextBtn && isRpg) {
|
|
||||||
const wizard = newChatModalEl?.querySelector('.modal-wizard');
|
|
||||||
const onStep3 = wizard?.querySelector('.wizard-page[data-step="3"]')?.classList.contains('active');
|
|
||||||
if (onStep3) nextBtn.disabled = newChatGenres.size === 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const newChatModalEl = document.getElementById('newChatModal');
|
|
||||||
let newChatWizard;
|
|
||||||
|
|
||||||
function isNewChatRpg() {
|
|
||||||
return document.querySelector('input[name="newChatRpg"]:checked')?.value === '1';
|
|
||||||
}
|
|
||||||
|
|
||||||
function syncNewChatStep3() {
|
|
||||||
const plain = document.getElementById('newChatPlainStep');
|
|
||||||
const rpg = document.getElementById('newChatRpgStep');
|
|
||||||
if (isNewChatRpg()) {
|
|
||||||
plain?.classList.add('hidden');
|
|
||||||
rpg?.classList.remove('hidden');
|
|
||||||
} else {
|
|
||||||
plain?.classList.remove('hidden');
|
|
||||||
rpg?.classList.add('hidden');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
export function openNewChatWizard() {
|
|
||||||
fillNewChatPersonaGrid();
|
|
||||||
resetGenreGrid(document.getElementById('newChatGenreGrid'), newChatGenres);
|
|
||||||
updateNewChatGenresLabel();
|
|
||||||
document.querySelector('input[name="newChatRpg"][value="0"]')?.click();
|
|
||||||
document.getElementById('newChatTitle').value = '';
|
|
||||||
syncNewChatStep3();
|
|
||||||
newChatWizard?.reset();
|
|
||||||
newChatModalEl?.classList.add('open');
|
|
||||||
}
|
|
||||||
|
|
||||||
export async function createNewChatFromWizard() {
|
|
||||||
const sid = 'sess_' + Math.random().toString(36).slice(2, 10);
|
|
||||||
setSessionId(sid);
|
|
||||||
setCurrentPersona(newChatPersonaId);
|
|
||||||
clearMessages();
|
|
||||||
|
|
||||||
const customTitle = document.getElementById('newChatTitle')?.value.trim();
|
|
||||||
const rpg = isNewChatRpg();
|
|
||||||
|
|
||||||
await fetch(`/sessions/${sid}`, {
|
|
||||||
method: 'PATCH',
|
|
||||||
headers: { 'Content-Type': 'application/json' },
|
|
||||||
body: JSON.stringify({ persona_id: newChatPersonaId, rpg_enabled: rpg }),
|
|
||||||
});
|
|
||||||
|
|
||||||
if (customTitle) {
|
|
||||||
await fetch(`/sessions/${sid}`, {
|
|
||||||
method: 'PATCH',
|
|
||||||
headers: { 'Content-Type': 'application/json' },
|
|
||||||
body: JSON.stringify({ title: customTitle }),
|
|
||||||
});
|
|
||||||
dom.headerTitle.textContent = customTitle;
|
|
||||||
} else {
|
|
||||||
const pName = personaIndex.get(newChatPersonaId)?.name || newChatPersonaId;
|
|
||||||
dom.headerTitle.textContent = rpg ? `${pName} — RPG` : `${pName} — новый чат`;
|
|
||||||
}
|
|
||||||
|
|
||||||
highlightPersona(newChatPersonaId);
|
|
||||||
await initChat();
|
|
||||||
|
|
||||||
if (rpg) {
|
|
||||||
const genreValue = [...newChatGenres].join(',') || 'adventure';
|
|
||||||
const settings = {
|
|
||||||
dice: document.getElementById('ncSettingDice')?.checked ?? true,
|
|
||||||
narrator: document.getElementById('ncSettingNarrator')?.checked ?? true,
|
|
||||||
quests: document.getElementById('ncSettingQuests')?.checked ?? true,
|
|
||||||
affinity: document.getElementById('ncSettingAffinity')?.checked ?? true,
|
|
||||||
choices: document.getElementById('ncSettingChoices')?.checked ?? true,
|
|
||||||
};
|
|
||||||
await bootstrapRpg(sid, newChatPersonaId, genreValue, settings);
|
|
||||||
}
|
|
||||||
|
|
||||||
await reloadChatFromServer(sid);
|
|
||||||
|
|
||||||
newChatModalEl?.classList.remove('open');
|
|
||||||
const sessionRes = await fetch(`/sessions/${sid}`);
|
|
||||||
if (sessionRes.ok) applySessionUi(await sessionRes.json());
|
|
||||||
await loadSessions();
|
|
||||||
}
|
|
||||||
|
|
||||||
export function initNewChatWizard() {
|
|
||||||
if (!newChatModalEl) return;
|
|
||||||
|
|
||||||
newChatWizard = initWizard(newChatModalEl.querySelector('.modal-wizard'), {
|
|
||||||
totalSteps: 3,
|
|
||||||
onStepChange(step) {
|
|
||||||
syncNewChatStep3();
|
|
||||||
const nextBtn = document.getElementById('newChatNext');
|
|
||||||
if (step === 3 && isNewChatRpg()) {
|
|
||||||
nextBtn.disabled = newChatGenres.size === 0;
|
|
||||||
} else {
|
|
||||||
nextBtn.disabled = false;
|
|
||||||
}
|
|
||||||
},
|
|
||||||
validateStep(step) {
|
|
||||||
if (step === 1 && !newChatPersonaId) {
|
|
||||||
alert('Выбери персонажа');
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
if (step === 3 && isNewChatRpg() && newChatGenres.size === 0) {
|
|
||||||
alert('Выбери хотя бы один жанр');
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
bindGenreGrid(document.getElementById('newChatGenreGrid'), newChatGenres, updateNewChatGenresLabel);
|
|
||||||
|
|
||||||
document.getElementById('newChatCancel')?.addEventListener('click', () => {
|
|
||||||
newChatModalEl.classList.remove('open');
|
|
||||||
newChatWizard.reset();
|
|
||||||
});
|
|
||||||
document.getElementById('newChatCreate')?.addEventListener('click', createNewChatFromWizard);
|
|
||||||
}
|
|
||||||
|
|
||||||
function updateChatSettingsGenresLabel() {
|
|
||||||
const el = document.getElementById('chatSettingsGenresLabel');
|
|
||||||
const labels = [...chatSettingsGenres].map(g => GENRE_LABELS[g] || g);
|
|
||||||
if (!el) return;
|
|
||||||
if (labels.length) {
|
|
||||||
el.textContent = `Выбрано: ${labels.join(' + ')}`;
|
|
||||||
el.classList.remove('hidden');
|
|
||||||
} else {
|
|
||||||
el.classList.add('hidden');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function loadRpgSettingsToDom(prefix, settings) {
|
|
||||||
document.getElementById(`${prefix}SettingDice`).checked = settings.dice !== false;
|
|
||||||
document.getElementById(`${prefix}SettingNarrator`).checked = settings.narrator !== false;
|
|
||||||
document.getElementById(`${prefix}SettingQuests`).checked = settings.quests !== false;
|
|
||||||
document.getElementById(`${prefix}SettingAffinity`).checked = settings.affinity !== false;
|
|
||||||
document.getElementById(`${prefix}SettingChoices`).checked = settings.choices !== false;
|
|
||||||
}
|
|
||||||
|
|
||||||
function readRpgSettingsFromDom(prefix) {
|
|
||||||
return {
|
|
||||||
dice: document.getElementById(`${prefix}SettingDice`)?.checked ?? true,
|
|
||||||
narrator: document.getElementById(`${prefix}SettingNarrator`)?.checked ?? true,
|
|
||||||
quests: document.getElementById(`${prefix}SettingQuests`)?.checked ?? true,
|
|
||||||
affinity: document.getElementById(`${prefix}SettingAffinity`)?.checked ?? true,
|
|
||||||
choices: document.getElementById(`${prefix}SettingChoices`)?.checked ?? true,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
export async function openChatSettings() {
|
|
||||||
if (!sessionId) return;
|
|
||||||
const res = await fetch(`/sessions/${sessionId}`);
|
|
||||||
if (!res.ok) return;
|
|
||||||
const s = await res.json();
|
|
||||||
|
|
||||||
document.getElementById('chatSettingsTitle').value = s.title || '';
|
|
||||||
const rpgOn = !!s.rpg_enabled;
|
|
||||||
document.getElementById('chatSettingsRpg').checked = rpgOn;
|
|
||||||
document.getElementById('chatSettingsRpgBlock').classList.toggle('hidden', !rpgOn);
|
|
||||||
|
|
||||||
chatSettingsGenres.clear();
|
|
||||||
(s.genre || 'adventure').split(',').forEach(g => {
|
|
||||||
const t = g.trim();
|
|
||||||
if (t) chatSettingsGenres.add(t);
|
|
||||||
});
|
|
||||||
resetGenreGrid(document.getElementById('chatSettingsGenreGrid'), chatSettingsGenres);
|
|
||||||
document.getElementById('chatSettingsGenreGrid')?.querySelectorAll('.genre-btn').forEach(btn => {
|
|
||||||
if (chatSettingsGenres.has(btn.dataset.genre)) btn.classList.add('selected');
|
|
||||||
});
|
|
||||||
updateChatSettingsGenresLabel();
|
|
||||||
|
|
||||||
let settings = {};
|
|
||||||
try { settings = JSON.parse(s.rpg_settings_json || '{}'); } catch { /* ignore */ }
|
|
||||||
loadRpgSettingsToDom('cs', settings);
|
|
||||||
|
|
||||||
let phase = '';
|
|
||||||
try {
|
|
||||||
const arc = JSON.parse(s.plot_arc_json || '{}');
|
|
||||||
phase = arc.phase || '';
|
|
||||||
} catch { /* ignore */ }
|
|
||||||
document.getElementById('chatSettingsMeta').innerHTML = [
|
|
||||||
`Симпатия: ${s.affinity ?? 0}`,
|
|
||||||
s.genre ? `Жанр: ${(s.genre || '').split(',').map(g => GENRE_LABELS[g.trim()] || g).join(' + ')}` : '',
|
|
||||||
phase ? `Фаза арки: ${phase}` : '',
|
|
||||||
].filter(Boolean).join('<br>');
|
|
||||||
|
|
||||||
document.getElementById('chatSettingsModal').classList.add('open');
|
|
||||||
}
|
|
||||||
|
|
||||||
export function initChatSettings() {
|
|
||||||
bindGenreGrid(
|
|
||||||
document.getElementById('chatSettingsGenreGrid'),
|
|
||||||
chatSettingsGenres,
|
|
||||||
updateChatSettingsGenresLabel,
|
|
||||||
);
|
|
||||||
|
|
||||||
document.getElementById('chatSettingsRpg')?.addEventListener('change', (e) => {
|
|
||||||
document.getElementById('chatSettingsRpgBlock').classList.toggle('hidden', !e.target.checked);
|
|
||||||
});
|
|
||||||
|
|
||||||
document.getElementById('chatSettingsCancel')?.addEventListener('click', () => {
|
|
||||||
document.getElementById('chatSettingsModal').classList.remove('open');
|
|
||||||
});
|
|
||||||
|
|
||||||
document.getElementById('chatSettingsSave')?.addEventListener('click', async () => {
|
|
||||||
if (!sessionId) return;
|
|
||||||
const title = document.getElementById('chatSettingsTitle').value.trim();
|
|
||||||
const rpgOn = document.getElementById('chatSettingsRpg').checked;
|
|
||||||
const genreValue = [...chatSettingsGenres].join(',') || 'adventure';
|
|
||||||
const settings = readRpgSettingsFromDom('cs');
|
|
||||||
|
|
||||||
const body = {
|
|
||||||
title: title || undefined,
|
|
||||||
rpg_enabled: rpgOn,
|
|
||||||
genre: genreValue,
|
|
||||||
rpg_settings_json: JSON.stringify(settings),
|
|
||||||
};
|
|
||||||
await fetch(`/sessions/${sessionId}`, {
|
|
||||||
method: 'PATCH',
|
|
||||||
headers: { 'Content-Type': 'application/json' },
|
|
||||||
body: JSON.stringify(body),
|
|
||||||
});
|
|
||||||
|
|
||||||
if (rpgOn) {
|
|
||||||
const sessionRes = await fetch(`/sessions/${sessionId}`);
|
|
||||||
const s = sessionRes.ok ? await sessionRes.json() : {};
|
|
||||||
let arc = {};
|
|
||||||
try { arc = JSON.parse(s.plot_arc_json || '{}'); } catch { /* ignore */ }
|
|
||||||
if (!arc || !Object.keys(arc).length) {
|
|
||||||
await bootstrapRpg(sessionId, currentPersona, genreValue, settings);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
document.getElementById('chatSettingsModal').classList.remove('open');
|
|
||||||
const updated = await (await fetch(`/sessions/${sessionId}`)).json();
|
|
||||||
applySessionUi(updated);
|
|
||||||
dom.headerTitle.textContent = updated.title || 'Новый чат';
|
|
||||||
await loadSessions();
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
export async function initSessions() {
|
export async function initSessions() {
|
||||||
await loadSessions();
|
await loadSessions();
|
||||||
if (sessionId) {
|
if (sessionId) {
|
||||||
@@ -453,4 +140,58 @@ export async function initSessions() {
|
|||||||
} else {
|
} else {
|
||||||
openNewChatWizard();
|
openNewChatWizard();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
dom.systemBlobRefresh?.addEventListener('click', async () => {
|
||||||
|
if (!sessionId) return;
|
||||||
|
dom.systemBlobRefresh.classList.add('spinning');
|
||||||
|
try {
|
||||||
|
const res = await fetch(`/chat/system/${sessionId}`);
|
||||||
|
if (res.ok) renderSystemBlob(await res.json());
|
||||||
|
} finally {
|
||||||
|
dom.systemBlobRefresh.classList.remove('spinning');
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
let _prevBlobSections = {};
|
||||||
|
|
||||||
|
function renderSystemBlob(blob) {
|
||||||
|
const tryFmt = (str, fallback = '') => {
|
||||||
|
try { return JSON.stringify(JSON.parse(str), null, 2); } catch { return str || fallback; }
|
||||||
|
};
|
||||||
|
|
||||||
|
const questLines = (blob.quests || []).map(q => {
|
||||||
|
const icon = q.status === 'done' ? '✓' : q.status === 'failed' ? '✗' : '◆';
|
||||||
|
return ` ${icon} [${q.status}] ${q.title}`;
|
||||||
|
}).join('\n');
|
||||||
|
|
||||||
|
const sections = {
|
||||||
|
system_prompt: blob.system_prompt ? `[system_prompt]\n${blob.system_prompt}` : '',
|
||||||
|
status_quo: blob.status_quo ? `[status_quo]\n${blob.status_quo}` : '',
|
||||||
|
affinity: blob.affinity != null ? `[affinity] ${blob.affinity}` : '',
|
||||||
|
genre: blob.genre ? `[genre] ${blob.genre}` : '',
|
||||||
|
rpg_settings: blob.rpg_settings_json && blob.rpg_settings_json !== '{}' ? `[rpg_settings]\n${tryFmt(blob.rpg_settings_json)}` : '',
|
||||||
|
outfit: blob.outfit_json && blob.outfit_json !== '[]' ? `[outfit]\n${tryFmt(blob.outfit_json)}` : '',
|
||||||
|
facts: blob.facts_json && blob.facts_json !== '[]' ? `[facts]\n${tryFmt(blob.facts_json)}` : '',
|
||||||
|
plot_arc: blob.plot_arc_json && blob.plot_arc_json !== '{}' ? `[plot_arc]\n${tryFmt(blob.plot_arc_json)}` : '',
|
||||||
|
quests: questLines ? `[quests]\n${questLines}` : '',
|
||||||
|
};
|
||||||
|
|
||||||
|
const el = dom.systemBlobContent;
|
||||||
|
el.innerHTML = '';
|
||||||
|
|
||||||
|
for (const [key, text] of Object.entries(sections)) {
|
||||||
|
if (!text) continue;
|
||||||
|
const span = document.createElement('span');
|
||||||
|
span.textContent = text;
|
||||||
|
if (_prevBlobSections[key] && _prevBlobSections[key] !== text) {
|
||||||
|
span.className = 'blob-changed';
|
||||||
|
setTimeout(() => span.classList.remove('blob-changed'), 3000);
|
||||||
|
}
|
||||||
|
el.appendChild(span);
|
||||||
|
el.appendChild(document.createTextNode('\n\n'));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!el.textContent.trim()) el.textContent = '—';
|
||||||
|
_prevBlobSections = { ...sections };
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -29,4 +29,5 @@ export const dom = {
|
|||||||
systemBlob: document.getElementById('systemBlob'),
|
systemBlob: document.getElementById('systemBlob'),
|
||||||
systemBlobContent: document.getElementById('systemBlobContent'),
|
systemBlobContent: document.getElementById('systemBlobContent'),
|
||||||
systemBlobToggle: document.getElementById('systemBlobToggle'),
|
systemBlobToggle: document.getElementById('systemBlobToggle'),
|
||||||
|
systemBlobRefresh: document.getElementById('systemBlobRefresh'),
|
||||||
};
|
};
|
||||||
|
|||||||
+36
-5
@@ -30,7 +30,9 @@ export function initWizard(modalEl, { totalSteps, onStepChange, validateStep })
|
|||||||
const dots = modalEl.querySelectorAll('.wizard-step-dot');
|
const dots = modalEl.querySelectorAll('.wizard-step-dot');
|
||||||
const prevBtn = modalEl.querySelector('[id$="Prev"]');
|
const prevBtn = modalEl.querySelector('[id$="Prev"]');
|
||||||
const nextBtn = modalEl.querySelector('[id$="Next"]');
|
const nextBtn = modalEl.querySelector('[id$="Next"]');
|
||||||
const saveBtn = modalEl.querySelector('[id$="Save"], [id$="Confirm"], [id$="Create"]');
|
const saveBtn = modalEl.querySelector(
|
||||||
|
'[id$="Save"], [id$="Confirm"], [id$="Create"], [id$="Import"]',
|
||||||
|
);
|
||||||
|
|
||||||
function render() {
|
function render() {
|
||||||
pages.forEach(p => p.classList.toggle('active', Number(p.dataset.step) === step));
|
pages.forEach(p => p.classList.toggle('active', Number(p.dataset.step) === step));
|
||||||
@@ -45,14 +47,17 @@ export function initWizard(modalEl, { totalSteps, onStepChange, validateStep })
|
|||||||
onStepChange?.(step);
|
onStepChange?.(step);
|
||||||
}
|
}
|
||||||
|
|
||||||
function goTo(next) {
|
async function goTo(next) {
|
||||||
if (next > step && validateStep && !validateStep(step)) return;
|
if (next > step && validateStep) {
|
||||||
|
const ok = await Promise.resolve(validateStep(step));
|
||||||
|
if (!ok) return;
|
||||||
|
}
|
||||||
step = Math.max(1, Math.min(totalSteps, next));
|
step = Math.max(1, Math.min(totalSteps, next));
|
||||||
render();
|
render();
|
||||||
}
|
}
|
||||||
|
|
||||||
prevBtn?.addEventListener('click', () => goTo(step - 1));
|
prevBtn?.addEventListener('click', () => { goTo(step - 1); });
|
||||||
nextBtn?.addEventListener('click', () => goTo(step + 1));
|
nextBtn?.addEventListener('click', () => { goTo(step + 1); });
|
||||||
|
|
||||||
render();
|
render();
|
||||||
|
|
||||||
@@ -96,6 +101,32 @@ export function getRpgSettingsFromDom(prefix = '') {
|
|||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export function fillGreetingSelect(selectEl, firstMes, alternates = []) {
|
||||||
|
if (!selectEl) return;
|
||||||
|
selectEl.innerHTML = '';
|
||||||
|
const main = document.createElement('option');
|
||||||
|
main.value = '0';
|
||||||
|
const mainPreview = (firstMes || '').replace(/\s+/g, ' ').trim();
|
||||||
|
main.textContent = mainPreview
|
||||||
|
? `Основное: ${mainPreview.slice(0, 50)}${mainPreview.length > 50 ? '…' : ''}`
|
||||||
|
: 'Основное (first_mes)';
|
||||||
|
selectEl.appendChild(main);
|
||||||
|
alternates.forEach((text, i) => {
|
||||||
|
const opt = document.createElement('option');
|
||||||
|
opt.value = String(i + 1);
|
||||||
|
const preview = String(text).replace(/\s+/g, ' ').trim();
|
||||||
|
opt.textContent = `Альт. ${i + 1}: ${preview.slice(0, 50)}${preview.length > 50 ? '…' : ''}`;
|
||||||
|
selectEl.appendChild(opt);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
export function getSelectedGreeting(selectEl, firstMes, alternates = []) {
|
||||||
|
const v = selectEl?.value ?? '0';
|
||||||
|
if (v === '0') return firstMes || '';
|
||||||
|
const idx = parseInt(v, 10) - 1;
|
||||||
|
return alternates[idx] ?? firstMes ?? '';
|
||||||
|
}
|
||||||
|
|
||||||
export function formatSessionDate(iso) {
|
export function formatSessionDate(iso) {
|
||||||
if (!iso) return '';
|
if (!iso) return '';
|
||||||
const d = new Date(iso.includes('T') ? iso : iso.replace(' ', 'T') + 'Z');
|
const d = new Date(iso.includes('T') ? iso : iso.replace(' ', 'T') + 'Z');
|
||||||
|
|||||||
@@ -0,0 +1,30 @@
|
|||||||
|
import asyncio, httpx, os, json
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
load_dotenv()
|
||||||
|
|
||||||
|
KEY = os.getenv("ROUTER_KEY")
|
||||||
|
URL = "https://openrouter.ai/api/v1/chat/completions"
|
||||||
|
|
||||||
|
async def test():
|
||||||
|
# Minimal test
|
||||||
|
payload = {
|
||||||
|
"model": "google/gemini-2.5-flash",
|
||||||
|
"messages": [{"role": "user", "content": "Say hi"}],
|
||||||
|
"stream": True,
|
||||||
|
}
|
||||||
|
headers = {"Authorization": f"Bearer {KEY}", "Content-Type": "application/json"}
|
||||||
|
async with httpx.AsyncClient(timeout=30) as c:
|
||||||
|
async with c.stream("POST", URL, headers=headers, json=payload) as r:
|
||||||
|
print("status:", r.status_code)
|
||||||
|
async for line in r.aiter_lines():
|
||||||
|
if line.startswith("data: ") and line[6:] != "[DONE]":
|
||||||
|
d = json.loads(line[6:])
|
||||||
|
content = d.get("choices", [{}])[0].get("delta", {}).get("content", "")
|
||||||
|
if content:
|
||||||
|
print("chunk:", repr(content))
|
||||||
|
return
|
||||||
|
if d.get("error"):
|
||||||
|
print("ERROR:", d["error"])
|
||||||
|
return
|
||||||
|
|
||||||
|
asyncio.run(test())
|
||||||
@@ -0,0 +1,40 @@
|
|||||||
|
import asyncio, httpx, os, json, sqlite3
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
load_dotenv()
|
||||||
|
|
||||||
|
KEY = os.getenv("ROUTER_KEY")
|
||||||
|
URL = "https://openrouter.ai/api/v1/chat/completions"
|
||||||
|
|
||||||
|
# Get actual messages from the new session
|
||||||
|
db = sqlite3.connect("data/chat.db")
|
||||||
|
db.row_factory = sqlite3.Row
|
||||||
|
rows = db.execute(
|
||||||
|
"SELECT role, content FROM messages WHERE session_id = (SELECT session_id FROM sessions ORDER BY updated_at DESC LIMIT 1) ORDER BY id"
|
||||||
|
).fetchall()
|
||||||
|
db.close()
|
||||||
|
|
||||||
|
messages = [{"role": r["role"], "content": r["content"]} for r in rows if (r["content"] or "").strip()]
|
||||||
|
print(f"Total messages: {len(messages)}")
|
||||||
|
for i, m in enumerate(messages):
|
||||||
|
print(f" [{i}] {m['role']} len={len(m['content'])} preview={repr(m['content'][:80])}")
|
||||||
|
|
||||||
|
async def test():
|
||||||
|
payload = {"model": "google/gemini-2.5-flash", "messages": messages, "stream": True}
|
||||||
|
headers = {"Authorization": f"Bearer {KEY}", "Content-Type": "application/json"}
|
||||||
|
async with httpx.AsyncClient(timeout=30) as c:
|
||||||
|
async with c.stream("POST", URL, headers=headers, json=payload) as r:
|
||||||
|
print("status:", r.status_code)
|
||||||
|
async for line in r.aiter_lines():
|
||||||
|
if not line.startswith("data: "): continue
|
||||||
|
d = line[6:]
|
||||||
|
if d == "[DONE]": break
|
||||||
|
parsed = json.loads(d)
|
||||||
|
if parsed.get("error"):
|
||||||
|
print("ERROR:", json.dumps(parsed["error"], indent=2))
|
||||||
|
return
|
||||||
|
content = parsed.get("choices", [{}])[0].get("delta", {}).get("content", "")
|
||||||
|
if content:
|
||||||
|
print("OK, got chunk:", repr(content[:50]))
|
||||||
|
return
|
||||||
|
|
||||||
|
asyncio.run(test())
|
||||||
@@ -0,0 +1,45 @@
|
|||||||
|
import asyncio, httpx, os, json, sqlite3
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
load_dotenv()
|
||||||
|
|
||||||
|
KEY = os.getenv("ROUTER_KEY")
|
||||||
|
URL = "https://openrouter.ai/api/v1/chat/completions"
|
||||||
|
|
||||||
|
db = sqlite3.connect("data/chat.db")
|
||||||
|
db.row_factory = sqlite3.Row
|
||||||
|
rows = db.execute(
|
||||||
|
"SELECT role, content FROM messages WHERE session_id = (SELECT session_id FROM sessions ORDER BY updated_at DESC LIMIT 1) ORDER BY id"
|
||||||
|
).fetchall()
|
||||||
|
db.close()
|
||||||
|
|
||||||
|
all_msgs = [{"role": r["role"], "content": r["content"]} for r in rows if (r["content"] or "").strip()]
|
||||||
|
|
||||||
|
async def try_msgs(msgs, label):
|
||||||
|
payload = {"model": "google/gemini-2.5-flash", "messages": msgs, "stream": True}
|
||||||
|
headers = {"Authorization": f"Bearer {KEY}", "Content-Type": "application/json"}
|
||||||
|
async with httpx.AsyncClient(timeout=30) as c:
|
||||||
|
async with c.stream("POST", URL, headers=headers, json=payload) as r:
|
||||||
|
async for line in r.aiter_lines():
|
||||||
|
if not line.startswith("data: "): continue
|
||||||
|
d = line[6:]
|
||||||
|
if d == "[DONE]": break
|
||||||
|
parsed = json.loads(d)
|
||||||
|
if parsed.get("error"):
|
||||||
|
print(f"[{label}] ERROR: {parsed['error']['message']}")
|
||||||
|
return
|
||||||
|
if parsed.get("choices", [{}])[0].get("delta", {}).get("content"):
|
||||||
|
print(f"[{label}] OK")
|
||||||
|
return
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
# Test subsets to find which message causes the error
|
||||||
|
await try_msgs(all_msgs[1:], "no system")
|
||||||
|
await try_msgs([all_msgs[0], all_msgs[2]], "system+user only")
|
||||||
|
await try_msgs([{"role": "user", "content": all_msgs[2]["content"]}], "user only")
|
||||||
|
# Print full system prompt
|
||||||
|
print("\n--- system prompt last 500 chars ---")
|
||||||
|
print(repr(all_msgs[0]["content"][-500:]))
|
||||||
|
print("\n--- user content ---")
|
||||||
|
print(repr(all_msgs[2]["content"]))
|
||||||
|
|
||||||
|
asyncio.run(main())
|
||||||
Reference in New Issue
Block a user