from fasthtml.common import *
from modules.database import load_history

def render_user_msg(text):
    return Div(P(text, style="margin:0; font-weight:bold; white-space: pre-wrap;"), cls="user-bubble")

def render_ai_msg(text, intention, nom_fournisseur, nom_modele, couleur):
    encart_info = Div(
        Span(f"🎯 {intention}", style="margin-right: 10px; color:#aaa; font-size:0.8em;"),
        Span(f"🏢 {nom_fournisseur}", style="margin-right: 10px; color:#aaa; font-size:0.8em;"),
        Span(f"🧠 {nom_modele}", style=f"margin-right: 10px; color:{couleur}; font-weight:bold; font-size:0.8em;"),
        style="border-bottom: 1px solid #333; padding-bottom: 10px; margin-bottom: 15px; display: flex; flex-wrap: wrap; gap: 5px;"
    )
    
    actions = Div(
        Button("📄 PDF", onclick="downloadResponsePDF(this)", cls="msg-action-btn", title="Télécharger en PDF"),
        Button("📘 DOC", onclick="downloadResponseDOC(this)", cls="msg-action-btn", title="Télécharger en Word"),
        Button("📝 TXT", onclick="downloadResponseTXT(this)", cls="msg-action-btn", title="Télécharger en Texte"),
        Button("🗄️ JSON", onclick="downloadResponseJSON(this)", cls="msg-action-btn", title="Télécharger en JSON"),
        Button("💬 WhatsApp", onclick="shareWhatsApp(this)", cls="msg-action-btn", title="Partager sur WhatsApp"),
        Button("📧 Email", onclick="shareEmail(this)", cls="msg-action-btn", title="Envoyer par Email"),
        cls="msg-actions"
    )
    
    return Div(
        encart_info,
        Div(text, cls="markdown-raw", style="display:none;"),
        Div(cls="markdown-rendered", style="line-height: 1.6;"),
        actions,
        cls="ai-bubble", style=f"border-left: 4px solid {couleur};"
    )

def render_sidebar():
    history = load_history()
    sorted_chats = sorted(history.items(), key=lambda x: (not x[1].get('pinned', False), -x[1].get('timestamp', 0)))
    
    items = []
    for chat_id, data in sorted_chats:
        pin_icon = "📌" if data.get('pinned', False) else "📍"
        items.append(
            Div(
                A(Span(data['title'], style="color:inherit;"), href=f"/load_chat/{chat_id}", cls="history-link"),
                Button(pin_icon, cls="history-btn", hx_post=f"/pin/{chat_id}", hx_target="#sidebar-content", hx_swap="outerHTML"),
                Button("✏️", cls="history-btn", hx_post=f"/rename/{chat_id}", hx_prompt="Nouveau nom :", hx_target="#sidebar-content", hx_swap="outerHTML"),
                Button("🗑️", cls="history-btn", hx_post=f"/delete/{chat_id}", hx_target="#sidebar-content", hx_swap="outerHTML"),
                cls="history-item", id=f"chat-item-{chat_id}"
            )
        )
    return Div(*items, id="sidebar-content", cls="sidebar-history", hx_swap_oob="true")