App_Function_Libraries/Gradio_UI/Transcript_comparison.py

# Transcript_comparison.py
# Description: Gradio UI tab for comparing transcripts
#
# Imports
import logging

#
# External Imports
import gradio as gr

from App_Function_Libraries.DB.DB_Manager import get_transcripts
from App_Function_Libraries.Gradio_UI.Gradio_Shared import browse_items
from App_Function_Libraries.Utils.Utils import format_transcription


#
# Local Imports

def get_transcript_options(media_id):
    transcripts = get_transcripts(media_id)
    return [f"{t[0]}: {t[1]} ({t[3]})" for t in transcripts]


def update_transcript_options(media_id):
    options = get_transcript_options(media_id)
    return gr.update(choices=options), gr.update(choices=options)

def compare_transcripts(media_id, transcript1_id, transcript2_id):
    try:
        transcripts = get_transcripts(media_id)
        transcript1 = next((t for t in transcripts if t[0] == int(transcript1_id)), None)
        transcript2 = next((t for t in transcripts if t[0] == int(transcript2_id)), None)

        if not transcript1 or not transcript2:
            return "One or both selected transcripts not found."

        comparison = f"Transcript 1 (Model: {transcript1[1]}, Created: {transcript1[3]}):\n\n"
        comparison += format_transcription(transcript1[2])
        comparison += f"\n\nTranscript 2 (Model: {transcript2[1]}, Created: {transcript2[3]}):\n\n"
        comparison += format_transcription(transcript2[2])

        return comparison
    except Exception as e:
        logging.error(f"Error in compare_transcripts: {str(e)}")
        return f"Error comparing transcripts: {str(e)}"


def create_compare_transcripts_tab():
    with gr.TabItem("Compare Transcripts", visible=True):
        gr.Markdown("# Compare Transcripts")

        with gr.Row():
            search_query_input = gr.Textbox(label="Search Query", placeholder="Enter your search query here...")
            search_type_input = gr.Radio(choices=["Title", "URL", "Keyword", "Content"], value="Title", label="Search By")
            search_button = gr.Button("Search")

        with gr.Row():
            media_id_output = gr.Dropdown(label="Select Media Item", choices=[], interactive=True)
            media_mapping = gr.State({})

        media_id_input = gr.Number(label="Media ID", visible=False)
        transcript1_dropdown = gr.Dropdown(label="Transcript 1")
        transcript2_dropdown = gr.Dropdown(label="Transcript 2")
        compare_button = gr.Button("Compare Transcripts")
        comparison_output = gr.Textbox(label="Comparison Result", lines=20)

        def update_media_dropdown(search_query, search_type):
            results = browse_items(search_query, search_type)
            item_options = [f"{item[1]} ({item[2]})" for item in results]
            new_item_mapping = {f"{item[1]} ({item[2]})": item[0] for item in results}
            return gr.update(choices=item_options), new_item_mapping

        search_button.click(
            fn=update_media_dropdown,
            inputs=[search_query_input, search_type_input],
            outputs=[media_id_output, media_mapping]
        )

        def load_selected_media_id(selected_media, media_mapping):
            if selected_media and media_mapping and selected_media in media_mapping:
                media_id = media_mapping[selected_media]
                return media_id
            return None

        media_id_output.change(
            fn=load_selected_media_id,
            inputs=[media_id_output, media_mapping],
            outputs=[media_id_input]
        )

        media_id_input.change(update_transcript_options, inputs=[media_id_input],
                              outputs=[transcript1_dropdown, transcript2_dropdown])
        compare_button.click(compare_transcripts, inputs=[media_id_input, transcript1_dropdown, transcript2_dropdown],
                             outputs=[comparison_output])