import gradio as gr
from google import genai
from youtube_transcript_api import YouTubeTranscriptApi
from youtube_transcript_api._errors import TranscriptsDisabled, NoTranscriptFound, VideoUnavailable


def get_transcript(video_url):
    video_id=video_url.split("v=")[1].split("&")[0]
    transcript=YouTubeTranscriptApi.get_transcript(video_id,languages=['en','tr'])
    raw_text=" ".join([entry['text'] for entry in transcript])
    return raw_text


def fn_sum_text(transkript_text, word_count, model_sel, lang_sel, action_sel,GEMINI_API_KEY, slider_enable):
    client=genai.Client(api_key=GEMINI_API_KEY)
    if slider_enable==True:
        prompt=f"{transkript_text} metni {word_count} sayıda kelimeyle {lang_sel} dilinde {action_sel}"
    else:
        prompt=f"{transkript_text} metni {lang_sel} dilinde {action_sel}"
    
    response=client.models.generate_content(
    model=model_sel,
    contents=[prompt]
    )
    return (response.text)


with gr.Blocks(theme=gr.themes.Soft()) as demo:
    with gr.Row():
        with gr.Column():
            video_url=gr.Textbox(placeholder="Youtube Video URL Giriniz.", label="Youtube Video URL")
            trs_btn=gr.Button('Transkripti Al')
            GEMINI_API_KEY=gr.Textbox(placeholder="GEMINI API KEY",type="password", label="GEMINI API KEY Giriniz")
            word_count=gr.Slider(minimum=50,
            maximum=1000,
            value=200,
            step=10)
            slider_enable = gr.Checkbox(value=True,label="Slider'i kullan")
            model_sel = gr.Dropdown(["gemini-2.0-flash", "gemini-2.0-flash-lite", "gemini-1.5-flash", "gemini-1.5-flash-8b", "gemini-1.5-pro"], value="gemini-2.0-flash", label="Model", info="Model seçiniz")
            lang_sel=gr.Dropdown(
                choices=['Türkçe',
                'İngilizce',
                'Almanca'],
                value='Türkçe',
                label="Dil Seçimi"
            )
            action_sel=gr.Dropdown(
                choices=['Özetle',
                'tam çeviri yap.'],
                value='Özetle',
                label="İşlem"
            )
            sum_btn=gr.Button('Özetle')
        with gr.Column():
            transkript_text=gr.Textbox(label='Transkripti', lines=5)
            sum_text=gr.Textbox(label='Özet', lines=5)
        trs_btn.click(fn=get_transcript,
        inputs=video_url,
        outputs=transkript_text
        )
        sum_btn.click(fn=fn_sum_text,
        inputs=[transkript_text, word_count, model_sel, lang_sel, action_sel, GEMINI_API_KEY, slider_enable],
        outputs=sum_text
        )

        

if __name__ == "__main__":
    demo.launch()