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

annotate love/poppy/apis/router.py @ 71:75de5903355c

Giagantic changes that update Dowa library to be more align with stb style array and hashmap. Updated Seobeo to be caching on server side instead of file level caching. Deleted bunch of things I don't really use.

author	June Park <parkjune1995@gmail.com>
date	Sun, 28 Dec 2025 20:34:22 -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 @ 71:75de5903355c