# from transformers import pipeline
# import gradio as gr

# # Input language translators (to English)
# input_translators = {
#     "Hindi": pipeline("translation_hi_to_en", model="Helsinki-NLP/opus-mt-hi-en"),
#     "French": pipeline("translation_fr_to_en", model="Helsinki-NLP/opus-mt-fr-en"),
#     "German": pipeline("translation_de_to_en", model="Helsinki-NLP/opus-mt-de-en"),
#     "English": None  # No translation needed
# }

# # Summarization model
# summarizer = pipeline("summarization", model="facebook/bart-large-cnn")

# # Output translators (English → target)
# output_translators = {
#     "None": None,
#     "Hindi": pipeline("translation_en_to_hi", model="Helsinki-NLP/opus-mt-en-hi"),
#     "French": pipeline("translation_en_to_fr", model="Helsinki-NLP/opus-mt-en-fr"),
#     "German": pipeline("translation_en_to_de", model="Helsinki-NLP/opus-mt-en-de"),
# }

# def summarize_multilang(text, input_lang, output_lang):
#     # Step 1: Translate to English (if needed)
#     if input_lang != "English":
#         translator = input_translators[input_lang]
#         text = translator(text)[0]['translation_text']
    
#     # Step 2: Summarize
#     summary = summarizer(text, max_length=130, min_length=30, do_sample=False)[0]['summary_text']
    
#     # Step 3: Translate summary (if needed)
#     if output_lang != "None":
#         summary = output_translators[output_lang](summary)[0]['translation_text']
    
#     return summary

# # Gradio interface
# gr.Interface(
#     fn=summarize_multilang,
#     inputs=[
#         gr.Textbox(lines=10, label="Input Text"),
#         gr.Dropdown(choices=["English", "Hindi", "French", "German"], label="Input Language"),
#         gr.Dropdown(choices=["None", "Hindi", "French", "German"], label="Translate Summary To")
#     ],
#     outputs=gr.Textbox(label="Final Summary"),
#     title="SummarAI",
#     description="Supports input in Hindi, French, German, or English. Summarizes and optionally translates the summary."
# ).launch()



from transformers import pipeline
import gradio as gr

# Input language translators (to English)
input_translators = {
    "Hindi": pipeline("translation_hi_to_en", model="Helsinki-NLP/opus-mt-hi-en"),
    "French": pipeline("translation_fr_to_en", model="Helsinki-NLP/opus-mt-fr-en"),
    "German": pipeline("translation_de_to_en", model="Helsinki-NLP/opus-mt-de-en"),
    "English": None
}

# Summarization model
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")

# Output translators (English → target)
output_translators = {
    "Hindi": pipeline("translation_en_to_hi", model="Helsinki-NLP/opus-mt-en-hi"),
    "French": pipeline("translation_en_to_fr", model="Helsinki-NLP/opus-mt-en-fr"),
    "German": pipeline("translation_en_to_de", model="Helsinki-NLP/opus-mt-en-de"),
}

def summarize_multilang(text, input_lang, toggle_translation, output_lang):
    # Step 1: Translate input to English (if needed)
    if input_lang != "English":
        translator = input_translators[input_lang]
        text = translator(text)[0]['translation_text']
    
    # Step 2: Summarize in English
    summary = summarizer(text, max_length=130, min_length=30, do_sample=False)[0]['summary_text']
    
    # Step 3: Optionally translate the summary
    if toggle_translation and output_lang in output_translators:
        summary = output_translators[output_lang](summary)[0]['translation_text']
    
    return summary

# Gradio Interface with Toggle
with gr.Blocks() as demo:
    gr.Markdown("#  SummarAI ")
    gr.Markdown("Summarize your content in multiple languages. Optionally translate the summary!")

    with gr.Row():
        input_text = gr.Textbox(lines=10, label="Input Text", placeholder="Paste or type your content here...")
    
    with gr.Row():
        input_lang = gr.Dropdown(choices=["English", "Hindi", "French", "German"], label="Input Language", value="English")
        toggle_translation = gr.Checkbox(label="Translate Summary?", value=False)
        output_lang = gr.Dropdown(choices=["Hindi", "French", "German"], label="Translate Summary To", visible=False)

    output_summary = gr.Textbox(label="Final Summary")

    # Toggle visibility logic
    def update_output_lang_visibility(toggle):
        return gr.update(visible=toggle)

    toggle_translation.change(update_output_lang_visibility, inputs=toggle_translation, outputs=output_lang)

    submit_btn = gr.Button("Summarize")
    submit_btn.click(
        fn=summarize_multilang,
        inputs=[input_text, input_lang, toggle_translation, output_lang],
        outputs=output_summary
    )

demo.launch()