Fixed RPG
This commit is contained in:
+99
-2
@@ -13,7 +13,13 @@ async def init_db():
|
||||
persona_id TEXT DEFAULT 'default',
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
title TEXT DEFAULT 'Новый чат'
|
||||
title TEXT DEFAULT 'Новый чат',
|
||||
rpg_enabled INTEGER DEFAULT 0,
|
||||
facts_json TEXT DEFAULT '[]',
|
||||
global_plot TEXT DEFAULT '',
|
||||
status_quo TEXT DEFAULT '',
|
||||
plot_arc_json TEXT DEFAULT '{}',
|
||||
rng_seed INTEGER
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS messages (
|
||||
@@ -38,7 +44,13 @@ async def init_db():
|
||||
sd_enabled INTEGER DEFAULT 0,
|
||||
lora_name TEXT DEFAULT '',
|
||||
lora_weight REAL DEFAULT 0.8,
|
||||
appearance_tags TEXT DEFAULT ''
|
||||
appearance_tags TEXT DEFAULT '',
|
||||
personality TEXT DEFAULT '',
|
||||
scenario TEXT DEFAULT '',
|
||||
first_mes TEXT DEFAULT '',
|
||||
mes_example TEXT DEFAULT '',
|
||||
lorebook_json TEXT DEFAULT '[]',
|
||||
avatar_path TEXT DEFAULT ''
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS characters (
|
||||
@@ -54,10 +66,16 @@ async def init_db():
|
||||
lora_weight REAL DEFAULT 0.8,
|
||||
appearance_tags TEXT DEFAULT '',
|
||||
lorebook_json TEXT DEFAULT '[]',
|
||||
avatar_path TEXT DEFAULT '',
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
||||
""")
|
||||
await _migrate_messages_columns(db)
|
||||
await _migrate_personas_columns(db)
|
||||
await _migrate_sessions_columns(db)
|
||||
await _migrate_characters_columns(db)
|
||||
await _migrate_rpg_quests(db)
|
||||
await _migrate_action_resolutions(db)
|
||||
await db.commit()
|
||||
|
||||
|
||||
@@ -68,3 +86,82 @@ async def _migrate_messages_columns(db):
|
||||
await db.execute("ALTER TABLE messages ADD COLUMN image_prompt TEXT")
|
||||
if "image_path" not in cols:
|
||||
await db.execute("ALTER TABLE messages ADD COLUMN image_path TEXT")
|
||||
|
||||
|
||||
async def _migrate_personas_columns(db):
|
||||
async with db.execute("PRAGMA table_info(personas)") as cur:
|
||||
cols = {row[1] for row in await cur.fetchall()}
|
||||
if "personality" not in cols:
|
||||
await db.execute("ALTER TABLE personas ADD COLUMN personality TEXT DEFAULT ''")
|
||||
if "scenario" not in cols:
|
||||
await db.execute("ALTER TABLE personas ADD COLUMN scenario TEXT DEFAULT ''")
|
||||
if "first_mes" not in cols:
|
||||
await db.execute("ALTER TABLE personas ADD COLUMN first_mes TEXT DEFAULT ''")
|
||||
if "mes_example" not in cols:
|
||||
await db.execute("ALTER TABLE personas ADD COLUMN mes_example TEXT DEFAULT ''")
|
||||
if "lorebook_json" not in cols:
|
||||
await db.execute("ALTER TABLE personas ADD COLUMN lorebook_json TEXT DEFAULT '[]'")
|
||||
if "avatar_path" not in cols:
|
||||
await db.execute("ALTER TABLE personas ADD COLUMN avatar_path TEXT DEFAULT ''")
|
||||
|
||||
|
||||
async def _migrate_sessions_columns(db):
|
||||
async with db.execute("PRAGMA table_info(sessions)") as cur:
|
||||
cols = {row[1] for row in await cur.fetchall()}
|
||||
if "rpg_enabled" not in cols:
|
||||
await db.execute("ALTER TABLE sessions ADD COLUMN rpg_enabled INTEGER DEFAULT 0")
|
||||
if "facts_json" not in cols:
|
||||
await db.execute("ALTER TABLE sessions ADD COLUMN facts_json TEXT DEFAULT '[]'")
|
||||
if "global_plot" not in cols:
|
||||
await db.execute("ALTER TABLE sessions ADD COLUMN global_plot TEXT DEFAULT ''")
|
||||
if "status_quo" not in cols:
|
||||
await db.execute("ALTER TABLE sessions ADD COLUMN status_quo TEXT DEFAULT ''")
|
||||
if "plot_arc_json" not in cols:
|
||||
await db.execute("ALTER TABLE sessions ADD COLUMN plot_arc_json TEXT DEFAULT '{}'")
|
||||
if "rng_seed" not in cols:
|
||||
await db.execute("ALTER TABLE sessions ADD COLUMN rng_seed INTEGER")
|
||||
if "affinity" not in cols:
|
||||
await db.execute("ALTER TABLE sessions ADD COLUMN affinity INTEGER DEFAULT 0")
|
||||
if "genre" not in cols:
|
||||
await db.execute("ALTER TABLE sessions ADD COLUMN genre TEXT DEFAULT 'adventure'")
|
||||
if "rpg_settings_json" not in cols:
|
||||
await db.execute("ALTER TABLE sessions ADD COLUMN rpg_settings_json TEXT DEFAULT '{}'")
|
||||
|
||||
|
||||
async def _migrate_rpg_quests(db):
|
||||
await db.executescript("""
|
||||
CREATE TABLE IF NOT EXISTS rpg_quests (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
session_id TEXT NOT NULL,
|
||||
title TEXT NOT NULL,
|
||||
status TEXT DEFAULT 'active',
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
||||
CREATE INDEX IF NOT EXISTS idx_rpg_quests_session ON rpg_quests(session_id);
|
||||
""")
|
||||
|
||||
|
||||
async def _migrate_action_resolutions(db):
|
||||
await db.executescript(
|
||||
"""
|
||||
CREATE TABLE IF NOT EXISTS action_resolutions (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
session_id TEXT NOT NULL,
|
||||
message_id INTEGER,
|
||||
intent_text TEXT NOT NULL,
|
||||
roll INTEGER NOT NULL,
|
||||
outcome TEXT NOT NULL,
|
||||
resolution_text TEXT NOT NULL,
|
||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||||
);
|
||||
CREATE INDEX IF NOT EXISTS idx_action_resolutions_session
|
||||
ON action_resolutions(session_id);
|
||||
"""
|
||||
)
|
||||
|
||||
|
||||
async def _migrate_characters_columns(db):
|
||||
async with db.execute("PRAGMA table_info(characters)") as cur:
|
||||
cols = {row[1] for row in await cur.fetchall()}
|
||||
if "avatar_path" not in cols:
|
||||
await db.execute("ALTER TABLE characters ADD COLUMN avatar_path TEXT DEFAULT ''")
|
||||
|
||||
+30
-1
@@ -5,6 +5,23 @@ class ChatRequest(BaseModel):
|
||||
message: str
|
||||
session_id: str
|
||||
persona_id: Optional[str] = "default"
|
||||
is_narrator_choice: bool = False
|
||||
skip_user_add: bool = False
|
||||
|
||||
|
||||
class MessageEditRequest(BaseModel):
|
||||
content: str
|
||||
truncate_after: bool = False
|
||||
|
||||
|
||||
class RegenerateRequest(BaseModel):
|
||||
session_id: str
|
||||
persona_id: Optional[str] = "default"
|
||||
message_id: Optional[int] = None
|
||||
|
||||
|
||||
class ForkSessionRequest(BaseModel):
|
||||
until_message_id: int
|
||||
|
||||
class ChatResponse(BaseModel):
|
||||
reply: str
|
||||
@@ -16,11 +33,17 @@ class PersonaCreate(BaseModel):
|
||||
name: str
|
||||
emoji: str = "🤖"
|
||||
description: str = ""
|
||||
prompt: str
|
||||
prompt: str = ""
|
||||
sd_enabled: bool = False
|
||||
lora_name: str = ""
|
||||
lora_weight: float = 0.8
|
||||
appearance_tags: str = ""
|
||||
personality: str = ""
|
||||
scenario: str = ""
|
||||
first_mes: str = ""
|
||||
mes_example: str = ""
|
||||
lorebook_json: str = "[]"
|
||||
avatar_path: str = ""
|
||||
|
||||
class PersonaResponse(BaseModel):
|
||||
persona_id: str
|
||||
@@ -33,3 +56,9 @@ class PersonaResponse(BaseModel):
|
||||
lora_name: str = ""
|
||||
lora_weight: float = 0.8
|
||||
appearance_tags: str = ""
|
||||
personality: str = ""
|
||||
scenario: str = ""
|
||||
first_mes: str = ""
|
||||
mes_example: str = ""
|
||||
lorebook_json: str = "[]"
|
||||
avatar_path: str = ""
|
||||
|
||||
@@ -56,6 +56,23 @@ async def patch_card(card_id: str, body: CardPatch):
|
||||
return await get_character(card_id)
|
||||
|
||||
|
||||
@router.post("/{card_id}/avatar")
|
||||
async def upload_avatar(card_id: str, file: UploadFile = File(...)):
|
||||
card = await get_character(card_id)
|
||||
if not card:
|
||||
raise HTTPException(status_code=404, detail="Карточка не найдена")
|
||||
content = await file.read()
|
||||
if not content.startswith(b"\x89PNG"):
|
||||
raise HTTPException(status_code=400, detail="Нужен PNG")
|
||||
from services.character_card import _save_avatar_bytes
|
||||
rel = _save_avatar_bytes(content, f"card_{card_id}")
|
||||
await update_character(card_id, {"avatar_path": rel})
|
||||
# sync persona
|
||||
from services.personas import patch_persona
|
||||
await patch_persona(f"card_{card_id}", {"avatar_path": rel})
|
||||
return {"avatar_path": f"/static/{rel}"}
|
||||
|
||||
|
||||
@router.post("/import")
|
||||
async def import_card(
|
||||
file: UploadFile = File(...),
|
||||
|
||||
+372
-10
@@ -1,22 +1,38 @@
|
||||
import json
|
||||
import os
|
||||
import random
|
||||
|
||||
import aiosqlite
|
||||
from fastapi import APIRouter
|
||||
from fastapi import APIRouter, HTTPException
|
||||
from fastapi.responses import StreamingResponse
|
||||
from pydantic import BaseModel
|
||||
|
||||
from database.db import DB_PATH
|
||||
from models.schemas import ChatRequest, ChatResponse
|
||||
from models.schemas import ChatRequest, ChatResponse, MessageEditRequest, RegenerateRequest
|
||||
from services.llm import send_message, stream_message
|
||||
from services.memory import (
|
||||
get_history,
|
||||
add_message,
|
||||
clear_history,
|
||||
get_or_create_session,
|
||||
get_session,
|
||||
update_session_title,
|
||||
update_session_persona,
|
||||
get_message_count,
|
||||
get_last_assistant_message_id,
|
||||
update_message_image,
|
||||
update_session_facts,
|
||||
update_session_status_quo,
|
||||
update_session_affinity,
|
||||
update_session_genre,
|
||||
update_session_rpg_settings,
|
||||
upsert_quest,
|
||||
get_quests,
|
||||
add_action_resolution,
|
||||
get_message,
|
||||
update_message_content,
|
||||
delete_messages_after,
|
||||
delete_message,
|
||||
)
|
||||
from services.personas import get_persona
|
||||
from services.sd_prompt import (
|
||||
@@ -27,12 +43,51 @@ from services.sd_prompt import (
|
||||
from services.lorebook import get_lorebook_context
|
||||
from services.character_card import get_character
|
||||
from services import sdbackend as sd_service
|
||||
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_narrator import narrator_pre, narrator_post
|
||||
|
||||
router = APIRouter(prefix="/chat", tags=["chat"])
|
||||
|
||||
DEFAULT_PROMPT = "Ты — полезный AI ассистент. Отвечай чётко и по делу."
|
||||
SD_AUTO_GENERATE = os.getenv("SD_AUTO_GENERATE", "false").lower() in ("1", "true", "yes")
|
||||
|
||||
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 ---\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:
|
||||
try:
|
||||
s = json.loads(session.get("rpg_settings_json") or "{}")
|
||||
return {**DEFAULT_RPG_SETTINGS, **s}
|
||||
except Exception:
|
||||
return DEFAULT_RPG_SETTINGS
|
||||
|
||||
|
||||
async def get_system_prompt(persona_id: str, history: list, user_message: str = "") -> str:
|
||||
persona = await get_persona(persona_id)
|
||||
@@ -41,11 +96,17 @@ async def get_system_prompt(persona_id: str, history: list, user_message: str =
|
||||
|
||||
prompt = persona["prompt"]
|
||||
|
||||
if persona.get("lorebook_json"):
|
||||
recent = [m for m in history if m["role"] in ("user", "assistant")][-5:]
|
||||
context = recent + [{"role": "user", "content": user_message}]
|
||||
lore = get_lorebook_context(persona.get("lorebook_json", "[]"), context)
|
||||
if lore:
|
||||
prompt = prompt + "\n\n" + lore
|
||||
|
||||
if persona_id.startswith("card_"):
|
||||
card_id = persona_id[5:]
|
||||
card = await get_character(card_id)
|
||||
if card:
|
||||
# match lorebook against recent context + current message
|
||||
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)
|
||||
@@ -60,20 +121,37 @@ async def get_chat_history(session_id: str):
|
||||
return await get_history(session_id)
|
||||
|
||||
|
||||
@router.get("/system/{session_id}")
|
||||
async def get_system_blob(session_id: str):
|
||||
history = await get_history(session_id)
|
||||
system_msg = next((m for m in history if m.get("role") == "system"), None)
|
||||
session = await get_session(session_id)
|
||||
return {
|
||||
"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 "",
|
||||
"plot_arc_json": session.get("plot_arc_json") if session else "{}",
|
||||
"rpg_enabled": bool(session.get("rpg_enabled")) if session else False,
|
||||
}
|
||||
|
||||
|
||||
@router.post("/init")
|
||||
async def init_chat(request: ChatRequest):
|
||||
"""Called when opening a new chat. Seeds system prompt and first_mes if card persona."""
|
||||
persona_id = request.persona_id or "default"
|
||||
await get_or_create_session(request.session_id, persona_id)
|
||||
history = await get_history(request.session_id)
|
||||
if history:
|
||||
return {"first_mes": None} # already initialized
|
||||
return {"first_mes": None}
|
||||
|
||||
system_prompt = await get_system_prompt(persona_id, [], "")
|
||||
await add_message(request.session_id, "system", system_prompt)
|
||||
|
||||
first_mes = None
|
||||
if persona_id.startswith("card_"):
|
||||
persona = await get_persona(persona_id)
|
||||
if persona and persona.get("first_mes"):
|
||||
first_mes = persona["first_mes"]
|
||||
await add_message(request.session_id, "assistant", first_mes)
|
||||
elif persona_id.startswith("card_"):
|
||||
card = await get_character(persona_id[5:])
|
||||
if card and card.get("first_mes"):
|
||||
first_mes = card["first_mes"]
|
||||
@@ -82,6 +160,50 @@ async def init_chat(request: ChatRequest):
|
||||
return {"first_mes": first_mes}
|
||||
|
||||
|
||||
class RpgBootstrapRequest(BaseModel):
|
||||
session_id: str
|
||||
persona_id: str = "default"
|
||||
genre: str = "adventure"
|
||||
|
||||
|
||||
@router.post("/rpg/bootstrap")
|
||||
async def rpg_bootstrap(req: RpgBootstrapRequest):
|
||||
await get_or_create_session(req.session_id, req.persona_id)
|
||||
session = await get_session(req.session_id)
|
||||
persona = await get_persona(req.persona_id) or {}
|
||||
|
||||
# Save genre
|
||||
await update_session_genre(req.session_id, req.genre)
|
||||
|
||||
arc_json = (session.get("plot_arc_json") or "{}") if session else "{}"
|
||||
try:
|
||||
arc = json.loads(arc_json) if isinstance(arc_json, str) else {}
|
||||
except Exception:
|
||||
arc = {}
|
||||
if not arc:
|
||||
facts_block = facts_to_prompt((session or {}).get("facts_json", "[]"))
|
||||
arc = await generate_plot_arc(
|
||||
persona.get("name", req.persona_id),
|
||||
persona.get("description", ""),
|
||||
persona.get("scenario", ""),
|
||||
persona.get("first_mes", ""),
|
||||
facts_block=facts_block,
|
||||
genre=req.genre,
|
||||
)
|
||||
if arc:
|
||||
from services.memory import update_session_plot_arc
|
||||
await update_session_plot_arc(req.session_id, json.dumps(arc, ensure_ascii=False))
|
||||
|
||||
# Seed quests from beats
|
||||
for beat in arc.get("beats", []):
|
||||
injection = beat.get("injection", "").strip()
|
||||
if injection:
|
||||
await upsert_quest(req.session_id, injection[:120])
|
||||
|
||||
quests = await get_quests(req.session_id)
|
||||
return {"plot_arc": arc, "quests": quests}
|
||||
|
||||
|
||||
@router.post("/stream")
|
||||
async def chat_stream(request: ChatRequest):
|
||||
persona_id = request.persona_id or "default"
|
||||
@@ -89,8 +211,114 @@ async def chat_stream(request: ChatRequest):
|
||||
await get_or_create_session(request.session_id, persona_id)
|
||||
|
||||
history = await get_history(request.session_id)
|
||||
session = await get_session(request.session_id)
|
||||
system_prompt = await get_system_prompt(persona_id, history, request.message)
|
||||
|
||||
arc = {}
|
||||
roll = None
|
||||
outcome = None
|
||||
resolution_text = ""
|
||||
narrator_msg = None # shown as narrator bubble before assistant reply
|
||||
rpg_settings = {}
|
||||
|
||||
if session and session.get("rpg_enabled"):
|
||||
rpg_settings = get_rpg_settings(session)
|
||||
facts_block = facts_to_prompt(session.get("facts_json", "[]"))
|
||||
if facts_block:
|
||||
system_prompt = system_prompt + "\n\n" + facts_block
|
||||
try:
|
||||
arc = json.loads(session.get("plot_arc_json") or "{}")
|
||||
except Exception:
|
||||
arc = {}
|
||||
if arc:
|
||||
system_prompt = system_prompt + "\n\n--- PlotArc ---\n" + json.dumps(
|
||||
{k: arc.get(k) for k in ("title", "phase", "next_beat_hint")}, ensure_ascii=False
|
||||
) + "\n---"
|
||||
status_quo = (session.get("status_quo") or "").strip()
|
||||
if status_quo:
|
||||
system_prompt = system_prompt + "\n\n--- Status quo ---\n" + status_quo + "\n---"
|
||||
if rpg_settings.get("affinity", True):
|
||||
aff = int(session.get("affinity") or 0)
|
||||
system_prompt = system_prompt + affinity_prompt_block(aff)
|
||||
|
||||
if rpg_settings.get("narrator", True):
|
||||
persona = await get_persona(persona_id) or {}
|
||||
recent_txt = "\n".join(
|
||||
f"{m['role']}: {m['content']}" for m in history[-8:]
|
||||
if m.get("role") in ("user", "assistant")
|
||||
)
|
||||
|
||||
# Phase 1: ask narrator if check is needed (no roll yet)
|
||||
pre = await narrator_pre(
|
||||
persona.get("name", persona_id),
|
||||
recent_txt,
|
||||
json.dumps(arc, ensure_ascii=False) if arc else "",
|
||||
facts_block,
|
||||
request.message,
|
||||
)
|
||||
|
||||
needs_check = pre.get("needs_check", False) and rpg_settings.get("dice", True)
|
||||
|
||||
if needs_check:
|
||||
# Phase 2: roll and get resolution
|
||||
roll = random.randint(1, 20)
|
||||
if roll == 1:
|
||||
outcome = "critical failure"
|
||||
elif roll <= 8:
|
||||
outcome = "failure"
|
||||
elif roll >= 20:
|
||||
outcome = "critical success"
|
||||
else:
|
||||
outcome = "success"
|
||||
|
||||
pre2 = await narrator_pre(
|
||||
persona.get("name", persona_id),
|
||||
recent_txt,
|
||||
json.dumps(arc, ensure_ascii=False) if arc else "",
|
||||
facts_block,
|
||||
request.message,
|
||||
roll=roll,
|
||||
outcome=outcome,
|
||||
)
|
||||
resolution_text = (pre2.get("resolution_text") or "").strip()
|
||||
directives = pre2.get("directives") or []
|
||||
pre_sq = (pre2.get("status_quo_update") or "").strip()
|
||||
else:
|
||||
directives = pre.get("directives") or []
|
||||
pre_sq = (pre.get("status_quo_update") or "").strip()
|
||||
|
||||
if directives:
|
||||
system_prompt = system_prompt + "\n\n--- Narrator directives ---\n" + "\n".join(f"- {d}" for d in directives) + "\n---"
|
||||
if pre_sq:
|
||||
await update_session_status_quo(request.session_id, pre_sq)
|
||||
session["status_quo"] = pre_sq
|
||||
|
||||
if resolution_text:
|
||||
await add_action_resolution(
|
||||
request.session_id,
|
||||
intent_text=request.message,
|
||||
roll=roll,
|
||||
outcome=outcome,
|
||||
resolution_text=resolution_text,
|
||||
message_id=None,
|
||||
)
|
||||
narrator_msg = {"roll": roll, "outcome": outcome, "text": resolution_text}
|
||||
|
||||
# Inject outcome into system prompt so character reply is consistent
|
||||
if roll is not None:
|
||||
system_prompt = (
|
||||
system_prompt
|
||||
+ f"\n\n--- Mechanics ---\n"
|
||||
+ f"Roll d20={roll}. Outcome: {outcome}.\n"
|
||||
+ "Your reply MUST be consistent with this outcome. Do NOT contradict the narrator resolution.\n"
|
||||
+ "---"
|
||||
)
|
||||
|
||||
# is_narrator_choice: wrap message so LLM understands context
|
||||
user_message_content = request.message
|
||||
if request.is_narrator_choice:
|
||||
user_message_content = f"[Player chose: {request.message}]"
|
||||
|
||||
if not history:
|
||||
await add_message(request.session_id, "system", system_prompt)
|
||||
elif history[0]["role"] == "system" and history[0]["content"] != system_prompt:
|
||||
@@ -103,12 +331,14 @@ async def chat_stream(request: ChatRequest):
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
await add_message(request.session_id, "user", request.message)
|
||||
if not request.skip_user_add:
|
||||
await add_message(request.session_id, "user", user_message_content)
|
||||
messages = await get_history(request.session_id)
|
||||
|
||||
full_reply = []
|
||||
|
||||
async def generate():
|
||||
nonlocal arc
|
||||
async for chunk in stream_message(
|
||||
[{"role": m["role"], "content": m["content"]} for m in messages]
|
||||
):
|
||||
@@ -133,10 +363,97 @@ async def chat_stream(request: ChatRequest):
|
||||
image_prompt=prompt_str,
|
||||
)
|
||||
|
||||
choices = []
|
||||
debug_blocks = []
|
||||
quests_updated = []
|
||||
|
||||
if session and session.get("rpg_enabled"):
|
||||
if not arc:
|
||||
persona = await get_persona(persona_id) or {}
|
||||
genre = (session.get("genre") or "adventure")
|
||||
arc = await generate_plot_arc(
|
||||
persona.get("name", persona_id),
|
||||
persona.get("description", ""),
|
||||
persona.get("scenario", ""),
|
||||
persona.get("first_mes", ""),
|
||||
facts_block=facts_to_prompt(session.get("facts_json", "[]")),
|
||||
genre=genre,
|
||||
)
|
||||
if arc:
|
||||
from services.memory import update_session_plot_arc
|
||||
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)})
|
||||
if rpg_settings.get("quests", True):
|
||||
for beat in arc.get("beats", []):
|
||||
inj = beat.get("injection", "").strip()
|
||||
if inj:
|
||||
await upsert_quest(request.session_id, inj[:120])
|
||||
|
||||
trig = should_advance_arc(request.message)
|
||||
if trig and arc:
|
||||
arc, beats = pop_matching_beats(arc, trig, max_beats=1)
|
||||
if beats:
|
||||
from services.memory import update_session_plot_arc
|
||||
await update_session_plot_arc(request.session_id, json.dumps(arc, ensure_ascii=False))
|
||||
inj = beats[0].get("injection", "")
|
||||
if inj:
|
||||
debug_blocks.append({"type": "narrator_injection", "text": inj})
|
||||
if rpg_settings.get("choices", True):
|
||||
beat_choices = beats[0].get("choices") or []
|
||||
if beat_choices:
|
||||
choices = choices + beat_choices
|
||||
|
||||
ctx = [m for m in (await get_history(request.session_id)) if m["role"] in ("user", "assistant")][-10:]
|
||||
new_facts = await extract_facts(ctx)
|
||||
if new_facts:
|
||||
merged = merge_facts(session.get("facts_json", "[]"), new_facts)
|
||||
await update_session_facts(request.session_id, merged)
|
||||
session["facts_json"] = merged
|
||||
debug_blocks.append({"type": "facts", "text": facts_to_prompt(merged)})
|
||||
|
||||
persona = await get_persona(persona_id) or {}
|
||||
ctx_txt = "\n".join(
|
||||
f"{m['role']}: {m['content']}" for m in ctx[-8:]
|
||||
if m.get("role") in ("user", "assistant")
|
||||
)
|
||||
post = await narrator_post(
|
||||
persona.get("name", persona_id),
|
||||
ctx_txt,
|
||||
json.dumps(arc, ensure_ascii=False) if arc else "",
|
||||
facts_to_prompt(session.get("facts_json", "[]")),
|
||||
)
|
||||
sq = (post.get("status_quo_update") or "").strip()
|
||||
if sq:
|
||||
await update_session_status_quo(request.session_id, sq)
|
||||
session["status_quo"] = sq
|
||||
debug_blocks.append({"type": "status_quo", "text": f"--- Status quo ---\n{sq}\n---"})
|
||||
|
||||
if rpg_settings.get("choices", True):
|
||||
extra_choices = post.get("choices") or []
|
||||
if extra_choices:
|
||||
choices = choices + extra_choices
|
||||
|
||||
if rpg_settings.get("affinity", True):
|
||||
delta = int(post.get("affinity_delta") or 0)
|
||||
if delta:
|
||||
await update_session_affinity(request.session_id, delta)
|
||||
|
||||
if rpg_settings.get("quests", True):
|
||||
for qu in (post.get("quest_updates") or []):
|
||||
title = (qu.get("title") or "").strip()
|
||||
status = qu.get("status", "active")
|
||||
if title:
|
||||
await upsert_quest(request.session_id, title[:120], status)
|
||||
quests_updated = await get_quests(request.session_id)
|
||||
|
||||
count = await get_message_count(request.session_id)
|
||||
if count == 2:
|
||||
title = request.message[:40] + ("…" if len(request.message) > 40 else "")
|
||||
await update_session_title(request.session_id, title)
|
||||
if count == 2 and not request.skip_user_add:
|
||||
persona = await get_persona(persona_id) or {}
|
||||
persona_name = persona.get("name", persona_id)
|
||||
preview = request.message[:40] + ("…" if len(request.message) > 40 else "")
|
||||
current = (session or {}).get("title") or "Новый чат"
|
||||
if current in ("", "Новый чат"):
|
||||
await update_session_title(request.session_id, f"{persona_name} — {preview}")
|
||||
|
||||
image_path = None
|
||||
image_error = None
|
||||
@@ -150,11 +467,20 @@ async def chat_stream(request: ChatRequest):
|
||||
else:
|
||||
image_error = err
|
||||
|
||||
# Fetch current affinity for UI
|
||||
updated_session = await get_session(request.session_id)
|
||||
affinity = updated_session.get("affinity", 0) if updated_session else 0
|
||||
|
||||
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,
|
||||
'narrator': narrator_msg,
|
||||
'affinity': affinity,
|
||||
'quests': quests_updated,
|
||||
})}\n\n"
|
||||
|
||||
return StreamingResponse(
|
||||
@@ -193,6 +519,42 @@ async def chat(request: ChatRequest):
|
||||
)
|
||||
|
||||
|
||||
@router.patch("/messages/{message_id}")
|
||||
async def edit_message(message_id: int, req: MessageEditRequest):
|
||||
msg = await get_message(message_id)
|
||||
if not msg:
|
||||
raise HTTPException(status_code=404, detail="Сообщение не найдено")
|
||||
await update_message_content(message_id, req.content)
|
||||
if req.truncate_after:
|
||||
await delete_messages_after(msg["session_id"], message_id)
|
||||
return {"status": "updated", "message_id": message_id}
|
||||
|
||||
|
||||
@router.post("/regenerate")
|
||||
async def regenerate_chat(req: RegenerateRequest):
|
||||
msg_id = req.message_id or await get_last_assistant_message_id(req.session_id)
|
||||
if not msg_id:
|
||||
raise HTTPException(status_code=400, detail="Нет сообщения для перегенерации")
|
||||
msg = await get_message(msg_id)
|
||||
if not msg or msg.get("role") != "assistant":
|
||||
raise HTTPException(status_code=400, detail="Неверное сообщение")
|
||||
await delete_message(msg_id)
|
||||
history = await get_history(req.session_id)
|
||||
last_user = next((m for m in reversed(history) if m["role"] == "user"), None)
|
||||
if not last_user:
|
||||
raise HTTPException(status_code=400, detail="Нет сообщения пользователя")
|
||||
user_text = last_user["content"]
|
||||
if user_text.startswith("[Player chose: ") and user_text.endswith("]"):
|
||||
user_text = user_text[15:-1]
|
||||
stream_req = ChatRequest(
|
||||
message=user_text,
|
||||
session_id=req.session_id,
|
||||
persona_id=req.persona_id,
|
||||
skip_user_add=True,
|
||||
)
|
||||
return await chat_stream(stream_req)
|
||||
|
||||
|
||||
@router.delete("/{session_id}")
|
||||
async def clear_chat(session_id: str):
|
||||
await clear_history(session_id)
|
||||
|
||||
+70
-2
@@ -1,6 +1,14 @@
|
||||
from fastapi import APIRouter, HTTPException
|
||||
from fastapi import APIRouter, HTTPException, File, UploadFile
|
||||
from pydantic import BaseModel
|
||||
from typing import Optional
|
||||
from models.schemas import PersonaCreate
|
||||
from services.personas import get_all_personas, get_persona, create_persona, delete_persona
|
||||
from services.personas import (
|
||||
get_all_personas,
|
||||
get_persona,
|
||||
create_persona,
|
||||
delete_persona,
|
||||
patch_persona,
|
||||
)
|
||||
|
||||
router = APIRouter(prefix="/personas", tags=["personas"])
|
||||
|
||||
@@ -31,10 +39,70 @@ async def create_new_persona(data: PersonaCreate):
|
||||
lora_name=data.lora_name,
|
||||
lora_weight=data.lora_weight,
|
||||
appearance_tags=data.appearance_tags,
|
||||
personality=data.personality,
|
||||
scenario=data.scenario,
|
||||
first_mes=data.first_mes,
|
||||
mes_example=data.mes_example,
|
||||
lorebook_json=data.lorebook_json,
|
||||
)
|
||||
return {"persona_id": data.persona_id, **persona}
|
||||
|
||||
|
||||
class PersonaPatch(BaseModel):
|
||||
name: Optional[str] = None
|
||||
emoji: Optional[str] = None
|
||||
description: Optional[str] = None
|
||||
prompt: Optional[str] = None
|
||||
sd_enabled: Optional[bool] = None
|
||||
lora_name: Optional[str] = None
|
||||
lora_weight: Optional[float] = None
|
||||
appearance_tags: Optional[str] = None
|
||||
personality: Optional[str] = None
|
||||
scenario: Optional[str] = None
|
||||
first_mes: Optional[str] = None
|
||||
mes_example: Optional[str] = None
|
||||
lorebook_json: Optional[str] = None
|
||||
avatar_path: Optional[str] = None
|
||||
|
||||
|
||||
@router.patch("/{persona_id}")
|
||||
async def patch_one_persona(persona_id: str, body: PersonaPatch):
|
||||
fields = {k: v for k, v in body.model_dump().items() if v is not None}
|
||||
ok = await patch_persona(persona_id, fields)
|
||||
if not ok:
|
||||
raise HTTPException(status_code=400, detail="Нельзя редактировать этого персонажа")
|
||||
persona = await get_persona(persona_id)
|
||||
if not persona:
|
||||
raise HTTPException(status_code=404, detail="Персонаж не найден")
|
||||
return {"persona_id": persona_id, **persona}
|
||||
|
||||
|
||||
@router.post("/{persona_id}/avatar")
|
||||
async def upload_persona_avatar(persona_id: str, file: UploadFile = File(...)):
|
||||
# only custom personas editable
|
||||
persona = await get_persona(persona_id)
|
||||
if not persona:
|
||||
raise HTTPException(status_code=404, detail="Персонаж не найден")
|
||||
if not persona.get("custom"):
|
||||
raise HTTPException(status_code=400, detail="Нельзя менять аватар встроенного персонажа")
|
||||
content = await file.read()
|
||||
if not content.startswith(b"\x89PNG"):
|
||||
raise HTTPException(status_code=400, detail="Нужен PNG")
|
||||
from pathlib import Path
|
||||
import uuid
|
||||
|
||||
avatars_dir = Path("static/avatars")
|
||||
avatars_dir.mkdir(parents=True, exist_ok=True)
|
||||
fname = f"persona_{persona_id}_{uuid.uuid4().hex[:8]}.png"
|
||||
path = avatars_dir / fname
|
||||
path.write_bytes(content)
|
||||
rel = f"avatars/{fname}"
|
||||
ok = await patch_persona(persona_id, {"avatar_path": rel})
|
||||
if not ok:
|
||||
raise HTTPException(status_code=400, detail="Нельзя изменить аватар")
|
||||
return {"avatar_path": f"/static/{rel}"}
|
||||
|
||||
|
||||
@router.delete("/{persona_id}")
|
||||
async def remove_persona(persona_id: str):
|
||||
if not await delete_persona(persona_id):
|
||||
|
||||
+44
-3
@@ -6,8 +6,19 @@ from services.memory import (
|
||||
update_session_title,
|
||||
update_session_persona,
|
||||
get_history,
|
||||
get_message_count
|
||||
get_message_count,
|
||||
update_session_rpg,
|
||||
update_session_facts,
|
||||
update_session_global_plot,
|
||||
update_session_status_quo,
|
||||
update_session_genre,
|
||||
update_session_rpg_settings,
|
||||
get_quests,
|
||||
get_last_message_preview,
|
||||
fork_session,
|
||||
get_session,
|
||||
)
|
||||
from models.schemas import ForkSessionRequest
|
||||
|
||||
router = APIRouter(prefix="/sessions", tags=["sessions"])
|
||||
|
||||
@@ -18,10 +29,20 @@ async def list_sessions():
|
||||
result = []
|
||||
for s in sessions:
|
||||
count = await get_message_count(s["session_id"])
|
||||
result.append({**s, "message_count": count})
|
||||
preview = await get_last_message_preview(s["session_id"])
|
||||
result.append({
|
||||
**s,
|
||||
"message_count": count,
|
||||
"last_message_preview": preview,
|
||||
})
|
||||
return result
|
||||
|
||||
|
||||
@router.get("/{session_id}/quests")
|
||||
async def list_quests(session_id: str):
|
||||
return await get_quests(session_id)
|
||||
|
||||
|
||||
@router.get("/{session_id}")
|
||||
async def get_session(session_id: str):
|
||||
sessions = await get_all_sessions()
|
||||
@@ -33,16 +54,36 @@ async def get_session(session_id: str):
|
||||
|
||||
@router.patch("/{session_id}")
|
||||
async def patch_session(session_id: str, data: dict):
|
||||
# ensure session exists before patching
|
||||
await get_or_create_session(session_id, data.get("persona_id", "default"))
|
||||
if "title" in data:
|
||||
await update_session_title(session_id, data["title"])
|
||||
if "persona_id" in data:
|
||||
await update_session_persona(session_id, data["persona_id"])
|
||||
if "rpg_enabled" in data:
|
||||
await update_session_rpg(session_id, bool(data["rpg_enabled"]))
|
||||
if "facts_json" in data:
|
||||
await update_session_facts(session_id, data["facts_json"])
|
||||
if "global_plot" in data:
|
||||
await update_session_global_plot(session_id, data["global_plot"])
|
||||
if "status_quo" in data:
|
||||
await update_session_status_quo(session_id, data["status_quo"])
|
||||
if "genre" in data:
|
||||
await update_session_genre(session_id, data["genre"])
|
||||
if "rpg_settings_json" in data:
|
||||
await update_session_rpg_settings(session_id, data["rpg_settings_json"])
|
||||
return {"status": "updated"}
|
||||
|
||||
|
||||
@router.post("/{session_id}/fork")
|
||||
async def fork_session_route(session_id: str, req: ForkSessionRequest):
|
||||
new_id = await fork_session(session_id, req.until_message_id)
|
||||
if not new_id:
|
||||
raise HTTPException(status_code=404, detail="Сессия не найдена")
|
||||
return {"session_id": new_id, "source_session_id": session_id}
|
||||
|
||||
|
||||
@router.delete("/{session_id}")
|
||||
async def remove_session(session_id: str):
|
||||
await delete_session(session_id)
|
||||
return {"status": "deleted", "session_id": session_id}
|
||||
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
import json
|
||||
import base64
|
||||
import uuid
|
||||
from pathlib import Path
|
||||
|
||||
import aiosqlite
|
||||
from database.db import DB_PATH
|
||||
@@ -110,8 +111,8 @@ async def save_character(card: dict, lora_name: str = "", lora_weight: float = 0
|
||||
await db.execute(
|
||||
"""INSERT OR REPLACE INTO characters
|
||||
(card_id, name, description, personality, scenario, first_mes,
|
||||
mes_example, raw_json, lora_name, lora_weight, appearance_tags, lorebook_json)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""",
|
||||
mes_example, raw_json, lora_name, lora_weight, appearance_tags, lorebook_json, avatar_path)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""",
|
||||
(
|
||||
card_id,
|
||||
card["name"],
|
||||
@@ -125,6 +126,7 @@ async def save_character(card: dict, lora_name: str = "", lora_weight: float = 0
|
||||
lora_weight,
|
||||
card.get("appearance_tags", ""),
|
||||
card["lorebook_json"],
|
||||
card.get("avatar_path", ""),
|
||||
),
|
||||
)
|
||||
await db.commit()
|
||||
@@ -171,7 +173,7 @@ async def update_appearance_tags(card_id: str, appearance_tags: str):
|
||||
|
||||
async def update_character(card_id: str, fields: dict) -> bool:
|
||||
allowed = {"name", "description", "personality", "scenario", "first_mes",
|
||||
"mes_example", "appearance_tags", "lora_name", "lora_weight"}
|
||||
"mes_example", "appearance_tags", "lora_name", "lora_weight", "avatar_path"}
|
||||
updates = {k: v for k, v in fields.items() if k in allowed}
|
||||
if not updates:
|
||||
return False
|
||||
@@ -190,6 +192,9 @@ async def import_card_file(content: bytes, filename: str, lora_name: str = "", l
|
||||
card = parse_png_card(content)
|
||||
if not card:
|
||||
raise ValueError("PNG does not contain character card metadata")
|
||||
# Use the PNG itself as avatar
|
||||
avatar_rel = _save_avatar_bytes(content, f"card_{card['card_id']}")
|
||||
card["avatar_path"] = avatar_rel
|
||||
else:
|
||||
card = parse_card_v2(json.loads(content.decode("utf-8")))
|
||||
|
||||
@@ -210,5 +215,15 @@ async def import_card_file(content: bytes, filename: str, lora_name: str = "", l
|
||||
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:
|
||||
avatars_dir = Path("static/avatars")
|
||||
avatars_dir.mkdir(parents=True, exist_ok=True)
|
||||
fname = f"{prefix}_{uuid.uuid4().hex[:8]}.png"
|
||||
path = avatars_dir / fname
|
||||
path.write_bytes(png_bytes)
|
||||
return f"avatars/{fname}"
|
||||
|
||||
@@ -31,6 +31,22 @@ async def send_message(messages: list) -> str:
|
||||
return data["choices"][0]["message"]["content"]
|
||||
|
||||
|
||||
async def send_message_with_model(messages: list, model: str) -> str:
|
||||
payload = {
|
||||
"model": model,
|
||||
"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):
|
||||
"""Стриминг — отдаём чанки по мере получения"""
|
||||
payload = {
|
||||
|
||||
+290
-1
@@ -36,6 +36,17 @@ async def get_all_sessions() -> list:
|
||||
return [dict(r) for r in rows]
|
||||
|
||||
|
||||
async def get_session(session_id: str) -> dict | None:
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
db.row_factory = aiosqlite.Row
|
||||
async with db.execute(
|
||||
"SELECT * FROM sessions WHERE session_id = ?",
|
||||
(session_id,),
|
||||
) as cursor:
|
||||
row = await cursor.fetchone()
|
||||
return dict(row) if row else None
|
||||
|
||||
|
||||
async def update_session_title(session_id: str, title: str):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
@@ -47,16 +58,118 @@ async def update_session_title(session_id: str, title: str):
|
||||
|
||||
async def update_session_persona(session_id: str, persona_id: str):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
db.row_factory = aiosqlite.Row
|
||||
async with db.execute(
|
||||
"SELECT persona_id FROM sessions WHERE session_id = ?",
|
||||
(session_id,),
|
||||
) as cur:
|
||||
row = await cur.fetchone()
|
||||
prev = row["persona_id"] if row else None
|
||||
|
||||
await db.execute(
|
||||
"UPDATE sessions SET persona_id = ?, updated_at = CURRENT_TIMESTAMP WHERE session_id = ?",
|
||||
(persona_id, session_id),
|
||||
)
|
||||
|
||||
# If persona changed, reset RPG state bound to the persona/arc.
|
||||
if prev is not None and prev != persona_id:
|
||||
await db.execute(
|
||||
"""UPDATE sessions
|
||||
SET facts_json = '[]',
|
||||
global_plot = '',
|
||||
status_quo = '',
|
||||
plot_arc_json = '{}'
|
||||
WHERE session_id = ?""",
|
||||
(session_id,),
|
||||
)
|
||||
await db.execute(
|
||||
"DELETE FROM action_resolutions WHERE session_id = ?",
|
||||
(session_id,),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def update_session_rpg(session_id: str, rpg_enabled: bool):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"UPDATE sessions SET rpg_enabled = ?, updated_at = CURRENT_TIMESTAMP WHERE session_id = ?",
|
||||
(1 if rpg_enabled else 0, session_id),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def update_session_facts(session_id: str, facts_json: str):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"UPDATE sessions SET facts_json = ?, updated_at = CURRENT_TIMESTAMP WHERE session_id = ?",
|
||||
(facts_json, session_id),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def update_session_global_plot(session_id: str, global_plot: str):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"UPDATE sessions SET global_plot = ?, updated_at = CURRENT_TIMESTAMP WHERE session_id = ?",
|
||||
(global_plot, session_id),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def update_session_status_quo(session_id: str, status_quo: str):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"UPDATE sessions SET status_quo = ?, updated_at = CURRENT_TIMESTAMP WHERE session_id = ?",
|
||||
(status_quo, session_id),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def update_session_plot_arc(session_id: str, plot_arc_json: str):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"UPDATE sessions SET plot_arc_json = ?, updated_at = CURRENT_TIMESTAMP WHERE session_id = ?",
|
||||
(plot_arc_json, session_id),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def add_action_resolution(
|
||||
session_id: str,
|
||||
intent_text: str,
|
||||
roll: int,
|
||||
outcome: str,
|
||||
resolution_text: str,
|
||||
message_id: int | None = None,
|
||||
):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"""INSERT INTO action_resolutions
|
||||
(session_id, message_id, intent_text, roll, outcome, resolution_text)
|
||||
VALUES (?, ?, ?, ?, ?, ?)""",
|
||||
(session_id, message_id, intent_text, roll, outcome, resolution_text),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def get_last_action_resolution(session_id: str) -> dict | None:
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
db.row_factory = aiosqlite.Row
|
||||
async with db.execute(
|
||||
"""SELECT * FROM action_resolutions
|
||||
WHERE session_id = ?
|
||||
ORDER BY id DESC LIMIT 1""",
|
||||
(session_id,),
|
||||
) as cur:
|
||||
row = await cur.fetchone()
|
||||
return dict(row) if row else None
|
||||
|
||||
|
||||
async def delete_session(session_id: str):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute("DELETE FROM messages WHERE session_id = ?", (session_id,))
|
||||
await db.execute("DELETE FROM rpg_quests WHERE session_id = ?", (session_id,))
|
||||
await db.execute("DELETE FROM action_resolutions WHERE session_id = ?", (session_id,))
|
||||
await db.execute("DELETE FROM sessions WHERE session_id = ?", (session_id,))
|
||||
await db.commit()
|
||||
|
||||
@@ -65,13 +178,14 @@ async def get_history(session_id: str) -> list:
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
db.row_factory = aiosqlite.Row
|
||||
async with db.execute(
|
||||
"""SELECT role, content, image_prompt, image_path
|
||||
"""SELECT id, role, content, image_prompt, image_path
|
||||
FROM messages WHERE session_id = ? ORDER BY id""",
|
||||
(session_id,),
|
||||
) as cursor:
|
||||
rows = await cursor.fetchall()
|
||||
return [
|
||||
{
|
||||
"id": r["id"],
|
||||
"role": r["role"],
|
||||
"content": r["content"],
|
||||
"image_prompt": r["image_prompt"],
|
||||
@@ -81,6 +195,114 @@ async def get_history(session_id: str) -> list:
|
||||
]
|
||||
|
||||
|
||||
async def get_message(message_id: int) -> dict | None:
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
db.row_factory = aiosqlite.Row
|
||||
async with db.execute(
|
||||
"SELECT id, session_id, role, content FROM messages WHERE id = ?",
|
||||
(message_id,),
|
||||
) as cursor:
|
||||
row = await cursor.fetchone()
|
||||
return dict(row) if row else None
|
||||
|
||||
|
||||
async def update_message_content(message_id: int, content: str):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"UPDATE messages SET content = ? WHERE id = ?",
|
||||
(content, message_id),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def delete_messages_after(session_id: str, message_id: int):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"DELETE FROM messages WHERE session_id = ? AND id > ?",
|
||||
(session_id, message_id),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def delete_message(message_id: int):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute("DELETE FROM messages WHERE id = ?", (message_id,))
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def get_last_message_preview(session_id: str, max_len: int = 80) -> str:
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
db.row_factory = aiosqlite.Row
|
||||
async with db.execute(
|
||||
"""SELECT content, role FROM messages
|
||||
WHERE session_id = ? AND role IN ('user', 'assistant')
|
||||
ORDER BY id DESC LIMIT 1""",
|
||||
(session_id,),
|
||||
) as cursor:
|
||||
row = await cursor.fetchone()
|
||||
if not row:
|
||||
return ""
|
||||
prefix = "Вы: " if row["role"] == "user" else "AI: "
|
||||
text = (row["content"] or "").replace("\n", " ").strip()
|
||||
if len(text) > max_len:
|
||||
text = text[:max_len] + "…"
|
||||
return prefix + text
|
||||
|
||||
|
||||
async def fork_session(source_session_id: str, until_message_id: int) -> str | None:
|
||||
source = await get_session(source_session_id)
|
||||
if not source:
|
||||
return None
|
||||
|
||||
import uuid
|
||||
new_id = "sess_" + uuid.uuid4().hex[:8]
|
||||
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"""INSERT INTO sessions
|
||||
(session_id, persona_id, title, rpg_enabled, facts_json, global_plot,
|
||||
status_quo, plot_arc_json, genre, rpg_settings_json, affinity)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""",
|
||||
(
|
||||
new_id,
|
||||
source["persona_id"],
|
||||
(source.get("title") or "Новый чат") + " (ветка)",
|
||||
source.get("rpg_enabled", 0),
|
||||
source.get("facts_json", "[]"),
|
||||
source.get("global_plot", ""),
|
||||
source.get("status_quo", ""),
|
||||
source.get("plot_arc_json", "{}"),
|
||||
source.get("genre", "adventure"),
|
||||
source.get("rpg_settings_json", "{}"),
|
||||
source.get("affinity", 0),
|
||||
),
|
||||
)
|
||||
async with db.execute(
|
||||
"""SELECT role, content, image_prompt, image_path FROM messages
|
||||
WHERE session_id = ? AND id <= ? ORDER BY id""",
|
||||
(source_session_id, until_message_id),
|
||||
) as cur:
|
||||
rows = await cur.fetchall()
|
||||
for r in rows:
|
||||
await db.execute(
|
||||
"""INSERT INTO messages (session_id, role, content, image_prompt, image_path)
|
||||
VALUES (?, ?, ?, ?, ?)""",
|
||||
(new_id, r[0], r[1], r[2], r[3]),
|
||||
)
|
||||
async with db.execute(
|
||||
"SELECT title, status FROM rpg_quests WHERE session_id = ?",
|
||||
(source_session_id,),
|
||||
) as cur:
|
||||
quests = await cur.fetchall()
|
||||
for q in quests:
|
||||
await db.execute(
|
||||
"INSERT INTO rpg_quests (session_id, title, status) VALUES (?, ?, ?)",
|
||||
(new_id, q[0], q[1]),
|
||||
)
|
||||
await db.commit()
|
||||
return new_id
|
||||
|
||||
|
||||
async def add_message(
|
||||
session_id: str,
|
||||
role: str,
|
||||
@@ -131,6 +353,73 @@ async def clear_history(session_id: str):
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def update_session_affinity(session_id: str, delta: int):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"UPDATE sessions SET affinity = affinity + ?, updated_at = CURRENT_TIMESTAMP WHERE session_id = ?",
|
||||
(delta, session_id),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def update_session_genre(session_id: str, genre: str):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"UPDATE sessions SET genre = ?, updated_at = CURRENT_TIMESTAMP WHERE session_id = ?",
|
||||
(genre, session_id),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def update_session_rpg_settings(session_id: str, settings_json: str):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"UPDATE sessions SET rpg_settings_json = ?, updated_at = CURRENT_TIMESTAMP WHERE session_id = ?",
|
||||
(settings_json, session_id),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def upsert_quest(session_id: str, title: str, status: str = "active"):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
async with db.execute(
|
||||
"SELECT id FROM rpg_quests WHERE session_id = ? AND title = ?",
|
||||
(session_id, title),
|
||||
) as cur:
|
||||
row = await cur.fetchone()
|
||||
if row:
|
||||
await db.execute(
|
||||
"UPDATE rpg_quests SET status = ? WHERE id = ?",
|
||||
(status, row[0]),
|
||||
)
|
||||
else:
|
||||
await db.execute(
|
||||
"INSERT INTO rpg_quests (session_id, title, status) VALUES (?, ?, ?)",
|
||||
(session_id, title, status),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def get_quests(session_id: str) -> list:
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
db.row_factory = aiosqlite.Row
|
||||
async with db.execute(
|
||||
"SELECT id, title, status FROM rpg_quests WHERE session_id = ? ORDER BY id",
|
||||
(session_id,),
|
||||
) as cur:
|
||||
rows = await cur.fetchall()
|
||||
return [dict(r) for r in rows]
|
||||
|
||||
|
||||
async def update_quest_status(session_id: str, title: str, status: str):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"UPDATE rpg_quests SET status = ? WHERE session_id = ? AND title = ?",
|
||||
(status, session_id, title),
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def get_message_count(session_id: str) -> int:
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
db.row_factory = aiosqlite.Row
|
||||
|
||||
+95
-4
@@ -63,9 +63,29 @@ def _row_to_persona(row: dict) -> dict:
|
||||
"lora_name": row["lora_name"] or "",
|
||||
"lora_weight": row["lora_weight"] if row["lora_weight"] is not None else 0.8,
|
||||
"appearance_tags": row["appearance_tags"] or "",
|
||||
"personality": row.get("personality", "") or "",
|
||||
"scenario": row.get("scenario", "") or "",
|
||||
"first_mes": row.get("first_mes", "") or "",
|
||||
"mes_example": row.get("mes_example", "") or "",
|
||||
"lorebook_json": row.get("lorebook_json", "[]") or "[]",
|
||||
"avatar_path": row.get("avatar_path", "") or "",
|
||||
}
|
||||
|
||||
|
||||
def build_persona_prompt(data: dict) -> str:
|
||||
parts = [
|
||||
f"You are {data.get('name', '').strip()}." if data.get("name") else "",
|
||||
f"Description: {data.get('description', '').strip()}",
|
||||
f"Personality: {data.get('personality', '').strip()}",
|
||||
f"Scenario: {data.get('scenario', '').strip()}",
|
||||
]
|
||||
ex = (data.get("mes_example") or "").strip()
|
||||
if ex:
|
||||
parts.append(f"Example dialogue:\n{ex}")
|
||||
parts.append("Stay in character. Reply as the character. Do not add image tags.")
|
||||
return "\n\n".join(p for p in parts if p and p.split(": ", 1)[-1].strip())
|
||||
|
||||
|
||||
async def get_all_personas() -> dict:
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
db.row_factory = aiosqlite.Row
|
||||
@@ -96,16 +116,33 @@ async def create_persona(
|
||||
lora_name: str = "",
|
||||
lora_weight: float = 0.8,
|
||||
appearance_tags: str = "",
|
||||
personality: str = "",
|
||||
scenario: str = "",
|
||||
first_mes: str = "",
|
||||
mes_example: str = "",
|
||||
lorebook_json: str = "[]",
|
||||
avatar_path: str = "",
|
||||
) -> dict:
|
||||
final_prompt = prompt.strip() or build_persona_prompt(
|
||||
{
|
||||
"name": name,
|
||||
"description": description,
|
||||
"personality": personality,
|
||||
"scenario": scenario,
|
||||
"mes_example": mes_example,
|
||||
}
|
||||
)
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute(
|
||||
"""INSERT INTO personas
|
||||
(persona_id, name, emoji, description, prompt, custom,
|
||||
sd_enabled, lora_name, lora_weight, appearance_tags)
|
||||
VALUES (?, ?, ?, ?, ?, 1, ?, ?, ?, ?)""",
|
||||
sd_enabled, lora_name, lora_weight, appearance_tags,
|
||||
personality, scenario, first_mes, mes_example, lorebook_json, avatar_path)
|
||||
VALUES (?, ?, ?, ?, ?, 1, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""",
|
||||
(
|
||||
persona_id, name, emoji, description, prompt,
|
||||
persona_id, name, emoji, description, final_prompt,
|
||||
1 if sd_enabled else 0, lora_name, lora_weight, appearance_tags,
|
||||
personality, scenario, first_mes, mes_example, lorebook_json, avatar_path,
|
||||
),
|
||||
)
|
||||
await db.commit()
|
||||
@@ -113,12 +150,18 @@ async def create_persona(
|
||||
"name": name,
|
||||
"emoji": emoji,
|
||||
"description": description,
|
||||
"prompt": prompt,
|
||||
"prompt": final_prompt,
|
||||
"custom": True,
|
||||
"sd_enabled": sd_enabled,
|
||||
"lora_name": lora_name,
|
||||
"lora_weight": lora_weight,
|
||||
"appearance_tags": appearance_tags,
|
||||
"personality": personality,
|
||||
"scenario": scenario,
|
||||
"first_mes": first_mes,
|
||||
"mes_example": mes_example,
|
||||
"lorebook_json": lorebook_json,
|
||||
"avatar_path": avatar_path,
|
||||
}
|
||||
|
||||
|
||||
@@ -166,3 +209,51 @@ async def update_persona_prompt(persona_id: str, prompt: str):
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
await db.execute("UPDATE personas SET prompt = ? WHERE persona_id = ?", (prompt, persona_id))
|
||||
await db.commit()
|
||||
|
||||
|
||||
async def patch_persona(persona_id: str, fields: dict) -> bool:
|
||||
allowed = {
|
||||
"name",
|
||||
"emoji",
|
||||
"description",
|
||||
"prompt",
|
||||
"sd_enabled",
|
||||
"lora_name",
|
||||
"lora_weight",
|
||||
"appearance_tags",
|
||||
"personality",
|
||||
"scenario",
|
||||
"first_mes",
|
||||
"mes_example",
|
||||
"lorebook_json",
|
||||
"avatar_path",
|
||||
}
|
||||
updates = {k: v for k, v in fields.items() if k in allowed}
|
||||
if not updates:
|
||||
return False
|
||||
|
||||
async with aiosqlite.connect(DB_PATH) as db:
|
||||
db.row_factory = aiosqlite.Row
|
||||
# disallow editing built-in personas
|
||||
async with db.execute("SELECT custom FROM personas WHERE persona_id = ?", (persona_id,)) as cur:
|
||||
row = await cur.fetchone()
|
||||
if not row or not row[0]:
|
||||
return False
|
||||
|
||||
# rebuild prompt if user didn't explicitly set it
|
||||
raw_fields = {"name", "description", "personality", "scenario", "mes_example"}
|
||||
if "prompt" not in updates and (raw_fields & updates.keys()):
|
||||
async with db.execute("SELECT * FROM personas WHERE persona_id = ?", (persona_id,)) as cur:
|
||||
existing = await cur.fetchone()
|
||||
if existing:
|
||||
merged = dict(existing)
|
||||
merged.update(updates)
|
||||
updates["prompt"] = build_persona_prompt(merged)
|
||||
|
||||
cols = ", ".join(f"{k} = ?" for k in updates)
|
||||
cur2 = await db.execute(
|
||||
f"UPDATE personas SET {cols} WHERE persona_id = ?",
|
||||
(*updates.values(), persona_id),
|
||||
)
|
||||
await db.commit()
|
||||
return cur2.rowcount > 0
|
||||
|
||||
@@ -0,0 +1,76 @@
|
||||
import json
|
||||
import os
|
||||
|
||||
from services.llm import send_message_with_model, send_message
|
||||
|
||||
FACTS_MODEL = os.getenv("RPG_FACTS_MODEL", "").strip() or "deepseek/deepseek-chat-v3"
|
||||
|
||||
FACTS_SYSTEM = """Extract stable facts from the conversation.
|
||||
Return ONLY valid JSON (no markdown), as an array of short strings.
|
||||
Rules:
|
||||
- Facts must be durable (names, relations, inventory, locations, world rules).
|
||||
- Do not include ephemeral actions unless they change state.
|
||||
- Avoid duplicates.
|
||||
- Keep each fact <= 120 chars.
|
||||
Example output:
|
||||
["User name is Alex", "We are in a ruined castle", "NPC Mira distrusts the user"]"""
|
||||
|
||||
|
||||
def merge_facts(existing_json: str, new_facts: list[str], limit: int = 80) -> str:
|
||||
try:
|
||||
existing = json.loads(existing_json or "[]")
|
||||
if not isinstance(existing, list):
|
||||
existing = []
|
||||
except json.JSONDecodeError:
|
||||
existing = []
|
||||
|
||||
seen = {str(x).strip() for x in existing if str(x).strip()}
|
||||
merged = [str(x).strip() for x in existing if str(x).strip()]
|
||||
for f in new_facts:
|
||||
s = str(f).strip()
|
||||
if not s or s in seen:
|
||||
continue
|
||||
seen.add(s)
|
||||
merged.append(s)
|
||||
|
||||
if len(merged) > limit:
|
||||
merged = merged[-limit:]
|
||||
return json.dumps(merged, ensure_ascii=False)
|
||||
|
||||
|
||||
async def extract_facts(context_messages: list[dict]) -> list[str]:
|
||||
# Build a compact transcript
|
||||
transcript = "\n".join(
|
||||
f"{m.get('role')}: {m.get('content','')}".strip()
|
||||
for m in context_messages
|
||||
if m.get("role") in ("user", "assistant")
|
||||
)[-6000:]
|
||||
|
||||
messages = [
|
||||
{"role": "system", "content": FACTS_SYSTEM},
|
||||
{"role": "user", "content": transcript},
|
||||
]
|
||||
|
||||
raw = await (send_message_with_model(messages, FACTS_MODEL) if FACTS_MODEL else send_message(messages))
|
||||
try:
|
||||
data = json.loads(raw.strip())
|
||||
if isinstance(data, list):
|
||||
return [str(x) for x in data][:40]
|
||||
except Exception:
|
||||
return []
|
||||
return []
|
||||
|
||||
|
||||
def facts_to_prompt(facts_json: str, max_items: int = 20) -> str:
|
||||
try:
|
||||
facts = json.loads(facts_json or "[]")
|
||||
if not isinstance(facts, list):
|
||||
return ""
|
||||
except json.JSONDecodeError:
|
||||
return ""
|
||||
facts = [str(x).strip() for x in facts if str(x).strip()]
|
||||
if not facts:
|
||||
return ""
|
||||
block = "\n".join(f"- {x}" for x in facts[-max_items:])
|
||||
return f"--- Facts (persistent memory) ---\n{block}\n---"
|
||||
|
||||
@@ -0,0 +1,111 @@
|
||||
import json
|
||||
import os
|
||||
import random
|
||||
|
||||
from services.llm import send_message_with_model
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
NARRATOR_MODEL = os.getenv("RPG_NARRATOR_MODEL", "").strip() or "deepseek/deepseek-chat-v3"
|
||||
|
||||
NARRATOR_PRE_SYSTEM = """You are the System/Narrator of an RPG chat.
|
||||
Decide if the user's action requires a skill/ability check (physical action, persuasion, deception, stealth, combat, etc.).
|
||||
Pure dialogue, questions, or passive observation do NOT require a check.
|
||||
Return ONLY valid JSON (no markdown):
|
||||
{
|
||||
"needs_check": true,
|
||||
"check_reason": "brief reason why a check is needed (e.g. 'jumping over a pit')",
|
||||
"directives": ["short imperative rules for the next character reply"],
|
||||
"resolution_text": "what actually happens as result of the action — written as narrator prose (1-2 sentences). Only if needs_check=true and roll/outcome provided.",
|
||||
"status_quo_update": "optional short update about the world state"
|
||||
}
|
||||
If needs_check=false: directives may still guide tone/pacing, resolution_text must be empty string.
|
||||
If needs_check=true and roll/outcome are provided: resolution_text MUST reflect the outcome.
|
||||
- critical failure (1): embarrassing or painful failure with extra complication
|
||||
- failure (2-8): action fails, partial or no progress
|
||||
- success (9-19): action succeeds as intended
|
||||
- critical success (20): spectacular success with bonus effect"""
|
||||
|
||||
NARRATOR_POST_SYSTEM = """You are the System/Narrator of an RPG chat.
|
||||
After the character replied, update persistent state.
|
||||
Return ONLY valid JSON (no markdown):
|
||||
{
|
||||
"status_quo_update": "what changed in the world/state (1-3 sentences)",
|
||||
"facts": ["durable facts only"],
|
||||
"choices": [{"id":"a","label":"..."}, ...],
|
||||
"affinity_delta": 0,
|
||||
"quest_updates": [{"title": "quest title", "status": "active|done|failed"}]
|
||||
}
|
||||
Rules:
|
||||
- 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.
|
||||
- choices: 0-4 options for what the player can do next."""
|
||||
|
||||
|
||||
async def narrator_pre(
|
||||
persona_name: str,
|
||||
context: str,
|
||||
global_plot: str,
|
||||
facts_block: str,
|
||||
user_message: str,
|
||||
roll: int | None = None,
|
||||
outcome: str | None = None,
|
||||
) -> dict:
|
||||
roll_block = f"Roll d20={roll}\nOutcome={outcome}\n\n" if roll is not None else ""
|
||||
user = (
|
||||
f"Persona: {persona_name}\n"
|
||||
f"{roll_block}"
|
||||
f"User action: {user_message}\n\n"
|
||||
f"Global plot:\n{global_plot}\n\n"
|
||||
f"Facts:\n{facts_block}\n\n"
|
||||
f"Recent context:\n{context}\n"
|
||||
)
|
||||
raw = await send_message_with_model(
|
||||
[{"role": "system", "content": NARRATOR_PRE_SYSTEM}, {"role": "user", "content": user}],
|
||||
NARRATOR_MODEL,
|
||||
)
|
||||
cleaned = raw.strip()
|
||||
if cleaned.startswith("```"):
|
||||
cleaned = cleaned.split("\n", 1)[1] if "\n" in cleaned else cleaned
|
||||
if cleaned.endswith("```"):
|
||||
cleaned = cleaned.rsplit("```", 1)[0]
|
||||
cleaned = cleaned.strip()
|
||||
try:
|
||||
data = json.loads(cleaned)
|
||||
if isinstance(data, dict):
|
||||
return data
|
||||
except Exception:
|
||||
logger.warning("Narrator-pre JSON parse failed. Raw=%.500s", raw)
|
||||
return {"needs_check": False, "directives": [], "status_quo_update": "", "resolution_text": ""}
|
||||
|
||||
|
||||
async def narrator_post(
|
||||
persona_name: str,
|
||||
context: str,
|
||||
global_plot: str,
|
||||
facts_block: str,
|
||||
) -> dict:
|
||||
user = (
|
||||
f"Persona: {persona_name}\n\n"
|
||||
f"Global plot:\n{global_plot}\n\n"
|
||||
f"Facts:\n{facts_block}\n\n"
|
||||
f"Recent context:\n{context}\n"
|
||||
)
|
||||
raw = await send_message_with_model(
|
||||
[{"role": "system", "content": NARRATOR_POST_SYSTEM}, {"role": "user", "content": user}],
|
||||
NARRATOR_MODEL,
|
||||
)
|
||||
cleaned = raw.strip()
|
||||
if cleaned.startswith("```"):
|
||||
cleaned = cleaned.split("\n", 1)[1] if "\n" in cleaned else cleaned
|
||||
if cleaned.endswith("```"):
|
||||
cleaned = cleaned.rsplit("```", 1)[0]
|
||||
cleaned = cleaned.strip()
|
||||
try:
|
||||
data = json.loads(cleaned)
|
||||
if isinstance(data, dict):
|
||||
return data
|
||||
except Exception:
|
||||
logger.warning("Narrator-post JSON parse failed. Raw=%.500s", raw)
|
||||
return {"status_quo_update": "", "facts": [], "choices": [], "affinity_delta": 0, "quest_updates": []}
|
||||
@@ -0,0 +1,105 @@
|
||||
import json
|
||||
import os
|
||||
|
||||
from services.llm import send_message_with_model, send_message
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
PLOT_MODEL = os.getenv("RPG_PLOT_MODEL", "").strip() or "deepseek/deepseek-chat-v3"
|
||||
|
||||
GENRE_LABELS = {
|
||||
"adventure": "Adventure",
|
||||
"horror": "Horror",
|
||||
"romance": "Romance",
|
||||
"slice_of_life": "Slice of Life",
|
||||
"fantasy": "Fantasy",
|
||||
"sci_fi": "Sci-Fi",
|
||||
}
|
||||
|
||||
ARC_SYSTEM = """You are a narrative designer for an RPG chat.
|
||||
Given the opening scene (greeting), character info, current facts, and genre(s), produce a STRUCTURED PLOT ARC.
|
||||
Return ONLY valid JSON (no markdown):
|
||||
{
|
||||
"title": "short arc title",
|
||||
"boundaries": ["things that must remain true to preserve immersion"],
|
||||
"phase": "opening|hook|complication|reveal|climax|aftermath",
|
||||
"cast": [{"name":"NPC name","role":"helper|antagonist|bystander","motivation":"..."}],
|
||||
"secrets": ["hidden truths not revealed yet"],
|
||||
"beats": [
|
||||
{"id":"b1","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",
|
||||
"choices":[{"id":"a","label":"..."},{"id":"b","label":"..."}]}
|
||||
],
|
||||
"next_beat_hint": "short hint for narrator what to push next"
|
||||
}
|
||||
Rules:
|
||||
- Respect the opening scene. Do not jump to unrelated characters immediately.
|
||||
- Beats must feel like natural developments fitting the genre(s). For cross-genre, blend tropes organically.
|
||||
- Keep injections immersive (in-world narration)."""
|
||||
|
||||
|
||||
def format_genres(genre: str) -> str:
|
||||
parts = [g.strip() for g in genre.replace("+", ",").split(",") if g.strip()]
|
||||
if not parts:
|
||||
return "Adventure"
|
||||
labels = [GENRE_LABELS.get(g, g.replace("_", " ").title()) for g in parts]
|
||||
if len(labels) == 1:
|
||||
return labels[0]
|
||||
return " + ".join(labels) + " (cross-genre blend)"
|
||||
|
||||
|
||||
async def generate_plot_arc(persona_name: str, persona_desc: str, persona_scenario: str, greeting: str, facts_block: str = "", genre: str = "adventure") -> dict:
|
||||
user = (
|
||||
f"Character: {persona_name}\n"
|
||||
f"Description: {persona_desc}\n"
|
||||
f"Scenario: {persona_scenario}\n"
|
||||
f"Greeting: {greeting}\n"
|
||||
f"Genre: {format_genres(genre)}\n"
|
||||
f"Facts:\n{facts_block}\n"
|
||||
).strip()
|
||||
|
||||
messages = [
|
||||
{"role": "system", "content": ARC_SYSTEM},
|
||||
{"role": "user", "content": user},
|
||||
]
|
||||
raw = await (send_message_with_model(messages, PLOT_MODEL) if PLOT_MODEL else send_message(messages))
|
||||
cleaned = raw.strip()
|
||||
# common OpenRouter formatting: fenced json
|
||||
if cleaned.startswith("```"):
|
||||
cleaned = cleaned.split("\n", 1)[1] if "\n" in cleaned else cleaned
|
||||
if cleaned.endswith("```"):
|
||||
cleaned = cleaned.rsplit("```", 1)[0]
|
||||
cleaned = cleaned.strip()
|
||||
try:
|
||||
data = json.loads(cleaned)
|
||||
return data if isinstance(data, dict) else {}
|
||||
except Exception:
|
||||
logger.warning("PlotArc JSON parse failed. Raw=%.500s", raw)
|
||||
return {}
|
||||
|
||||
|
||||
def should_advance_arc(user_text: str) -> str | None:
|
||||
t = (user_text or "").lower()
|
||||
if any(x in t for x in ["отдыха", "ночлег", "спим", "сон", "разбить лагерь", "лагерь", "отдохн"]):
|
||||
return "event_driven:rest"
|
||||
if any(x in t for x in ["идем дальше", "пойдем дальше", "в путь", "продолжаем путь", "уходим", "возвращаемся", "переходим"]):
|
||||
return "event_driven:travel"
|
||||
if any(x in t for x in ["помоги", "помочь", "нужна помощь", "спасите", "help"]):
|
||||
return "event_driven:help_request"
|
||||
return None
|
||||
|
||||
|
||||
def pop_matching_beats(arc: dict, trigger: str, max_beats: int = 1) -> tuple[dict, list[dict]]:
|
||||
beats = arc.get("beats", [])
|
||||
if not isinstance(beats, list):
|
||||
return arc, []
|
||||
matched, remaining = [], []
|
||||
for b in beats:
|
||||
if len(matched) < max_beats and isinstance(b, dict) and b.get("trigger") == trigger:
|
||||
matched.append(b)
|
||||
else:
|
||||
remaining.append(b)
|
||||
arc["beats"] = remaining
|
||||
return arc, matched
|
||||
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 77 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 71 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 71 KiB |
@@ -5,6 +5,7 @@ body {
|
||||
color: #e0e0e0;
|
||||
font-family: 'Segoe UI', sans-serif;
|
||||
height: 100vh;
|
||||
height: 100dvh;
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
overflow: hidden;
|
||||
@@ -45,6 +46,21 @@ header h1 { font-size: 1.1rem; color: #e94560; }
|
||||
white-space: nowrap;
|
||||
}
|
||||
|
||||
.rpg-toggle {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: 6px;
|
||||
font-size: 0.75rem;
|
||||
color: #888;
|
||||
border: 1px solid #0f3460;
|
||||
border-radius: 10px;
|
||||
padding: 4px 10px;
|
||||
cursor: pointer;
|
||||
user-select: none;
|
||||
}
|
||||
.rpg-toggle input { accent-color: #e94560; }
|
||||
.rpg-toggle:hover { border-color: #e94560; color: #e94560; }
|
||||
|
||||
.app-body { display: flex; flex: 1; overflow: hidden; }
|
||||
|
||||
.sidebar {
|
||||
@@ -96,6 +112,7 @@ header h1 { font-size: 1.1rem; color: #e94560; }
|
||||
.session-item:hover { background: #1a1a2e; }
|
||||
.session-item.active { background: #1a1a2e; border-left-color: #e94560; }
|
||||
.session-item .s-title { flex: 1; font-size: 0.82rem; color: #ccc; overflow: hidden; text-overflow: ellipsis; white-space: nowrap; }
|
||||
.session-item .s-companion { flex: 1; font-size: 0.72rem; color: #777; margin-top: 2px; overflow: hidden; text-overflow: ellipsis; white-space: nowrap; }
|
||||
.session-item .s-meta { font-size: 0.7rem; color: #555; }
|
||||
.session-item .s-del { background: none; border: none; color: #555; cursor: pointer; opacity: 0; }
|
||||
.session-item:hover .s-del { opacity: 1; }
|
||||
@@ -111,6 +128,38 @@ header h1 { font-size: 1.1rem; color: #e94560; }
|
||||
border-bottom: 1px solid #0f3460;
|
||||
}
|
||||
|
||||
.system-blob {
|
||||
border-bottom: 1px solid #0f3460;
|
||||
background: #11162a;
|
||||
padding: 8px 16px;
|
||||
}
|
||||
.system-blob-header {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: space-between;
|
||||
color: #888;
|
||||
font-size: 0.8rem;
|
||||
margin-bottom: 6px;
|
||||
}
|
||||
.system-blob-header button {
|
||||
background: transparent;
|
||||
border: 1px solid #0f3460;
|
||||
border-radius: 8px;
|
||||
color: #888;
|
||||
padding: 4px 10px;
|
||||
cursor: pointer;
|
||||
}
|
||||
.system-blob-header button:hover { border-color: #e94560; color: #e94560; }
|
||||
.system-blob-content {
|
||||
white-space: pre-wrap;
|
||||
word-break: break-word;
|
||||
font-size: 0.78rem;
|
||||
color: #aaa;
|
||||
max-height: 140px;
|
||||
overflow: auto;
|
||||
margin: 0;
|
||||
}
|
||||
|
||||
.persona-card {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
@@ -128,6 +177,13 @@ header h1 { font-size: 1.1rem; color: #e94560; }
|
||||
.persona-card.active { border-color: #e94560; background: #1f1535; }
|
||||
.persona-card .emoji { font-size: 1.2rem; }
|
||||
.persona-card .pname { font-size: 0.7rem; color: #ccc; }
|
||||
.persona-card .avatar {
|
||||
width: 28px;
|
||||
height: 28px;
|
||||
border-radius: 50%;
|
||||
object-fit: cover;
|
||||
border: 1px solid #0f3460;
|
||||
}
|
||||
.persona-card .del-btn {
|
||||
position: absolute; top: -5px; right: -5px;
|
||||
width: 14px; height: 14px;
|
||||
@@ -224,6 +280,27 @@ header h1 { font-size: 1.1rem; color: #e94560; }
|
||||
.chat-image { margin-top: 8px; max-width: 100%; border-radius: 8px; border: 1px solid #0f3460; }
|
||||
.image-error { margin-top: 6px; font-size: 0.75rem; color: #888; }
|
||||
|
||||
.choice-row {
|
||||
display: flex;
|
||||
flex-wrap: wrap;
|
||||
gap: 6px;
|
||||
margin-top: 8px;
|
||||
}
|
||||
.choice-btn {
|
||||
background: #16213e;
|
||||
border: 1px solid #0f3460;
|
||||
border-radius: 10px;
|
||||
color: #ccc;
|
||||
font-size: 0.8rem;
|
||||
padding: 6px 10px;
|
||||
cursor: pointer;
|
||||
}
|
||||
.choice-btn:hover {
|
||||
border-color: #e94560;
|
||||
color: #e94560;
|
||||
}
|
||||
|
||||
|
||||
.typing {
|
||||
align-self: flex-start;
|
||||
display: flex; gap: 4px;
|
||||
@@ -240,6 +317,7 @@ header h1 { font-size: 1.1rem; color: #e94560; }
|
||||
.input-area {
|
||||
display: flex; gap: 10px;
|
||||
padding: 12px 16px;
|
||||
padding-bottom: max(12px, env(safe-area-inset-bottom));
|
||||
border-top: 1px solid #0f3460;
|
||||
}
|
||||
|
||||
@@ -276,6 +354,8 @@ textarea:focus { border-color: #e94560; }
|
||||
display: none; position: fixed; inset: 0;
|
||||
background: rgba(0,0,0,0.7);
|
||||
z-index: 100; align-items: center; justify-content: center;
|
||||
padding: 16px;
|
||||
overflow-y: auto;
|
||||
}
|
||||
.modal-overlay.open { display: flex; }
|
||||
|
||||
@@ -283,8 +363,48 @@ textarea:focus { border-color: #e94560; }
|
||||
background: #16213e; border: 1px solid #0f3460;
|
||||
border-radius: 16px; padding: 24px;
|
||||
width: 100%; max-width: 440px;
|
||||
max-height: calc(100vh - 32px);
|
||||
display: flex; flex-direction: column; gap: 12px;
|
||||
margin: auto;
|
||||
}
|
||||
.modal-wizard { max-width: 480px; }
|
||||
.modal-wizard-header { flex-shrink: 0; }
|
||||
.modal-wizard-header h2 { margin-bottom: 8px; }
|
||||
.modal-wizard-body {
|
||||
flex: 1; min-height: 0;
|
||||
overflow-y: auto;
|
||||
padding-right: 4px;
|
||||
}
|
||||
.modal-wizard-footer {
|
||||
flex-shrink: 0;
|
||||
margin-top: 4px;
|
||||
}
|
||||
.wizard-steps {
|
||||
display: flex; align-items: center; justify-content: center;
|
||||
gap: 0; margin-bottom: 4px;
|
||||
}
|
||||
.wizard-step-dot {
|
||||
width: 28px; height: 28px; border-radius: 50%;
|
||||
background: #1a1a2e; border: 1px solid #0f3460;
|
||||
color: #666; font-size: 0.75rem;
|
||||
display: flex; align-items: center; justify-content: center;
|
||||
transition: all 0.15s;
|
||||
}
|
||||
.wizard-step-dot.active { border-color: #e94560; color: #e94560; background: #1f0a14; }
|
||||
.wizard-step-dot.done { border-color: #9b7fd4; color: #9b7fd4; }
|
||||
.wizard-step-line { width: 40px; height: 1px; background: #0f3460; }
|
||||
.wizard-page { display: none; flex-direction: column; gap: 12px; }
|
||||
.wizard-page.active { display: flex; }
|
||||
.wizard-page-title { font-size: 0.85rem; color: #9b7fd4; margin: 0 0 4px; }
|
||||
.wizard-hint { font-size: 0.8rem; color: #666; margin: -4px 0 4px; }
|
||||
.selected-genres-label { font-size: 0.8rem; color: #e94560; margin-top: 4px; }
|
||||
.wizard-nav { display: flex; gap: 8px; }
|
||||
.wizard-nav-btn {
|
||||
padding: 8px 18px; border-radius: 8px; border: none; cursor: pointer;
|
||||
background: #0f3460; color: #ccc;
|
||||
}
|
||||
.wizard-nav-btn:hover { color: #e0e0e0; }
|
||||
.wizard-nav-btn:disabled { opacity: 0.45; cursor: not-allowed; }
|
||||
.modal h2 { font-size: 1.1rem; color: #e94560; }
|
||||
.modal label { display: flex; flex-direction: column; gap: 4px; font-size: 0.8rem; color: #888; }
|
||||
.modal input, .modal textarea {
|
||||
@@ -293,6 +413,7 @@ textarea:focus { border-color: #e94560; }
|
||||
padding: 8px 10px; outline: none; font-family: inherit;
|
||||
}
|
||||
.modal-buttons { display: flex; gap: 8px; justify-content: flex-end; }
|
||||
.modal-wizard-footer { justify-content: space-between; align-items: center; }
|
||||
.modal-buttons button { padding: 8px 18px; border-radius: 8px; border: none; cursor: pointer; }
|
||||
#modalCancel, #cardModalCancel { background: #0f3460; color: #aaa; }
|
||||
#modalSave, #cardModalImport { background: #e94560; color: white; }
|
||||
@@ -304,3 +425,168 @@ textarea:focus { border-color: #e94560; }
|
||||
}
|
||||
.empty-state .big { font-size: 2.5rem; }
|
||||
.hidden { display: none !important; }
|
||||
|
||||
/* Narrator message bubble */
|
||||
.message.narrator { align-self: center; max-width: 80%; }
|
||||
.message.narrator .label { color: #9b7fd4; font-size: 0.75rem; margin-bottom: 4px; }
|
||||
.message.narrator .bubble {
|
||||
background: #1a1230;
|
||||
border: 1px solid #4a2d8a;
|
||||
border-left: 3px solid #9b7fd4;
|
||||
border-radius: 12px;
|
||||
padding: 10px 14px;
|
||||
color: #ccc;
|
||||
font-size: 0.9rem;
|
||||
}
|
||||
|
||||
/* Dice block inside narrator bubble */
|
||||
.dice-block {
|
||||
display: flex; align-items: center; gap: 8px;
|
||||
margin-bottom: 8px;
|
||||
padding: 4px 8px;
|
||||
border-radius: 8px;
|
||||
background: #0f0a1e;
|
||||
font-size: 0.8rem;
|
||||
width: fit-content;
|
||||
}
|
||||
.dice-icon { font-size: 1.1rem; }
|
||||
.dice-roll { font-size: 1.3rem; font-weight: bold; }
|
||||
.dice-outcome { font-size: 0.75rem; text-transform: uppercase; letter-spacing: 0.05em; }
|
||||
.outcome-crit-fail { border: 1px solid #c0392b; }
|
||||
.outcome-crit-fail .dice-roll { color: #e74c3c; }
|
||||
.outcome-crit-fail .dice-outcome { color: #e74c3c; }
|
||||
.outcome-fail { border: 1px solid #555; }
|
||||
.outcome-fail .dice-roll { color: #aaa; }
|
||||
.outcome-fail .dice-outcome { color: #888; }
|
||||
.outcome-success { border: 1px solid #27ae60; }
|
||||
.outcome-success .dice-roll { color: #2ecc71; }
|
||||
.outcome-success .dice-outcome { color: #2ecc71; }
|
||||
.outcome-crit-success { border: 1px solid #f39c12; }
|
||||
.outcome-crit-success .dice-roll { color: #f1c40f; }
|
||||
.outcome-crit-success .dice-outcome { color: #f1c40f; }
|
||||
.narrator-text { white-space: pre-wrap; line-height: 1.5; }
|
||||
|
||||
/* Affinity display in header */
|
||||
.affinity-display {
|
||||
font-size: 0.8rem; padding: 4px 10px;
|
||||
border: 1px solid #0f3460; border-radius: 10px;
|
||||
color: #aaa; white-space: nowrap;
|
||||
}
|
||||
.affinity-display.affinity-high { border-color: #e94560; color: #e94560; }
|
||||
.affinity-display.affinity-low { border-color: #555; color: #666; }
|
||||
|
||||
/* Quest panel in sidebar */
|
||||
.quest-panel {
|
||||
border-top: 1px solid #0f3460;
|
||||
padding: 10px 14px;
|
||||
flex-shrink: 0;
|
||||
}
|
||||
.quest-panel-header {
|
||||
font-size: 0.75rem; color: #888;
|
||||
text-transform: uppercase; letter-spacing: 0.05em;
|
||||
margin-bottom: 6px;
|
||||
}
|
||||
.quest-item {
|
||||
font-size: 0.8rem; padding: 4px 0;
|
||||
color: #bbb; line-height: 1.4;
|
||||
border-bottom: 1px solid #0f3460;
|
||||
}
|
||||
.quest-item:last-child { border-bottom: none; }
|
||||
.quest-done { color: #555; text-decoration: line-through; }
|
||||
.quest-failed { color: #c0392b; }
|
||||
|
||||
/* Genre modal grid */
|
||||
.genre-grid {
|
||||
display: grid; grid-template-columns: 1fr 1fr;
|
||||
gap: 8px; margin: 4px 0;
|
||||
}
|
||||
.genre-btn {
|
||||
background: #1a1a2e; border: 1px solid #0f3460;
|
||||
border-radius: 8px; color: #ccc;
|
||||
padding: 10px 8px; cursor: pointer;
|
||||
font-size: 0.85rem; text-align: center;
|
||||
transition: all 0.15s;
|
||||
}
|
||||
.genre-btn:hover { border-color: #9b7fd4; color: #e0e0e0; }
|
||||
.genre-btn.selected { border-color: #e94560; color: #e94560; background: #1f0a14; }
|
||||
|
||||
/* RPG settings checkboxes */
|
||||
.rpg-settings-grid {
|
||||
display: grid; grid-template-columns: 1fr 1fr;
|
||||
gap: 6px; margin: 4px 0;
|
||||
}
|
||||
.rpg-settings-grid label {
|
||||
display: flex; align-items: center; gap: 6px;
|
||||
font-size: 0.8rem; color: #aaa; cursor: pointer;
|
||||
flex-direction: row;
|
||||
}
|
||||
.rpg-settings-grid input[type=checkbox] { accent-color: #e94560; }
|
||||
|
||||
.header-icon-btn {
|
||||
background: transparent; border: 1px solid #0f3460;
|
||||
border-radius: 10px; color: #aaa; padding: 4px 10px;
|
||||
cursor: pointer; font-size: 1rem;
|
||||
}
|
||||
.header-icon-btn:hover { border-color: #e94560; color: #e94560; }
|
||||
.rpg-badge {
|
||||
font-size: 0.7rem; padding: 3px 8px;
|
||||
border: 1px solid #9b7fd4; border-radius: 8px;
|
||||
color: #9b7fd4; text-transform: uppercase;
|
||||
letter-spacing: 0.05em;
|
||||
}
|
||||
.persona-pick-grid {
|
||||
display: grid; grid-template-columns: repeat(auto-fill, minmax(100px, 1fr));
|
||||
gap: 8px;
|
||||
}
|
||||
.persona-pick-card {
|
||||
background: #1a1a2e; border: 1px solid #0f3460;
|
||||
border-radius: 10px; padding: 10px 8px;
|
||||
cursor: pointer; text-align: center;
|
||||
font-size: 0.8rem; color: #ccc;
|
||||
transition: all 0.15s;
|
||||
}
|
||||
.persona-pick-card:hover { border-color: #9b7fd4; }
|
||||
.persona-pick-card.selected { border-color: #e94560; color: #e94560; background: #1f0a14; }
|
||||
.persona-pick-card .emoji { font-size: 1.5rem; display: block; margin-bottom: 4px; }
|
||||
.rpg-mode-option {
|
||||
display: flex; align-items: center; gap: 8px;
|
||||
font-size: 0.9rem; color: #ccc; cursor: pointer;
|
||||
flex-direction: row !important;
|
||||
padding: 8px 0;
|
||||
}
|
||||
.chat-settings-meta {
|
||||
margin-top: 12px; padding: 10px;
|
||||
background: #1a1a2e; border-radius: 8px;
|
||||
font-size: 0.8rem; color: #888; line-height: 1.5;
|
||||
}
|
||||
.session-item .s-preview {
|
||||
font-size: 0.75rem; color: #666;
|
||||
white-space: nowrap; overflow: hidden; text-overflow: ellipsis;
|
||||
margin-top: 2px;
|
||||
}
|
||||
.session-item .s-row {
|
||||
display: flex; align-items: center; gap: 6px; flex-wrap: wrap;
|
||||
}
|
||||
.session-item .s-badge {
|
||||
font-size: 0.65rem; padding: 1px 5px;
|
||||
border: 1px solid #9b7fd4; border-radius: 4px;
|
||||
color: #9b7fd4;
|
||||
}
|
||||
.session-item .s-date { font-size: 0.7rem; color: #555; margin-left: auto; }
|
||||
.session-item .s-title[contenteditable] {
|
||||
outline: none; border-bottom: 1px dashed #e94560;
|
||||
}
|
||||
.message-actions {
|
||||
display: flex; gap: 4px; margin-top: 4px; flex-wrap: wrap;
|
||||
}
|
||||
.message-actions button {
|
||||
background: #0f3460; border: none; border-radius: 6px;
|
||||
color: #aaa; font-size: 0.7rem; padding: 3px 8px; cursor: pointer;
|
||||
}
|
||||
.message-actions button:hover { color: #e94560; }
|
||||
.message .bubble-edit {
|
||||
width: 100%; min-height: 60px;
|
||||
background: #1a1a2e; border: 1px solid #e94560;
|
||||
border-radius: 8px; color: #e0e0e0; padding: 8px;
|
||||
font-family: inherit; resize: vertical;
|
||||
}
|
||||
|
||||
+217
-27
@@ -2,7 +2,7 @@
|
||||
<html lang="ru">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0, viewport-fit=cover">
|
||||
<title>AI Chat</title>
|
||||
<link rel="stylesheet" href="/static/css/app.css">
|
||||
</head>
|
||||
@@ -12,6 +12,9 @@
|
||||
<button id="sidebarToggle" type="button">☰</button>
|
||||
<h1>🤖 AI Chat</h1>
|
||||
<span class="header-title" id="headerTitle">Новый чат</span>
|
||||
<span id="rpgBadge" class="rpg-badge hidden" title="RPG режим">RPG</span>
|
||||
<button id="chatSettingsBtn" type="button" class="header-icon-btn" title="Настройки чата">⚙️</button>
|
||||
<span id="affinityDisplay" class="affinity-display hidden"></span>
|
||||
</header>
|
||||
|
||||
<div class="app-body">
|
||||
@@ -21,10 +24,21 @@
|
||||
<button id="newChatBtn" type="button">+ Новый</button>
|
||||
</div>
|
||||
<div class="session-list" id="sessionList"></div>
|
||||
<div class="quest-panel hidden" id="questPanel">
|
||||
<div class="quest-panel-header">Квесты</div>
|
||||
<div id="questList"></div>
|
||||
</div>
|
||||
</aside>
|
||||
|
||||
<div class="main">
|
||||
<div class="persona-bar" id="personaBar"></div>
|
||||
<div class="system-blob" id="systemBlob">
|
||||
<div class="system-blob-header">
|
||||
<span>System</span>
|
||||
<button type="button" id="systemBlobToggle">Скрыть</button>
|
||||
</div>
|
||||
<pre class="system-blob-content" id="systemBlobContent">—</pre>
|
||||
</div>
|
||||
<div class="messages" id="messages">
|
||||
<div class="empty-state" id="emptyState">
|
||||
<span class="big">💬</span>
|
||||
@@ -41,33 +55,72 @@
|
||||
</div>
|
||||
|
||||
<div class="modal-overlay" id="modalOverlay">
|
||||
<div class="modal">
|
||||
<h2>✨ Новый персонаж</h2>
|
||||
<label>ID (латиницей)
|
||||
<input type="text" id="pId" placeholder="my_hero">
|
||||
</label>
|
||||
<label>Имя
|
||||
<input type="text" id="pName" placeholder="Мой герой">
|
||||
</label>
|
||||
<label>Эмодзи
|
||||
<input type="text" id="pEmoji" placeholder="🦸" maxlength="4">
|
||||
</label>
|
||||
<label>Описание
|
||||
<input type="text" id="pDesc" placeholder="Краткое описание">
|
||||
</label>
|
||||
<label>Системный промт
|
||||
<textarea id="pPrompt" rows="4" placeholder="Ты — ..."></textarea>
|
||||
</label>
|
||||
<label><input type="checkbox" id="pSdEnabled"> Генерировать SD-промпт</label>
|
||||
<label>LoRA
|
||||
<input type="text" id="pLora" placeholder="CharacterLoRA">
|
||||
</label>
|
||||
<label>Теги внешности (SD)
|
||||
<input type="text" id="pAppearance" placeholder="blue hair, elf ears">
|
||||
</label>
|
||||
<div class="modal-buttons">
|
||||
<div class="modal modal-wizard">
|
||||
<div class="modal-wizard-header">
|
||||
<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>
|
||||
<span class="wizard-step-line"></span>
|
||||
<span class="wizard-step-dot" data-step="3">3</span>
|
||||
</div>
|
||||
</div>
|
||||
<div class="modal-wizard-body">
|
||||
<div class="wizard-page active" data-step="1">
|
||||
<p class="wizard-page-title">Основное</p>
|
||||
<label>ID (латиницей)
|
||||
<input type="text" id="pId" placeholder="my_hero">
|
||||
</label>
|
||||
<label>Имя
|
||||
<input type="text" id="pName" placeholder="Мой герой">
|
||||
</label>
|
||||
<label>Эмодзи
|
||||
<input type="text" id="pEmoji" placeholder="🦸" maxlength="4">
|
||||
</label>
|
||||
<label>Описание
|
||||
<input type="text" id="pDesc" placeholder="Краткое описание">
|
||||
</label>
|
||||
</div>
|
||||
<div class="wizard-page" data-step="2">
|
||||
<p class="wizard-page-title">Характер и сценарий</p>
|
||||
<label>Личность
|
||||
<textarea id="pPersonality" rows="3" placeholder="calm, confident, sarcastic..."></textarea>
|
||||
</label>
|
||||
<label>Сценарий / мир
|
||||
<textarea id="pScenario" rows="3" placeholder="где вы находитесь, что происходит, правила мира"></textarea>
|
||||
</label>
|
||||
<label>Первое сообщение (first_mes)
|
||||
<textarea id="pFirstMes" rows="3" placeholder="приветствие персонажа"></textarea>
|
||||
</label>
|
||||
<label>Пример диалога (mes_example)
|
||||
<textarea id="pMesExample" rows="3" placeholder="пример стиля речи персонажа"></textarea>
|
||||
</label>
|
||||
</div>
|
||||
<div class="wizard-page" data-step="3">
|
||||
<p class="wizard-page-title">Дополнительно</p>
|
||||
<label>Lorebook JSON (опционально)
|
||||
<textarea id="pLorebook" rows="3" placeholder='[]'></textarea>
|
||||
</label>
|
||||
<label>Системный промт (опционально, если пусто — соберём автоматически)
|
||||
<textarea id="pPrompt" rows="3" placeholder=""></textarea>
|
||||
</label>
|
||||
<label><input type="checkbox" id="pSdEnabled"> Генерировать SD-промпт</label>
|
||||
<label>LoRA
|
||||
<input type="text" id="pLora" placeholder="CharacterLoRA">
|
||||
</label>
|
||||
<label>Теги внешности (SD)
|
||||
<input type="text" id="pAppearance" placeholder="blue hair, elf ears">
|
||||
</label>
|
||||
</div>
|
||||
</div>
|
||||
<div class="modal-buttons modal-wizard-footer">
|
||||
<button id="modalCancel" type="button">Отмена</button>
|
||||
<button id="modalSave" type="button">Создать</button>
|
||||
<div class="wizard-nav">
|
||||
<button id="modalPrev" type="button" class="wizard-nav-btn hidden">← Назад</button>
|
||||
<button id="modalNext" type="button" class="wizard-nav-btn">Далее →</button>
|
||||
<button id="modalSave" type="button" class="hidden">Создать</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
@@ -95,6 +148,9 @@
|
||||
<div class="modal" style="max-width:560px;max-height:90vh;overflow-y:auto">
|
||||
<h2>✏️ Редактор карточки</h2>
|
||||
<input type="hidden" id="editCardId">
|
||||
<label>Аватар (PNG)
|
||||
<input type="file" id="editCardAvatar" accept=".png">
|
||||
</label>
|
||||
<label>Имя <input type="text" id="editName"></label>
|
||||
<label>Описание <textarea id="editDescription" rows="4"></textarea></label>
|
||||
<label>Личность <textarea id="editPersonality" rows="3"></textarea></label>
|
||||
@@ -111,6 +167,140 @@
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div class="modal-overlay" id="personaEditOverlay">
|
||||
<div class="modal" style="max-width:560px;max-height:90vh;overflow-y:auto">
|
||||
<h2>✏️ Редактор персонажа</h2>
|
||||
<input type="hidden" id="editPersonaId">
|
||||
<label>Аватар (PNG)
|
||||
<input type="file" id="editPAvatar" accept=".png">
|
||||
</label>
|
||||
<label>Имя <input type="text" id="editPName"></label>
|
||||
<label>Эмодзи <input type="text" id="editPEmoji" maxlength="4"></label>
|
||||
<label>Описание <textarea id="editPDesc" rows="3"></textarea></label>
|
||||
<label>Личность <textarea id="editPPersonality" rows="3"></textarea></label>
|
||||
<label>Сценарий <textarea id="editPScenario" rows="3"></textarea></label>
|
||||
<label>Первое сообщение <textarea id="editPFirstMes" rows="3"></textarea></label>
|
||||
<label>Пример диалога <textarea id="editPMesExample" rows="3"></textarea></label>
|
||||
<label>Lorebook JSON <textarea id="editPLorebook" rows="3"></textarea></label>
|
||||
<label>Системный промпт (опционально) <textarea id="editPPrompt" rows="3"></textarea></label>
|
||||
<label><input type="checkbox" id="editPSdEnabled"> Генерировать SD-промпт</label>
|
||||
<label>LoRA <input type="text" id="editPLora"></label>
|
||||
<label>Вес LoRA <input type="number" id="editPLoraWeight" value="0.8" min="0" max="2" step="0.1"></label>
|
||||
<label>Теги внешности (SD) <input type="text" id="editPAppearance"></label>
|
||||
<div class="modal-buttons">
|
||||
<button id="personaEditCancel" type="button">Отмена</button>
|
||||
<button id="personaEditSave" type="button" style="background:#e94560;color:white">Сохранить</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div class="modal-overlay" id="newChatModal">
|
||||
<div class="modal modal-wizard">
|
||||
<div class="modal-wizard-header">
|
||||
<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>
|
||||
<span class="wizard-step-line"></span>
|
||||
<span class="wizard-step-dot" data-step="3">3</span>
|
||||
</div>
|
||||
</div>
|
||||
<div class="modal-wizard-body">
|
||||
<div class="wizard-page active" data-step="1">
|
||||
<p class="wizard-page-title">Персонаж</p>
|
||||
<div class="persona-pick-grid" id="newChatPersonaGrid"></div>
|
||||
</div>
|
||||
<div class="wizard-page" data-step="2">
|
||||
<p class="wizard-page-title">Режим</p>
|
||||
<label class="rpg-mode-option">
|
||||
<input type="radio" name="newChatRpg" value="0" checked> Обычный чат
|
||||
</label>
|
||||
<label class="rpg-mode-option">
|
||||
<input type="radio" name="newChatRpg" value="1"> RPG режим
|
||||
</label>
|
||||
</div>
|
||||
<div class="wizard-page" data-step="3">
|
||||
<div id="newChatPlainStep">
|
||||
<p class="wizard-page-title">Название (опционально)</p>
|
||||
<label>Название чата
|
||||
<input type="text" id="newChatTitle" placeholder="Оставь пустым — сгенерируем автоматически">
|
||||
</label>
|
||||
</div>
|
||||
<div id="newChatRpgStep" class="hidden">
|
||||
<p class="wizard-page-title">Жанры и настройки RPG</p>
|
||||
<p class="wizard-hint">Можно выбрать несколько жанров</p>
|
||||
<div class="genre-grid" id="newChatGenreGrid">
|
||||
<button type="button" class="genre-btn" data-genre="adventure">⚔️ Приключение</button>
|
||||
<button type="button" class="genre-btn" data-genre="horror">👻 Хоррор</button>
|
||||
<button type="button" class="genre-btn" data-genre="romance">💕 Романтика</button>
|
||||
<button type="button" class="genre-btn" data-genre="slice_of_life">☕ Повседневность</button>
|
||||
<button type="button" class="genre-btn" data-genre="fantasy">🧙 Фэнтези</button>
|
||||
<button type="button" class="genre-btn" data-genre="sci_fi">🚀 Sci-Fi</button>
|
||||
</div>
|
||||
<p class="selected-genres-label hidden" id="newChatGenresLabel"></p>
|
||||
<div class="rpg-settings-grid" style="margin-top:12px">
|
||||
<label><input type="checkbox" id="ncSettingDice" checked> 🎲 Проверки d20</label>
|
||||
<label><input type="checkbox" id="ncSettingNarrator" checked> 📖 Нарратор</label>
|
||||
<label><input type="checkbox" id="ncSettingQuests" checked> 📜 Квесты</label>
|
||||
<label><input type="checkbox" id="ncSettingAffinity" checked> 💖 Симпатия</label>
|
||||
<label><input type="checkbox" id="ncSettingChoices" checked> 🔘 Кнопки выбора</label>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
<div class="modal-buttons modal-wizard-footer">
|
||||
<button id="newChatCancel" type="button">Отмена</button>
|
||||
<div class="wizard-nav">
|
||||
<button id="newChatPrev" type="button" class="wizard-nav-btn hidden">← Назад</button>
|
||||
<button id="newChatNext" type="button" class="wizard-nav-btn">Далее →</button>
|
||||
<button id="newChatCreate" type="button" class="hidden" style="background:#e94560;color:white">Создать</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div class="modal-overlay" id="chatSettingsModal">
|
||||
<div class="modal modal-wizard" style="max-width:520px">
|
||||
<div class="modal-wizard-header">
|
||||
<h2>⚙️ Настройки чата</h2>
|
||||
</div>
|
||||
<div class="modal-wizard-body">
|
||||
<label>Название чата
|
||||
<input type="text" id="chatSettingsTitle">
|
||||
</label>
|
||||
<label class="rpg-mode-option">
|
||||
<input type="checkbox" id="chatSettingsRpg"> RPG режим
|
||||
</label>
|
||||
<div id="chatSettingsRpgBlock" class="hidden">
|
||||
<p class="wizard-page-title">Жанры</p>
|
||||
<div class="genre-grid" id="chatSettingsGenreGrid">
|
||||
<button type="button" class="genre-btn" data-genre="adventure">⚔️ Приключение</button>
|
||||
<button type="button" class="genre-btn" data-genre="horror">👻 Хоррор</button>
|
||||
<button type="button" class="genre-btn" data-genre="romance">💕 Романтика</button>
|
||||
<button type="button" class="genre-btn" data-genre="slice_of_life">☕ Повседневность</button>
|
||||
<button type="button" class="genre-btn" data-genre="fantasy">🧙 Фэнтези</button>
|
||||
<button type="button" class="genre-btn" data-genre="sci_fi">🚀 Sci-Fi</button>
|
||||
</div>
|
||||
<p class="selected-genres-label hidden" id="chatSettingsGenresLabel"></p>
|
||||
<p class="wizard-page-title" style="margin-top:12px">Настройки RPG</p>
|
||||
<div class="rpg-settings-grid">
|
||||
<label><input type="checkbox" id="csSettingDice"> 🎲 Проверки d20</label>
|
||||
<label><input type="checkbox" id="csSettingNarrator"> 📖 Нарратор</label>
|
||||
<label><input type="checkbox" id="csSettingQuests"> 📜 Квесты</label>
|
||||
<label><input type="checkbox" id="csSettingAffinity"> 💖 Симпатия</label>
|
||||
<label><input type="checkbox" id="csSettingChoices"> 🔘 Кнопки выбора</label>
|
||||
</div>
|
||||
<div class="chat-settings-meta" id="chatSettingsMeta"></div>
|
||||
</div>
|
||||
</div>
|
||||
<div class="modal-buttons modal-wizard-footer">
|
||||
<button id="chatSettingsCancel" type="button">Отмена</button>
|
||||
<button id="chatSettingsSave" type="button" style="background:#e94560;color:white">Сохранить</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<script type="module" src="/static/js/app.js"></script>
|
||||
</body>
|
||||
</html>
|
||||
|
||||
+12
-2
@@ -1,5 +1,7 @@
|
||||
import { toggleSidebar, dom } from './state.js';
|
||||
import { initSessions, createNewChat } from './sessions.js';
|
||||
import {
|
||||
initSessions, openNewChatWizard, initNewChatWizard, initChatSettings, openChatSettings,
|
||||
} from './sessions.js';
|
||||
import { loadPersonas, initPersonaModals } from './personas.js';
|
||||
import { sendMessage, clearHistory } from './chat.js';
|
||||
|
||||
@@ -8,7 +10,8 @@ document.getElementById('sidebarToggle').addEventListener('click', () => {
|
||||
document.getElementById('sidebar').classList.toggle('collapsed', !open);
|
||||
});
|
||||
|
||||
document.getElementById('newChatBtn').addEventListener('click', createNewChat);
|
||||
document.getElementById('newChatBtn').addEventListener('click', openNewChatWizard);
|
||||
document.getElementById('chatSettingsBtn')?.addEventListener('click', openChatSettings);
|
||||
|
||||
dom.inputEl.addEventListener('input', () => {
|
||||
dom.inputEl.style.height = 'auto';
|
||||
@@ -25,6 +28,13 @@ dom.inputEl.addEventListener('keydown', (e) => {
|
||||
dom.sendBtn.addEventListener('click', sendMessage);
|
||||
dom.clearBtn.addEventListener('click', clearHistory);
|
||||
|
||||
dom.systemBlobToggle?.addEventListener('click', () => {
|
||||
const hidden = dom.systemBlobContent.classList.toggle('hidden');
|
||||
dom.systemBlobToggle.textContent = hidden ? 'Показать' : 'Скрыть';
|
||||
});
|
||||
|
||||
initPersonaModals();
|
||||
initNewChatWizard();
|
||||
initChatSettings();
|
||||
await initSessions();
|
||||
loadPersonas();
|
||||
|
||||
+331
-54
@@ -53,6 +53,102 @@ export function createImagePromptBlock(promptText) {
|
||||
return block;
|
||||
}
|
||||
|
||||
const OUTCOME_CLASS = {
|
||||
'critical failure': 'outcome-crit-fail',
|
||||
'failure': 'outcome-fail',
|
||||
'success': 'outcome-success',
|
||||
'critical success': 'outcome-crit-success',
|
||||
};
|
||||
|
||||
function renderNarratorMessage(narrator) {
|
||||
// narrator = { roll, outcome, text }
|
||||
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);
|
||||
dom.messagesEl.appendChild(wrapper);
|
||||
dom.messagesEl.scrollTop = dom.messagesEl.scrollHeight;
|
||||
return wrapper;
|
||||
}
|
||||
|
||||
function renderChoices(wrapper, choices) {
|
||||
if (!choices || !choices.length) return;
|
||||
const row = document.createElement('div');
|
||||
row.className = 'choice-row';
|
||||
for (const c of choices) {
|
||||
const btn = document.createElement('button');
|
||||
btn.type = 'button';
|
||||
btn.className = 'choice-btn';
|
||||
btn.textContent = c.label;
|
||||
btn.addEventListener('click', () => {
|
||||
sendMessage(c.label, true);
|
||||
});
|
||||
row.appendChild(btn);
|
||||
}
|
||||
wrapper.appendChild(row);
|
||||
}
|
||||
|
||||
function renderDebugBlocks(wrapper, blocks) {
|
||||
if (!blocks || !blocks.length) return;
|
||||
for (const b of blocks) {
|
||||
if (!b?.text) continue;
|
||||
if (b.type === 'narrator_injection') {
|
||||
// Show beat injections as narrator bubbles (no dice)
|
||||
const w = document.createElement('div');
|
||||
w.className = 'message narrator';
|
||||
const lbl = document.createElement('div');
|
||||
lbl.className = 'label';
|
||||
lbl.textContent = '📖 Рассказчик';
|
||||
const bub = document.createElement('div');
|
||||
bub.className = 'bubble';
|
||||
bub.textContent = b.text;
|
||||
w.appendChild(lbl);
|
||||
w.appendChild(bub);
|
||||
dom.messagesEl.appendChild(w);
|
||||
}
|
||||
// facts/status_quo/plot_arc — silently skip (debug only, not shown to user)
|
||||
}
|
||||
}
|
||||
|
||||
export function updateQuestPanel(quests) {
|
||||
const list = document.getElementById('questList');
|
||||
if (!list) return;
|
||||
list.innerHTML = '';
|
||||
for (const q of quests) {
|
||||
const el = document.createElement('div');
|
||||
el.className = `quest-item quest-${q.status}`;
|
||||
el.textContent = (q.status === 'done' ? '✅ ' : q.status === 'failed' ? '❌ ' : '🔸 ') + q.title;
|
||||
list.appendChild(el);
|
||||
}
|
||||
}
|
||||
|
||||
export function updateAffinityDisplay(affinity) {
|
||||
const el = dom.affinityDisplay;
|
||||
if (!el) return;
|
||||
el.classList.remove('hidden');
|
||||
const hearts = affinity >= 10 ? '❤️❤️❤️' : affinity >= 5 ? '❤️❤️' : affinity >= 1 ? '❤️' : affinity <= -5 ? '💔' : '🤍';
|
||||
el.textContent = `${hearts} ${affinity > 0 ? '+' : ''}${affinity}`;
|
||||
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();
|
||||
@@ -86,7 +182,208 @@ export function appendChatImage(wrapper, imagePath) {
|
||||
wrapper.appendChild(img);
|
||||
}
|
||||
|
||||
export function addMessage(role, content = '', imagePrompt = null, imagePath = null) {
|
||||
function attachMessageActions(wrapper, messageId, role) {
|
||||
if (!messageId) return;
|
||||
wrapper.dataset.messageId = String(messageId);
|
||||
|
||||
const actions = document.createElement('div');
|
||||
actions.className = 'message-actions';
|
||||
|
||||
const editBtn = document.createElement('button');
|
||||
editBtn.type = 'button';
|
||||
editBtn.textContent = '✏️';
|
||||
editBtn.title = 'Редактировать';
|
||||
editBtn.addEventListener('click', () => startEditMessage(wrapper, messageId));
|
||||
actions.appendChild(editBtn);
|
||||
|
||||
if (role === 'assistant') {
|
||||
const regenBtn = document.createElement('button');
|
||||
regenBtn.type = 'button';
|
||||
regenBtn.textContent = '🔄';
|
||||
regenBtn.title = 'Перегенерировать';
|
||||
regenBtn.addEventListener('click', () => regenerateMessage(messageId, wrapper));
|
||||
actions.appendChild(regenBtn);
|
||||
}
|
||||
|
||||
const branchBtn = document.createElement('button');
|
||||
branchBtn.type = 'button';
|
||||
branchBtn.textContent = '🌿';
|
||||
branchBtn.title = 'Ветка отсюда';
|
||||
branchBtn.addEventListener('click', () => forkFromMessage(messageId));
|
||||
actions.appendChild(branchBtn);
|
||||
|
||||
wrapper.appendChild(actions);
|
||||
}
|
||||
|
||||
async function startEditMessage(wrapper, messageId) {
|
||||
const bubble = wrapper.querySelector('.bubble');
|
||||
if (!bubble || wrapper.querySelector('.bubble-edit')) return;
|
||||
const original = bubble.textContent;
|
||||
const ta = document.createElement('textarea');
|
||||
ta.className = 'bubble-edit';
|
||||
ta.value = original;
|
||||
bubble.replaceWith(ta);
|
||||
wrapper.querySelector('.message-actions')?.remove();
|
||||
|
||||
const saveRow = document.createElement('div');
|
||||
saveRow.className = 'message-actions';
|
||||
const saveBtn = document.createElement('button');
|
||||
saveBtn.textContent = 'Сохранить';
|
||||
const cancelBtn = document.createElement('button');
|
||||
cancelBtn.textContent = 'Отмена';
|
||||
saveRow.appendChild(saveBtn);
|
||||
saveRow.appendChild(cancelBtn);
|
||||
wrapper.appendChild(saveRow);
|
||||
|
||||
const truncate = role => confirm(
|
||||
role === 'user'
|
||||
? 'Удалить все сообщения после этого? (рекомендуется)'
|
||||
: 'Удалить все сообщения после этого?',
|
||||
);
|
||||
|
||||
cancelBtn.addEventListener('click', () => reloadChatFromServer(sessionId));
|
||||
saveBtn.addEventListener('click', async () => {
|
||||
const role = wrapper.classList.contains('user') ? 'user' : 'assistant';
|
||||
const doTruncate = truncate(role);
|
||||
const res = await fetch(`/chat/messages/${messageId}`, {
|
||||
method: 'PATCH',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ content: ta.value.trim(), truncate_after: doTruncate }),
|
||||
});
|
||||
if (!res.ok) { alert('Ошибка сохранения'); return; }
|
||||
await reloadChatFromServer(sessionId);
|
||||
const { loadSessions } = await import('./sessions.js');
|
||||
loadSessions();
|
||||
});
|
||||
}
|
||||
|
||||
async function regenerateMessage(messageId, wrapper) {
|
||||
if (!sessionId) return;
|
||||
wrapper?.remove();
|
||||
showTyping();
|
||||
dom.sendBtn.disabled = true;
|
||||
try {
|
||||
const res = await fetch('/chat/regenerate', {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({
|
||||
session_id: sessionId,
|
||||
persona_id: currentPersona,
|
||||
message_id: messageId,
|
||||
}),
|
||||
});
|
||||
if (!res.ok) throw new Error('Ошибка: ' + res.status);
|
||||
removeTyping();
|
||||
await consumeStream(res);
|
||||
} catch (err) {
|
||||
removeTyping();
|
||||
addMessage('assistant', '⚠️ ' + err.message);
|
||||
} finally {
|
||||
dom.sendBtn.disabled = false;
|
||||
}
|
||||
}
|
||||
|
||||
async function forkFromMessage(messageId) {
|
||||
if (!sessionId) return;
|
||||
const res = await fetch(`/sessions/${sessionId}/fork`, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ until_message_id: messageId }),
|
||||
});
|
||||
const data = await res.json();
|
||||
if (!res.ok) { alert(data.detail || 'Ошибка'); return; }
|
||||
const { switchSession, loadSessions } = await import('./sessions.js');
|
||||
await switchSession(data.session_id);
|
||||
await loadSessions();
|
||||
}
|
||||
|
||||
export async function reloadChatFromServer(id) {
|
||||
const sid = id || sessionId;
|
||||
if (!sid) return;
|
||||
const histRes = await fetch(`/chat/history/${sid}`);
|
||||
if (!histRes.ok) return;
|
||||
const messages = await histRes.json();
|
||||
clearMessages();
|
||||
messages.filter(m => m.role !== 'system').forEach(m => {
|
||||
addMessage(
|
||||
m.role === 'user' ? 'user' : 'assistant',
|
||||
m.content,
|
||||
m.image_prompt,
|
||||
m.image_path ? `/static/${m.image_path}` : null,
|
||||
m.id,
|
||||
);
|
||||
});
|
||||
}
|
||||
|
||||
async function consumeStream(res) {
|
||||
const reader = res.body.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let buffer = '';
|
||||
let bubble = null;
|
||||
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
|
||||
buffer += decoder.decode(value, { stream: true });
|
||||
const lines = buffer.split('\n');
|
||||
buffer = lines.pop();
|
||||
|
||||
for (const line of lines) {
|
||||
if (!line.startsWith('data: ')) continue;
|
||||
try {
|
||||
const data = JSON.parse(line.slice(6));
|
||||
if (data.chunk !== undefined) {
|
||||
if (!bubble) {
|
||||
bubble = addMessage('assistant', '');
|
||||
bubble.classList.add('typing-active');
|
||||
}
|
||||
bubble.textContent += data.chunk;
|
||||
bubble.textContent = bubble.textContent.replace(/\[IMAGE_PROMPT:.*?\]/gs, '').trim();
|
||||
dom.messagesEl.scrollTop = dom.messagesEl.scrollHeight;
|
||||
}
|
||||
if (data.done) {
|
||||
bubble?.classList.remove('typing-active');
|
||||
if (data.narrator && !bubble) {
|
||||
renderNarratorMessage(data.narrator);
|
||||
} else if (data.narrator && bubble) {
|
||||
const assistantWrapper = bubble.parentElement;
|
||||
dom.messagesEl.insertBefore(buildNarratorWrapper(data.narrator), assistantWrapper);
|
||||
}
|
||||
if (data.image_prompt && bubble) {
|
||||
bubble.parentElement.appendChild(createImagePromptBlock(data.image_prompt));
|
||||
}
|
||||
if (data.image_path && bubble) {
|
||||
appendChatImage(bubble.parentElement, data.image_path);
|
||||
}
|
||||
if (data.image_error && bubble) {
|
||||
const err = document.createElement('div');
|
||||
err.className = 'image-error';
|
||||
err.textContent = '🖼 ' + data.image_error;
|
||||
bubble.parentElement.appendChild(err);
|
||||
}
|
||||
if (data.choices && bubble) {
|
||||
renderChoices(bubble.parentElement, data.choices);
|
||||
}
|
||||
if (data.debug) {
|
||||
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');
|
||||
loadSessions();
|
||||
}
|
||||
} catch { /* skip */ }
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
export function addMessage(role, content = '', imagePrompt = null, imagePath = null, messageId = null) {
|
||||
updateEmptyState();
|
||||
|
||||
const wrapper = document.createElement('div');
|
||||
@@ -149,6 +446,8 @@ export function addMessage(role, content = '', imagePrompt = null, imagePath = n
|
||||
if (prompt) wrapper.appendChild(createImagePromptBlock(prompt));
|
||||
if (imagePath) appendChatImage(wrapper, imagePath);
|
||||
|
||||
attachMessageActions(wrapper, messageId, role);
|
||||
|
||||
dom.messagesEl.appendChild(wrapper);
|
||||
dom.messagesEl.scrollTop = dom.messagesEl.scrollHeight;
|
||||
return bubble;
|
||||
@@ -175,75 +474,32 @@ export function clearMessages() {
|
||||
}
|
||||
}
|
||||
|
||||
export async function sendMessage() {
|
||||
const text = dom.inputEl.value.trim();
|
||||
export async function sendMessage(text, isNarratorChoice = false) {
|
||||
if (typeof text !== 'string') text = dom.inputEl.value.trim();
|
||||
if (!text || !sessionId) return;
|
||||
|
||||
dom.inputEl.value = '';
|
||||
dom.inputEl.style.height = 'auto';
|
||||
dom.sendBtn.disabled = true;
|
||||
|
||||
addMessage('user', text);
|
||||
addMessage('user', isNarratorChoice ? `[${text}]` : text);
|
||||
showTyping();
|
||||
|
||||
try {
|
||||
const res = await fetch('/chat/stream', {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ message: text, session_id: sessionId, persona_id: currentPersona }),
|
||||
body: JSON.stringify({
|
||||
message: text,
|
||||
session_id: sessionId,
|
||||
persona_id: currentPersona,
|
||||
is_narrator_choice: isNarratorChoice,
|
||||
}),
|
||||
});
|
||||
if (!res.ok) throw new Error('Ошибка сервера: ' + res.status);
|
||||
|
||||
const reader = res.body.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let buffer = '';
|
||||
let bubble = null;
|
||||
|
||||
removeTyping();
|
||||
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
|
||||
buffer += decoder.decode(value, { stream: true });
|
||||
const lines = buffer.split('\n');
|
||||
buffer = lines.pop();
|
||||
|
||||
for (const line of lines) {
|
||||
if (!line.startsWith('data: ')) continue;
|
||||
try {
|
||||
const data = JSON.parse(line.slice(6));
|
||||
|
||||
if (data.chunk !== undefined) {
|
||||
if (!bubble) {
|
||||
bubble = addMessage('assistant', '');
|
||||
bubble.classList.add('typing-active');
|
||||
}
|
||||
bubble.textContent += data.chunk;
|
||||
bubble.textContent = bubble.textContent.replace(/\[IMAGE_PROMPT:.*?\]/gs, '').trim();
|
||||
dom.messagesEl.scrollTop = dom.messagesEl.scrollHeight;
|
||||
}
|
||||
|
||||
if (data.done) {
|
||||
bubble?.classList.remove('typing-active');
|
||||
if (data.image_prompt && bubble) {
|
||||
bubble.parentElement.appendChild(createImagePromptBlock(data.image_prompt));
|
||||
}
|
||||
if (data.image_path && bubble) {
|
||||
appendChatImage(bubble.parentElement, data.image_path);
|
||||
}
|
||||
if (data.image_error && bubble) {
|
||||
const err = document.createElement('div');
|
||||
err.className = 'image-error';
|
||||
err.textContent = '🖼 ' + data.image_error;
|
||||
bubble.parentElement.appendChild(err);
|
||||
}
|
||||
const { loadSessions } = await import('./sessions.js');
|
||||
loadSessions();
|
||||
}
|
||||
} catch { /* skip */ }
|
||||
}
|
||||
}
|
||||
await consumeStream(res);
|
||||
} catch (err) {
|
||||
removeTyping();
|
||||
addMessage('assistant', '⚠️ Ошибка: ' + err.message);
|
||||
@@ -253,6 +509,27 @@ export async function sendMessage() {
|
||||
}
|
||||
}
|
||||
|
||||
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() {
|
||||
if (!sessionId) return;
|
||||
await fetch(`/chat/${sessionId}`, { method: 'DELETE' });
|
||||
|
||||
+116
-5
@@ -1,5 +1,10 @@
|
||||
import { currentPersona, setCurrentPersona, sessionId } from './state.js';
|
||||
import { initChat } from './chat.js';
|
||||
import { initWizard } from './utils.js';
|
||||
|
||||
export let personaIndex = new Map();
|
||||
|
||||
let createWizard;
|
||||
|
||||
export function highlightPersona(personaId) {
|
||||
document.querySelectorAll('.persona-card').forEach(c => {
|
||||
@@ -10,6 +15,7 @@ export function highlightPersona(personaId) {
|
||||
export async function loadPersonas() {
|
||||
const res = await fetch('/personas/');
|
||||
const personas = await res.json();
|
||||
personaIndex = new Map(personas.map(p => [p.persona_id, p]));
|
||||
const bar = document.getElementById('personaBar');
|
||||
bar.innerHTML = '';
|
||||
|
||||
@@ -18,11 +24,14 @@ export async function loadPersonas() {
|
||||
card.className = 'persona-card' + (p.persona_id === currentPersona ? ' active' : '');
|
||||
card.dataset.id = p.persona_id;
|
||||
const isCard = p.persona_id.startsWith('card_');
|
||||
const isCustomPersona = p.custom && !isCard;
|
||||
const avatar = p.avatar_path ? `/static/${p.avatar_path}` : '';
|
||||
card.innerHTML = `
|
||||
<span class="emoji">${p.emoji}</span>
|
||||
${avatar ? `<img class="avatar" src="${avatar}" alt="">` : `<span class="emoji">${p.emoji}</span>`}
|
||||
<span class="pname">${p.name}</span>
|
||||
${p.custom ? `<button class="del-btn" type="button">✕</button>` : ''}
|
||||
${isCard ? `<button class="edit-btn" type="button">✏️</button>` : ''}
|
||||
${isCustomPersona ? `<button class="edit-persona-btn" type="button">✏️</button>` : ''}
|
||||
`;
|
||||
card.addEventListener('click', () => selectPersona(p.persona_id));
|
||||
card.querySelector('.del-btn')?.addEventListener('click', async (e) => {
|
||||
@@ -48,6 +57,27 @@ export async function loadPersonas() {
|
||||
document.getElementById('editLoraWeight').value = data.lora_weight ?? 0.8;
|
||||
document.getElementById('cardEditOverlay').classList.add('open');
|
||||
});
|
||||
|
||||
card.querySelector('.edit-persona-btn')?.addEventListener('click', async (e) => {
|
||||
e.stopPropagation();
|
||||
const r = await fetch(`/personas/${p.persona_id}`);
|
||||
const data = await r.json();
|
||||
document.getElementById('editPersonaId').value = p.persona_id;
|
||||
document.getElementById('editPName').value = data.name || '';
|
||||
document.getElementById('editPEmoji').value = data.emoji || '';
|
||||
document.getElementById('editPDesc').value = data.description || '';
|
||||
document.getElementById('editPPersonality').value = data.personality || '';
|
||||
document.getElementById('editPScenario').value = data.scenario || '';
|
||||
document.getElementById('editPFirstMes').value = data.first_mes || '';
|
||||
document.getElementById('editPMesExample').value = data.mes_example || '';
|
||||
document.getElementById('editPLorebook').value = data.lorebook_json || '[]';
|
||||
document.getElementById('editPPrompt').value = data.prompt || '';
|
||||
document.getElementById('editPSdEnabled').checked = !!data.sd_enabled;
|
||||
document.getElementById('editPLora').value = data.lora_name || '';
|
||||
document.getElementById('editPLoraWeight').value = data.lora_weight ?? 0.8;
|
||||
document.getElementById('editPAppearance').value = data.appearance_tags || '';
|
||||
document.getElementById('personaEditOverlay').classList.add('open');
|
||||
});
|
||||
bar.appendChild(card);
|
||||
});
|
||||
|
||||
@@ -55,7 +85,10 @@ export async function loadPersonas() {
|
||||
addBtn.type = 'button';
|
||||
addBtn.className = 'persona-add';
|
||||
addBtn.innerHTML = '➕<span>Создать</span>';
|
||||
addBtn.addEventListener('click', () => document.getElementById('modalOverlay').classList.add('open'));
|
||||
addBtn.addEventListener('click', () => {
|
||||
document.getElementById('modalOverlay').classList.add('open');
|
||||
createWizard?.reset();
|
||||
});
|
||||
bar.appendChild(addBtn);
|
||||
|
||||
const importBtn = document.createElement('button');
|
||||
@@ -80,8 +113,24 @@ export async function selectPersona(personaId) {
|
||||
}
|
||||
|
||||
export function initPersonaModals() {
|
||||
const createModal = document.getElementById('modalOverlay');
|
||||
createWizard = initWizard(createModal.querySelector('.modal-wizard'), {
|
||||
totalSteps: 3,
|
||||
validateStep(step) {
|
||||
if (step !== 1) return true;
|
||||
const id = document.getElementById('pId').value.trim();
|
||||
const name = document.getElementById('pName').value.trim();
|
||||
if (!id || !name) {
|
||||
alert('Заполни ID и имя');
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
},
|
||||
});
|
||||
|
||||
document.getElementById('modalCancel').addEventListener('click', () => {
|
||||
document.getElementById('modalOverlay').classList.remove('open');
|
||||
createModal.classList.remove('open');
|
||||
createWizard.reset();
|
||||
});
|
||||
document.getElementById('cardModalCancel').addEventListener('click', () => {
|
||||
document.getElementById('cardModalOverlay').classList.remove('open');
|
||||
@@ -90,6 +139,14 @@ export function initPersonaModals() {
|
||||
document.getElementById('cardEditOverlay').classList.remove('open');
|
||||
});
|
||||
|
||||
// custom persona editor (reuses create modal fields)
|
||||
const personaEditCancel = document.getElementById('personaEditCancel');
|
||||
if (personaEditCancel) {
|
||||
personaEditCancel.addEventListener('click', () => {
|
||||
document.getElementById('personaEditOverlay').classList.remove('open');
|
||||
});
|
||||
}
|
||||
|
||||
document.getElementById('modalSave').addEventListener('click', async () => {
|
||||
const data = {
|
||||
persona_id: document.getElementById('pId').value.trim(),
|
||||
@@ -100,9 +157,14 @@ export function initPersonaModals() {
|
||||
sd_enabled: document.getElementById('pSdEnabled').checked,
|
||||
lora_name: document.getElementById('pLora').value.trim(),
|
||||
appearance_tags: document.getElementById('pAppearance').value.trim(),
|
||||
personality: document.getElementById('pPersonality').value.trim(),
|
||||
scenario: document.getElementById('pScenario').value.trim(),
|
||||
first_mes: document.getElementById('pFirstMes').value.trim(),
|
||||
mes_example: document.getElementById('pMesExample').value.trim(),
|
||||
lorebook_json: document.getElementById('pLorebook').value.trim() || '[]',
|
||||
};
|
||||
if (!data.persona_id || !data.name || !data.prompt) {
|
||||
alert('Заполни ID, имя и промт');
|
||||
if (!data.persona_id || !data.name) {
|
||||
alert('Заполни ID и имя');
|
||||
return;
|
||||
}
|
||||
await fetch('/personas/', {
|
||||
@@ -111,6 +173,7 @@ export function initPersonaModals() {
|
||||
body: JSON.stringify(data),
|
||||
});
|
||||
document.getElementById('modalOverlay').classList.remove('open');
|
||||
createWizard.reset();
|
||||
await loadPersonas();
|
||||
await selectPersona(data.persona_id);
|
||||
});
|
||||
@@ -134,6 +197,15 @@ export function initPersonaModals() {
|
||||
body: JSON.stringify(body),
|
||||
});
|
||||
if (!res.ok) { alert('Ошибка сохранения'); return; }
|
||||
|
||||
const avatarFile = document.getElementById('editCardAvatar')?.files?.[0];
|
||||
if (avatarFile) {
|
||||
const form = new FormData();
|
||||
form.append('file', avatarFile);
|
||||
await fetch(`/characters/${cardId}/avatar`, { method: 'POST', body: form });
|
||||
document.getElementById('editCardAvatar').value = '';
|
||||
}
|
||||
|
||||
document.getElementById('cardEditOverlay').classList.remove('open');
|
||||
await loadPersonas();
|
||||
});
|
||||
@@ -160,5 +232,44 @@ export function initPersonaModals() {
|
||||
await loadPersonas();
|
||||
await selectPersona(data.persona_id);
|
||||
});
|
||||
|
||||
const personaEditSave = document.getElementById('personaEditSave');
|
||||
if (personaEditSave) {
|
||||
personaEditSave.addEventListener('click', async () => {
|
||||
const personaId = document.getElementById('editPersonaId').value;
|
||||
const body = {
|
||||
name: document.getElementById('editPName').value.trim() || undefined,
|
||||
emoji: document.getElementById('editPEmoji').value.trim() || undefined,
|
||||
description: document.getElementById('editPDesc').value.trim() || undefined,
|
||||
personality: document.getElementById('editPPersonality').value.trim() || undefined,
|
||||
scenario: document.getElementById('editPScenario').value.trim() || undefined,
|
||||
first_mes: document.getElementById('editPFirstMes').value.trim() || undefined,
|
||||
mes_example: document.getElementById('editPMesExample').value.trim() || undefined,
|
||||
lorebook_json: document.getElementById('editPLorebook').value.trim() || undefined,
|
||||
prompt: document.getElementById('editPPrompt').value.trim() || undefined,
|
||||
sd_enabled: document.getElementById('editPSdEnabled').checked,
|
||||
lora_name: document.getElementById('editPLora').value.trim() || undefined,
|
||||
lora_weight: parseFloat(document.getElementById('editPLoraWeight').value) || undefined,
|
||||
appearance_tags: document.getElementById('editPAppearance').value.trim() || undefined,
|
||||
};
|
||||
const res = await fetch(`/personas/${personaId}`, {
|
||||
method: 'PATCH',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify(body),
|
||||
});
|
||||
if (!res.ok) { alert('Ошибка сохранения'); return; }
|
||||
|
||||
const avatarFile = document.getElementById('editPAvatar')?.files?.[0];
|
||||
if (avatarFile) {
|
||||
const form = new FormData();
|
||||
form.append('file', avatarFile);
|
||||
await fetch(`/personas/${personaId}/avatar`, { method: 'POST', body: form });
|
||||
document.getElementById('editPAvatar').value = '';
|
||||
}
|
||||
|
||||
document.getElementById('personaEditOverlay').classList.remove('open');
|
||||
await loadPersonas();
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
+400
-30
@@ -1,6 +1,13 @@
|
||||
import { sessionId, setSessionId, setCurrentPersona, currentPersona, dom } from './state.js';
|
||||
import { clearMessages, addMessage, initChat } from './chat.js';
|
||||
import { highlightPersona } from './personas.js';
|
||||
import {
|
||||
sessionId, setSessionId, setCurrentPersona, currentPersona, dom, setRpgEnabled,
|
||||
} from './state.js';
|
||||
import {
|
||||
clearMessages, addMessage, initChat, updateQuestPanel, updateAffinityDisplay, reloadChatFromServer,
|
||||
} from './chat.js';
|
||||
import { highlightPersona, personaIndex, loadPersonas } from './personas.js';
|
||||
import {
|
||||
initWizard, GENRE_LABELS, bindGenreGrid, resetGenreGrid, formatSessionDate,
|
||||
} from './utils.js';
|
||||
|
||||
function escapeTitle(t) {
|
||||
const d = document.createElement('div');
|
||||
@@ -8,6 +15,38 @@ function escapeTitle(t) {
|
||||
return d.innerHTML;
|
||||
}
|
||||
|
||||
let newChatPersonaId = currentPersona;
|
||||
const newChatGenres = new Set();
|
||||
const chatSettingsGenres = new Set();
|
||||
|
||||
export function applySessionUi(session) {
|
||||
if (!session) return;
|
||||
dom.headerTitle.textContent = session.title || 'Новый чат';
|
||||
const rpgOn = !!session.rpg_enabled;
|
||||
setRpgEnabled(rpgOn);
|
||||
document.getElementById('rpgBadge')?.classList.toggle('hidden', !rpgOn);
|
||||
|
||||
let settings = { quests: true, affinity: true };
|
||||
try {
|
||||
settings = { ...settings, ...JSON.parse(session.rpg_settings_json || '{}') };
|
||||
} catch { /* ignore */ }
|
||||
|
||||
document.getElementById('questPanel')?.classList.toggle('hidden', !rpgOn || !settings.quests);
|
||||
if (rpgOn && settings.affinity) {
|
||||
updateAffinityDisplay(session.affinity ?? 0);
|
||||
dom.affinityDisplay?.classList.remove('hidden');
|
||||
} else {
|
||||
dom.affinityDisplay?.classList.add('hidden');
|
||||
}
|
||||
|
||||
if (rpgOn && settings.quests) {
|
||||
fetch(`/sessions/${session.session_id}/quests`)
|
||||
.then(r => r.ok ? r.json() : [])
|
||||
.then(q => updateQuestPanel(q))
|
||||
.catch(() => {});
|
||||
}
|
||||
}
|
||||
|
||||
export async function loadSessions() {
|
||||
const res = await fetch('/sessions/');
|
||||
const sessions = await res.json();
|
||||
@@ -16,16 +55,49 @@ export async function loadSessions() {
|
||||
sessions.forEach(s => {
|
||||
const item = document.createElement('div');
|
||||
item.className = 'session-item' + (s.session_id === sessionId ? ' active' : '');
|
||||
const personaName = personaIndex.get(s.persona_id)?.name || s.persona_id || 'default';
|
||||
const dateStr = formatSessionDate(s.updated_at);
|
||||
item.innerHTML = `
|
||||
<div class="s-title">${escapeTitle(s.title || 'Новый чат')}</div>
|
||||
<div class="s-row">
|
||||
<div class="s-title" title="Двойной клик — переименовать">${escapeTitle(s.title || 'Новый чат')}</div>
|
||||
${s.rpg_enabled ? '<span class="s-badge">RPG</span>' : ''}
|
||||
${dateStr ? `<span class="s-date">${dateStr}</span>` : ''}
|
||||
</div>
|
||||
<div class="s-companion">С: ${escapeTitle(personaName)}</div>
|
||||
<div class="s-preview">${escapeTitle(s.last_message_preview || '')}</div>
|
||||
<div class="s-meta">${s.message_count} сообщ.</div>
|
||||
<button class="s-del" type="button">🗑</button>
|
||||
`;
|
||||
item.addEventListener('click', () => switchSession(s.session_id));
|
||||
item.addEventListener('click', (e) => {
|
||||
if (e.target.closest('.s-del') || item.querySelector('.s-title')?.isContentEditable) return;
|
||||
switchSession(s.session_id);
|
||||
});
|
||||
const titleEl = item.querySelector('.s-title');
|
||||
titleEl.addEventListener('dblclick', (e) => {
|
||||
e.stopPropagation();
|
||||
titleEl.contentEditable = 'true';
|
||||
titleEl.focus();
|
||||
});
|
||||
titleEl.addEventListener('blur', async () => {
|
||||
titleEl.contentEditable = 'false';
|
||||
const t = titleEl.textContent.trim();
|
||||
if (t && t !== (s.title || 'Новый чат')) {
|
||||
await fetch(`/sessions/${s.session_id}`, {
|
||||
method: 'PATCH',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ title: t }),
|
||||
});
|
||||
if (s.session_id === sessionId) dom.headerTitle.textContent = t;
|
||||
loadSessions();
|
||||
}
|
||||
});
|
||||
titleEl.addEventListener('keydown', (e) => {
|
||||
if (e.key === 'Enter') { e.preventDefault(); titleEl.blur(); }
|
||||
});
|
||||
item.querySelector('.s-del').addEventListener('click', async (e) => {
|
||||
e.stopPropagation();
|
||||
await fetch(`/sessions/${s.session_id}`, { method: 'DELETE' });
|
||||
if (s.session_id === sessionId) createNewChat();
|
||||
if (s.session_id === sessionId) openNewChatWizard();
|
||||
else loadSessions();
|
||||
});
|
||||
dom.sessionList.appendChild(item);
|
||||
@@ -41,37 +113,335 @@ export async function switchSession(id) {
|
||||
|
||||
export async function loadChatHistory(id) {
|
||||
const sessionRes = await fetch(`/sessions/${id}`);
|
||||
let session = null;
|
||||
if (sessionRes.ok) {
|
||||
const s = await sessionRes.json();
|
||||
dom.headerTitle.textContent = s.title || 'Новый чат';
|
||||
if (s.persona_id) {
|
||||
setCurrentPersona(s.persona_id);
|
||||
highlightPersona(s.persona_id);
|
||||
session = await sessionRes.json();
|
||||
if (session.persona_id) {
|
||||
setCurrentPersona(session.persona_id);
|
||||
highlightPersona(session.persona_id);
|
||||
}
|
||||
applySessionUi(session);
|
||||
}
|
||||
|
||||
const histRes = await fetch(`/chat/history/${id}`);
|
||||
if (!histRes.ok) return;
|
||||
try {
|
||||
const blobRes = await fetch(`/chat/system/${id}`);
|
||||
if (blobRes.ok) {
|
||||
const blob = await blobRes.json();
|
||||
const parts = [];
|
||||
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 */ }
|
||||
|
||||
const messages = await histRes.json();
|
||||
clearMessages();
|
||||
messages.filter(m => m.role !== 'system').forEach(m => {
|
||||
addMessage(
|
||||
m.role === 'user' ? 'user' : 'assistant',
|
||||
m.content,
|
||||
m.image_prompt,
|
||||
m.image_path ? `/static/${m.image_path}` : null,
|
||||
);
|
||||
});
|
||||
await reloadChatFromServer(id);
|
||||
}
|
||||
|
||||
export async function createNewChat() {
|
||||
setSessionId('sess_' + Math.random().toString(36).slice(2, 10));
|
||||
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();
|
||||
dom.headerTitle.textContent = 'Новый чат';
|
||||
highlightPersona(currentPersona);
|
||||
|
||||
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();
|
||||
loadSessions();
|
||||
|
||||
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() {
|
||||
@@ -79,8 +449,8 @@ export async function initSessions() {
|
||||
if (sessionId) {
|
||||
const check = await fetch(`/sessions/${sessionId}`);
|
||||
if (check.ok) await switchSession(sessionId);
|
||||
else createNewChat();
|
||||
else openNewChatWizard();
|
||||
} else {
|
||||
createNewChat();
|
||||
openNewChatWizard();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
export let sessionId = localStorage.getItem('chat_session_id') || null;
|
||||
export let currentPersona = localStorage.getItem('persona_id') || 'default';
|
||||
export let sidebarOpen = true;
|
||||
export let rpgEnabled = false;
|
||||
export function toggleSidebar() { sidebarOpen = !sidebarOpen; return sidebarOpen; }
|
||||
|
||||
export function setSessionId(id) {
|
||||
@@ -13,6 +14,8 @@ export function setCurrentPersona(id) {
|
||||
localStorage.setItem('persona_id', id);
|
||||
}
|
||||
|
||||
export function setRpgEnabled(v) { rpgEnabled = !!v; }
|
||||
|
||||
export const dom = {
|
||||
messagesEl: document.getElementById('messages'),
|
||||
inputEl: document.getElementById('input'),
|
||||
@@ -21,4 +24,9 @@ export const dom = {
|
||||
sessionList: document.getElementById('sessionList'),
|
||||
headerTitle: document.getElementById('headerTitle'),
|
||||
emptyState: document.getElementById('emptyState'),
|
||||
affinityDisplay: document.getElementById('affinityDisplay'),
|
||||
rpgBadge: document.getElementById('rpgBadge'),
|
||||
systemBlob: document.getElementById('systemBlob'),
|
||||
systemBlobContent: document.getElementById('systemBlobContent'),
|
||||
systemBlobToggle: document.getElementById('systemBlobToggle'),
|
||||
};
|
||||
|
||||
@@ -14,3 +14,94 @@ export async function copyToClipboard(text) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
export const GENRE_LABELS = {
|
||||
adventure: 'Приключение',
|
||||
horror: 'Хоррор',
|
||||
romance: 'Романтика',
|
||||
slice_of_life: 'Повседневность',
|
||||
fantasy: 'Фэнтези',
|
||||
sci_fi: 'Sci-Fi',
|
||||
};
|
||||
|
||||
export function initWizard(modalEl, { totalSteps, onStepChange, validateStep }) {
|
||||
let step = 1;
|
||||
const pages = modalEl.querySelectorAll('.wizard-page');
|
||||
const dots = modalEl.querySelectorAll('.wizard-step-dot');
|
||||
const prevBtn = modalEl.querySelector('[id$="Prev"]');
|
||||
const nextBtn = modalEl.querySelector('[id$="Next"]');
|
||||
const saveBtn = modalEl.querySelector('[id$="Save"], [id$="Confirm"], [id$="Create"]');
|
||||
|
||||
function render() {
|
||||
pages.forEach(p => p.classList.toggle('active', Number(p.dataset.step) === step));
|
||||
dots.forEach(d => {
|
||||
const n = Number(d.dataset.step);
|
||||
d.classList.toggle('active', n === step);
|
||||
d.classList.toggle('done', n < step);
|
||||
});
|
||||
prevBtn?.classList.toggle('hidden', step <= 1);
|
||||
nextBtn?.classList.toggle('hidden', step >= totalSteps);
|
||||
saveBtn?.classList.toggle('hidden', step < totalSteps);
|
||||
onStepChange?.(step);
|
||||
}
|
||||
|
||||
function goTo(next) {
|
||||
if (next > step && validateStep && !validateStep(step)) return;
|
||||
step = Math.max(1, Math.min(totalSteps, next));
|
||||
render();
|
||||
}
|
||||
|
||||
prevBtn?.addEventListener('click', () => goTo(step - 1));
|
||||
nextBtn?.addEventListener('click', () => goTo(step + 1));
|
||||
|
||||
render();
|
||||
|
||||
return {
|
||||
reset() { step = 1; render(); },
|
||||
getStep: () => step,
|
||||
goTo,
|
||||
render,
|
||||
};
|
||||
}
|
||||
|
||||
export function bindGenreGrid(gridEl, selectedSet, onChange) {
|
||||
gridEl.addEventListener('click', (e) => {
|
||||
const btn = e.target.closest('.genre-btn');
|
||||
if (!btn) return;
|
||||
const genre = btn.dataset.genre;
|
||||
if (selectedSet.has(genre)) {
|
||||
selectedSet.delete(genre);
|
||||
btn.classList.remove('selected');
|
||||
} else {
|
||||
selectedSet.add(genre);
|
||||
btn.classList.add('selected');
|
||||
}
|
||||
onChange?.();
|
||||
});
|
||||
}
|
||||
|
||||
export function resetGenreGrid(gridEl, selectedSet) {
|
||||
selectedSet.clear();
|
||||
gridEl.querySelectorAll('.genre-btn').forEach(b => b.classList.remove('selected'));
|
||||
}
|
||||
|
||||
export function getRpgSettingsFromDom(prefix = '') {
|
||||
const id = (name) => document.getElementById(prefix + name);
|
||||
return {
|
||||
dice: id('settingDice')?.checked ?? true,
|
||||
narrator: id('settingNarrator')?.checked ?? true,
|
||||
quests: id('settingQuests')?.checked ?? true,
|
||||
affinity: id('settingAffinity')?.checked ?? true,
|
||||
choices: id('settingChoices')?.checked ?? true,
|
||||
};
|
||||
}
|
||||
|
||||
export function formatSessionDate(iso) {
|
||||
if (!iso) return '';
|
||||
const d = new Date(iso.includes('T') ? iso : iso.replace(' ', 'T') + 'Z');
|
||||
if (Number.isNaN(d.getTime())) return '';
|
||||
const now = new Date();
|
||||
const sameDay = d.toDateString() === now.toDateString();
|
||||
if (sameDay) return d.toLocaleTimeString('ru-RU', { hour: '2-digit', minute: '2-digit' });
|
||||
return d.toLocaleDateString('ru-RU', { day: 'numeric', month: 'short' });
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user