import gradio as gr
from google import genai

client = genai.Client() # uses GOOGLE_API_KEY from env

def generate_text(input_text):
    generated_text = client.models.generate_content(
        model="tunedModels/texthighlighter-c0uqcsiv5s5v",
        contents=input_text
    ).text
    highlighted_text = input_text
    if generated_text is not None:
        generated_text = generated_text.strip()
        highlighted_text = input_text.replace(generated_text, f'<span style="background-color: #fff8c6;">{generated_text}</span>')
    return f"<p>{highlighted_text}</p>"

with gr.Blocks() as demo:
    gr.Markdown("## Text Highligher App")
    input_text = gr.Textbox(label="Enter your paragraph", lines=5)
    output_text = gr.HTML()
    submit = gr.Button("Highlight")
    submit.click(generate_text, inputs=input_text, outputs=output_text)

demo.launch()