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

import gradio as gr

try:
    from retrieval import (
        candid_blog,
        candid_help,
        candid_learning,
        # cds,
        issuelab,
        # news,
        # up_orgs,
        youtube
    )
except ImportError:
    from .retrieval import (
        candid_blog,
        candid_help,
        candid_learning,
        # cds,
        issuelab,
        # news,
        # up_orgs,
        youtube
    )


# TODO summarize messages instead
def filter_messages(messages, k=10):
    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 = f"""<a href={s_url} target='_blank' rel='noreferrer' class='ssearch-source'> \
<button class='ssearch-source-btn'>{s_name}</button></a>"""
            docs_html.append(s_html)
        docs_html_insert = "".join(s for s in docs_html)
        html = f"""<div id='ssearch-sources'>{docs_html_insert}</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 gr.HTML(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 gr.HTML(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: gr.components.Component) -> str:
    if not thread_id:
        thread_id = uuid4().hex
    return thread_id