# Trash.py
# Gradio UI for managing trashed items in the database
#
# Imports
from typing import Tuple, List

import gradio as gr
#
# Local Imports
from App_Function_Libraries.DB.DB_Manager import (
    get_trashed_items, user_delete_item, empty_trash,
    get_transcripts, fetch_item_details,
    search_media_database, mark_as_trash,
)


#
############################################################################################################
#
# Functions:


def list_trash():
    items = get_trashed_items()
    return "\n".join(
        [f"ID: {item['id']}, Title: {item['title']}, Trashed on: {item['trash_date']}" for item in items])


def delete_item(media_id, force):
    return user_delete_item(media_id, force)


def empty_trash_ui(days):
    deleted, remaining = empty_trash(days)
    return f"Deleted {deleted} items. {remaining} items remain in trash."


def get_media_transcripts(media_id):
    transcripts = get_transcripts(media_id)
    return "\n\n".join([f"Transcript ID: {t[0]}\nModel: {t[1]}\nCreated: {t[3]}\n{t[2][:200]}..." for t in transcripts])


def get_media_summaries(media_id):
    _, summary, _ = fetch_item_details(media_id)
    return summary if summary else "No summary available."


def get_media_prompts(media_id):
    prompt, _, _ = fetch_item_details(media_id)
    return prompt if prompt else "No prompt available."


def search_and_mark_trash(search_query: str) -> Tuple[List[Tuple[int, str, str]], str]:
    try:
        results = search_media_database(search_query)
        if not results:
            return [], "No items found matching the search query."
        return results, "Search completed successfully."
    except Exception as e:
        return [], f"Error during search: {str(e)}"


def mark_item_as_trash(media_id: int) -> str:
    try:
        mark_as_trash(media_id)
        return f"Item with ID {media_id} has been marked as trash."
    except Exception as e:
        return f"Error marking item as trash: {str(e)}"


def create_search_and_mark_trash_tab():
    with gr.TabItem("Search and Mark as Trash"):
        gr.Markdown("# Search for Items and Mark as Trash")

        search_input = gr.Textbox(label="Search Query")
        search_button = gr.Button("Search")
        search_results = gr.Dropdown(label="Search Results", choices=[], interactive=True)
        search_status = gr.Textbox(label="Search Status")

        mark_trash_button = gr.Button("Mark Selected Item as Trash")
        mark_trash_status = gr.Textbox(label="Mark as Trash Status")

        def update_search_results(query):
            results, status = search_and_mark_trash(query)
            choices = [f"{id}: {title} ({url})" for id, title, url in results]
            return choices, status

        search_button.click(
            update_search_results,
            inputs=[search_input],
            outputs=[search_results, search_status]
        )

        def mark_selected_as_trash(selected_item):
            if selected_item:
                media_id = int(selected_item.split(":")[0])
                return mark_item_as_trash(media_id)
            return "No item selected."

        mark_trash_button.click(
            mark_selected_as_trash,
            inputs=[search_results],
            outputs=[mark_trash_status]
        )


def create_view_trash_tab():
    with gr.TabItem("View Trash"):
        view_button = gr.Button("View Trash")
        trash_list = gr.Textbox(label="Trashed Items")
        view_button.click(list_trash, inputs=[], outputs=trash_list)


def create_delete_trash_tab():
    with gr.TabItem("Delete DB Item"):
        gr.Markdown("# Delete Items from Databases")

        media_id_input = gr.Number(label="Media ID")
        media_force_checkbox = gr.Checkbox(label="Force Delete")
        media_delete_button = gr.Button("Delete Media")
        media_delete_output = gr.Textbox(label="Delete Result")

        media_delete_button.click(
            delete_item,
            inputs=[media_id_input, media_force_checkbox],
            outputs=media_delete_output
        )


def create_empty_trash_tab():
    with gr.TabItem("Empty Trash"):
        days_input = gr.Slider(minimum=15, maximum=90, step=5, label="Delete items older than (days)")
        empty_button = gr.Button("Empty Trash")
        empty_output = gr.Textbox(label="Result")
        empty_button.click(empty_trash_ui, inputs=[days_input], outputs=empty_output)

#
# End of File
############################################################################################################