import gradio as gr
from google import genai
import os

client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])

def diacritize(text):

    prompt = f"""
أضف التشكيل الكامل للنص العربي التالي مع مراعاة المعنى:

{text}

أعد النص فقط بعد التشكيل بدون أي شرح.
"""

    response = client.models.generate_content(
        model="gemini-2.5-flash",
        contents=prompt
    )

    return response.text


demo = gr.Interface(
    fn=diacritize,
    inputs=gr.Textbox(lines=6, label="Input Text"),
    outputs=gr.Textbox(label="Diacritized Text"),
    title="Arabic Diacritization API"
)

demo.launch()