/home/mrjunejune/zenbu: love/poppy/apis/router.py annotate

annotate love/poppy/apis/router.py @ 149:f41ac17926d2

[Config] Added ctags scripts and actual tags.

author	June Park <parkjune1995@gmail.com>
date	Sat, 10 Jan 2026 07:07:10 -0800
parents	cf9caa4abc3e
children

rev	line source
38 cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	1 from typing import Optional
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	2 from uuid import UUID
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	3 from fastapi import (
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	4 APIRouter,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	5 Depends,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	6 HTTPException,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	7 status,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	8 WebSocket,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	9 WebSocketDisconnect,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	10 Query,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	11 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	12 from sqlalchemy.orm import selectinload
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	13 from sqlmodel import select, desc, exists
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	14 import json
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	15 import asyncio
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	16 import os
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	17
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	18 from xai_sdk import Client
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	19 from xai_sdk.chat import user, system, assistant
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	20 from xai_sdk.tools import mcp, web_search, code_execution
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	21
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	22 from .schemas import MessageOut, ChatOut, ChatWithMessages, Chats
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	23 from db.models import Chat, Message, MessageAsset
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	24 from utils.database import get_session
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	25 # from utils.redis import append_message, get_all_messages_from_redis
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	26 from utils.logger import logger
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	27
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	28 import sys
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	29
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	30
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	31 GROK_API_KEY = os.getenv("XAI_API_KEY", "NO-KEY")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	32 TOOLS = [
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	33 web_search(),
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	34 code_execution(),
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	35 # mcp(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	36 # server_url="https://mcp.babocoder.com/mcp",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	37 # server_label="powerpoint-generator",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	38 # server_description="This will create powerpoitn slides and files for you."
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	39 # )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	40 ]
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	41 # xAI SDK Client
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	42 client = Client(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	43 api_key=GROK_API_KEY,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	44 timeout=3600, # Take this out when you are deploying to your local server.
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	45 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	46
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	47 router = APIRouter(tags=["chat"])
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	48
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	49 # TODO: Make this into something more useful
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	50 SYSTEM_PROMPT = """
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	51 You are a dog lover and everytime someone mentioned about their dog. You should be the most excited person alive.
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	52 """
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	53
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	54 TITLE_GENERATION_PROMPT = """Based on the following user message, generate a short, descriptive title (max 6 words) for this chat conversation.
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	55 Only return the title text, nothing else.
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	56
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	57 User message: {user_message}"""
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	58
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	59
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	60 async def generate_chat_title(user_message: str) -> str:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	61 try:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	62 title_client = Client(api_key=GROK_API_KEY, timeout=30)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	63 title_chat = title_client.chat.create(model="grok-4-fast")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	64 title_chat.append(system(TITLE_GENERATION_PROMPT.format(user_message=user_message)))
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	65
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	66 response = title_chat.sample()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	67 title = response.content.strip()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	68
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	69 logger.info("Title: \n")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	70 logger.info(title)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	71
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	72 # Ensure title isn't too long
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	73 if len(title) > 100:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	74 title = title[:97] + "..."
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	75
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	76 return title if title else "New Chat"
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	77 except Exception as e:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	78 logger.error(f"Failed to generate title: {e}")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	79 return "New Chat"
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	80
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	81 async def generate_image(user_message: str) -> str:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	82 try:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	83 image_client = Client(api_key=GROK_API_KEY, timeout=300)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	84 response = image_client.image.sample(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	85 model="grok-2-image",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	86 prompt=user_message,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	87 image_format="url"
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	88 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	89 logger.info("Image URL: \n")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	90 logger.info(response.url)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	91
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	92 return response.url
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	93 except Exception as e:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	94 logger.error(f"Failed to generate title: {e}")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	95 return "failed image url"
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	96
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	97
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	98 async def should_update_title(chat: Chat, session) -> bool:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	99 stmt = select(Message).where(Message.chat_id == chat.id)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	100 result = await session.execute(stmt)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	101 message_count = len(result.scalars().all())
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	102 logger.info(f"message_count: {message_count}")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	103
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	104 # Update title if we have exactly 2 messages and title is still default
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	105 return message_count == 2 and chat.title == "New Chat"
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	106
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	107
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	108 # --- REST ---
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	109 @router.get("/chats", response_model=Chats)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	110 async def get_chats(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	111 session=Depends(get_session),
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	112 limit: int = Query(50, ge=1, le=100),
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	113 cursor: Optional[str] = Query(None, description="Chat ID to paginate from")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	114 ):
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	115 has_messages = exists(Message).where(Message.chat_id == Chat.id)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	116
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	117 stmt = select(Chat).where(has_messages).order_by(desc(Chat.created_at))
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	118
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	119 # If cursor provided, filter to chats older than cursor chat
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	120 if cursor:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	121 try:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	122 cursor_chat_id = UUID(cursor)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	123 cursor_chat = await session.get(Chat, cursor_chat_id)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	124 if cursor_chat:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	125 stmt = stmt.where(Chat.created_at < cursor_chat.created_at)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	126 except (ValueError, TypeError):
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	127 pass # Invalid cursor, ignore it
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	128
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	129 stmt = stmt.limit(limit)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	130
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	131 result = await session.execute(stmt)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	132 recent_chats = result.scalars().all()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	133
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	134 return Chats(chats=recent_chats)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	135
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	136 @router.post("/chats", response_model=ChatOut, status_code=status.HTTP_201_CREATED)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	137 async def create_chat(session=Depends(get_session)):
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	138 chat = Chat()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	139 session.add(chat)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	140 await session.commit()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	141 await session.refresh(chat)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	142 return chat
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	143
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	144
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	145 @router.get("/chats/{chat_id_str}/messages", response_model=ChatWithMessages)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	146 async def get_chat_messages(chat_id_str: str, session=Depends(get_session)):
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	147 chat_id = UUID(chat_id_str)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	148 chat = await session.get(Chat, chat_id)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	149 if not chat:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	150 raise HTTPException(404, "Chat not found")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	151
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	152 # try:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	153 # all_msgs = await get_all_messages_from_redis()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	154 # chat_msgs = [m for m in all_msgs if m["chatId"] == chat_id_str]
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	155 # if chat_msgs:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	156 # messages = [
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	157 # MessageOut(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	158 # id=UUID(m.get("id", "00000000-0000-0000-0000-000000000000")), # or store id in Redis!
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	159 # role=m["role"],
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	160 # content=m["content"],
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	161 # created_at=None,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	162 # image_url=m.get("image_url"), # ← make sure Redis also stores this
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	163 # )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	164 # for m in chat_msgs
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	165 # ]
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	166 # return ChatWithMessages(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	167 # id=chat.id,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	168 # title=chat.title,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	169 # created_at=chat.created_at,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	170 # messages=messages,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	171 # )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	172 # except Exception as e:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	173 # pass
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	174
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	175 stmt = (
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	176 select(Message)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	177 .options(selectinload(Message.assets))
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	178 .where(Message.chat_id == chat_id)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	179 .order_by(Message.created_at)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	180 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	181 result = await session.execute(stmt)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	182 messages = result.scalars().all()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	183 return ChatWithMessages(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	184 id=chat.id,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	185 title=chat.title,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	186 created_at=chat.created_at,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	187 messages=[MessageOut.from_orm_with_assets(m) for m in messages],
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	188 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	189
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	190
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	191 @router.websocket("/chats/{chat_id_str}/ws")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	192 async def chat_stream(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	193 websocket: WebSocket,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	194 chat_id_str: str,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	195 session=Depends(get_session),
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	196 use_grok: bool = Query(True), # TODO: use url for this, but for now this is fine.
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	197 ):
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	198 chat_id = UUID(chat_id_str)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	199
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	200 await websocket.accept()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	201
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	202 chat = None
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	203 history = []
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	204
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	205 # Load chat from DB
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	206 db_chat = await session.get(Chat, chat_id)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	207 if not db_chat:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	208 await websocket.close(code=1008)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	209 return
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	210
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	211 # Load Message
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	212 if use_grok:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	213 try:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	214 all_msgs = await get_all_messages_from_redis()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	215 chat_msgs = [m for m in all_msgs if m["chatId"] == chat_id_str]
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	216 if chat_msgs:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	217 history = [
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	218 {"role": m["role"], "content": m["content"]} for m in chat_msgs
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	219 ]
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	220 else:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	221 stmt = (
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	222 select(Message)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	223 .where(Message.chat_id == chat_id)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	224 .order_by(Message.created_at)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	225 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	226 result = await session.execute(stmt)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	227 db_msgs = result.scalars().all()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	228 history = [{"role": m.role, "content": m.content} for m in db_msgs]
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	229 except Exception as e:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	230 logger.error("History load failed:", e)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	231 history = []
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	232
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	233 chat = client.chat.create(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	234 model="grok-4-fast",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	235 tools=TOOLS
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	236 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	237 chat.append(system(SYSTEM_PROMPT))
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	238
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	239 # order does not matter so should be fine...
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	240 for msg in history:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	241 if msg["role"] == "user":
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	242 chat.append(user(msg["content"]))
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	243 elif msg["role"] == "assistant":
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	244 chat.append(assistant(msg["content"]))
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	245
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	246 logger.info(f"This should only be called once: {history}")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	247 # Listen to websocket
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	248 try:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	249 while True:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	250 full_response = ""
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	251 assistant_msg = None
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	252
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	253 try:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	254 data = await websocket.receive_text()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	255 # TODO: Create seralizer for this.
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	256 user_msg = json.loads(data)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	257 if user_msg.get("role") != "user":
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	258 await websocket.close(code=1008)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	259 return
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	260 user_content = user_msg["content"]
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	261
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	262 msg = Message(chat_id=chat_id, role="user", content=user_content)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	263 session.add(msg)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	264 await session.commit()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	265 await session.refresh(msg)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	266 # await append_message(chat_id_str, "user", user_content)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	267
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	268 # Actual API calls
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	269 if use_grok:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	270
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	271 # Can use clarifying agent model for this, but I feel like that would be over kill..
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	272 image_triggers = [
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	273 "/img ",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	274 "generate image ",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	275 "create image ",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	276 "draw ",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	277 "make an image of ",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	278 "show me a picture of ",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	279 "image: ",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	280 "img: ",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	281 "🖼️",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	282 "picture of ",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	283 "generate a photo of ",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	284 ]
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	285 is_image_call = False
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	286 image_url = ""
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	287 for trigger in image_triggers:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	288 if trigger in user_content:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	289 is_image_call = True
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	290 break
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	291
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	292 if is_image_call:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	293 image_url = await generate_image(user_content)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	294 logger.info(f"Image got generated and we are sending assistant prompt?")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	295 chat.append(assistant(f"Your image is here. Please view {image_url}!"))
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	296 await websocket.send_json(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	297 {
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	298 "chatId": chat_id_str,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	299 "url": f"{image_url}",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	300 "action": "image",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	301 }
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	302 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	303 else:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	304 logger.info(f"User content sent: {user_content}")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	305 chat.append(user(user_content))
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	306
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	307 # AI response....
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	308 # Maybe stream the token into redis?, but probably over kill.
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	309 is_thinking = True
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	310 if use_grok:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	311 for response, chunk in chat.stream():
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	312 for tool_call in chunk.tool_calls:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	313 logger.info(f"\nCalling tool: {tool_call.function.name} with arguments: {tool_call.function.arguments}")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	314 # if response.usage.reasoning_tokens and is_thinking:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	315 # logger.info(f"\rThinking... ({response.usage.reasoning_tokens} tokens)")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	316 if chunk.content and is_thinking:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	317 logger.info("\n\nAnalysis Results:")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	318 is_thinking = False
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	319 token = chunk.content
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	320 if not token:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	321 continue
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	322 full_response = response.content
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	323 await websocket.send_json(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	324 {
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	325 "chatId": chat_id_str,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	326 "content": token,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	327 "action": "append",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	328 }
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	329 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	330 await asyncio.sleep(0.015)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	331 logger.info("\n\nCitations:")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	332 logger.info(response.citations)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	333 logger.info("\n\nUsage:")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	334 logger.info(response.usage)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	335 logger.info(response.server_side_tool_usage)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	336 logger.info("\n\nServer Side Tool Calls:")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	337 logger.info(response.tool_calls)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	338 else:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	339 mock_reply = f"[Mock] Echo: {user_content}"
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	340 for token in mock_reply.split():
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	341 token += " "
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	342 full_response += token
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	343 await websocket.send_json(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	344 {
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	345 "chatId": chat_id_str,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	346 "content": token,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	347 "action": "append",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	348 }
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	349 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	350 await asyncio.sleep(0.2)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	351
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	352 if use_grok and full_response:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	353 chat.append(assistant(full_response))
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	354
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	355 # Async update redis
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	356 # task = asyncio.create_task(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	357 # append_message(chat_id_str, "assistant", full_response)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	358 # )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	359 # asyncio.shield(task)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	360
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	361 assistant_msg = Message(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	362 chat_id=chat_id, role="assistant", content=full_response
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	363 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	364 session.add(assistant_msg)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	365 await session.commit()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	366 await session.refresh(assistant_msg)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	367
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	368 # Now create assets linked to the message
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	369 if is_image_call and image_url:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	370 asset = MessageAsset(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	371 message_id=assistant_msg.id,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	372 asset_type="image",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	373 url=image_url,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	374 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	375 session.add(asset)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	376 await session.commit()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	377 await session.refresh(assistant_msg)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	378
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	379 # Getting titles
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	380 # Generate title after first exchange
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	381 if use_grok and await should_update_title(db_chat, session):
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	382 new_title = await generate_chat_title(user_content)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	383 db_chat.title = new_title
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	384 session.add(db_chat)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	385 await session.commit()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	386 logger.info(f"Updated chat title to: {new_title}")
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	387 await websocket.send_json(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	388 {
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	389 "chatId": chat_id_str,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	390 "title": new_title,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	391 "action": "title_updated"
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	392 }
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	393 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	394 else:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	395 db_chat.title = "Automatic title"
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	396 session.add(db_chat)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	397 await session.commit()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	398 await websocket.send_json(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	399 {
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	400 "chatId": chat_id_str,
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	401 "title": "Automatic title",
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	402 "action": "title_updated"
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	403 }
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	404 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	405
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	406 await websocket.send_json(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	407 {"chatId": chat_id_str, "content": "", "action": "done"}
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	408 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	409 except WebSocketDisconnect:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	410 break
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	411 except json.JSONDecodeError:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	412 await websocket.send_json({"error": "Invalid JSON"})
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	413 except Exception as e:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	414 exc_type, exc_value, exc_traceback = sys.exc_info()
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	415 line_number = exc_traceback.tb_lineno
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	416 logger.error(
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	417 f"An error occurred on line {line_number}: {type(e).__name__} - {e}"
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	418 )
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	419 await websocket.send_json({"error": str(e)})
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	420 except Exception as e:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	421 logger.error("Error: ", e)
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	422 finally:
cf9caa4abc3e [Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations. MrJuneJune <me@mrjunejune.com> parents: diff changeset	423 pass

Mercurial

annotate love/poppy/apis/router.py @ 149:f41ac17926d2