added mesuurments
This commit is contained in:
@@ -10,9 +10,11 @@ from sqlalchemy import select
|
||||
from app.config import get_settings
|
||||
from app.db.base import SessionLocal
|
||||
from app.db.models import User
|
||||
from app.homelab.comfyui import ComfyUIClient
|
||||
from app.character.service import CharacterService
|
||||
from app.homelab.comfyui import ComfyUIClient, random_rofl_pose
|
||||
from app.homelab.context import resolve_timezone
|
||||
from app.homelab.digest import build_morning_digest
|
||||
from app.homelab.image_gen import generate_image
|
||||
from app.homelab.monitoring import check_netdata_alerts
|
||||
from app.homelab_scoped.notices import post_chat_notice
|
||||
from app.homelab.state import get_state, set_state
|
||||
@@ -140,10 +142,15 @@ async def _tick_rofl() -> None:
|
||||
if not await _comfyui_reachable(client.base_url):
|
||||
continue
|
||||
|
||||
prompt = client.random_rofl_prompt()
|
||||
card = CharacterService(db, user.id).get_card().get("data", {})
|
||||
appearance = (card.get("appearance_tags") or "").strip()
|
||||
if not appearance:
|
||||
continue
|
||||
|
||||
pose = random_rofl_pose()
|
||||
try:
|
||||
result = await asyncio.wait_for(
|
||||
client.generate_image(prompt),
|
||||
generate_image(db, user_id=user.id, draw_self=True, scene_description=pose),
|
||||
timeout=settings.comfyui_timeout_sec + 15,
|
||||
)
|
||||
except (asyncio.TimeoutError, httpx.TimeoutException, httpx.ConnectError) as exc:
|
||||
@@ -155,7 +162,7 @@ async def _tick_rofl() -> None:
|
||||
|
||||
url = result.get("url", "")
|
||||
post_chat_notice(
|
||||
f"🎨 **Рофл дня**\n\n\n\n_{prompt}_",
|
||||
f"🎨 **Рофл дня**\n\n\n\n_{pose}_",
|
||||
user.id,
|
||||
)
|
||||
set_state(db, count_key, str(count + 1))
|
||||
|
||||
Reference in New Issue
Block a user