from typing import List, Dict, Union, Any
from uuid import uuid4

from ask_candid.retrieval.sources import (
    candid_blog,
    candid_help,
    candid_learning,
    issuelab,
    youtube
)


def filter_messages(messages, k=10):
    # TODO summarize messages instead
    return messages[-k:]


def html_format_doc(doc: Dict[str, Any], source: str, show_chunks=False) -> str:
    height_px = 200
    html = ""

    if source == "news":
        # html = news.article_card_html(doc, height_px, show_chunks)
        pass
    elif source == "transactions":
        # html = cds.transaction_card_html(doc, height_px, show_chunks)
        pass
    elif source == "organizations":
        # html = up_orgs.organization_card_html(doc, 400, show_chunks)
        pass
    elif source == "issuelab":
        html = issuelab.issuelab_card_html(doc, height_px, show_chunks)
    elif source == "youtube":
        html = youtube.build_card_html(doc, 400, show_chunks)
    elif source == "candid_blog":
        html = candid_blog.build_card_html(doc, height_px, show_chunks)
    elif source == "candid_learning":
        html = candid_learning.build_card_html(doc, height_px, show_chunks)
    elif source == "candid_help":
        html = candid_help.build_card_html(doc, height_px, show_chunks)
    return html


def html_format_docs_chat(docs):
    """
    Formats Candid sources into a line of buttons
    """
    html = ""
    if docs:
        docs_html = []
        for doc in docs:
            s_name = doc.metadata.get("source", "Source")
            s_url = doc.metadata.get("url", "URL")

            s_html = (
                "<span class='source-item'>"
                f"<a href={s_url} target='_blank' rel='noreferrer' class='ssearch-source'>"
                f"{doc.metadata['title']} ({s_name})</a></span>"
            )

            docs_html.append(s_html)

        html = f"<h2>Candid Resources</h2><div id='ssearch-sources'>{'<br>'.join(docs_html)}</div>"
    return html


def format_chat_response(chatbot: List[Any]) -> List[Any]:
    """We have sources appended as one more tuple. Here we concatinate HTML of sources
        with the AI response
    Returns:
        _type_: updated chatbot message as HTML
    """
    if chatbot:
        sources = chatbot[-1][1]
        chatbot.pop(-1)
        chatbot[-1][1] = chatbot[-1][1] + sources
    return chatbot


def format_chat_ag_response(chatbot: List[Any]) -> List[Any]:
    """If we called retriever, we appended sources as as one more message. Here we concatinate HTML of sources
        with the AI response
    Returns:
        _type_: updated chatbot message as HTML
    """
    sources = ""
    if chatbot:
        title = chatbot[-1]["metadata"].get("title", None)
        if title == "Sources HTML":
            sources = chatbot[-1]["content"]
            chatbot.pop(-1)
            chatbot[-1]["content"] = chatbot[-1]["content"] + sources
    return chatbot


def valid_inputs(*args) -> bool:
    return any(a is not None or (isinstance(a, str) and a.strip() != '') for a in args)


def get_session_id(thread_id: Union[str, None]) -> str:
    if not thread_id:
        thread_id = uuid4().hex
    return thread_id