import gradio as gr
from summarizer.summarizer import Summarizer

summarizer = Summarizer()

def summarize(text, target_lang="en"):
    if not text:
        return "Error: Please provide text for summarization."
    try:
        summary = summarizer.summarize_text(text, target_lang)
        return summary
    except Exception as e:
        return f"Summarization failed: {str(e)}"

iface = gr.Interface(
    fn=summarize,
    inputs=[
        gr.Textbox(lines=10, label="Enter Clinical Case Text"),
        gr.Textbox(label="Target Language Code (default 'en')"),
    ],
    outputs="text",
    title="CareCompanion Summarizer",
    description="Summarizes clinical case descriptions into diagnosis, comorbidities, and treatment plan. (Add target language code like 'en', 'hi', 'gu', etc.)",
)

iface.launch()