import gradio as gr
from transformers import pipeline

# יצירת צינור AI (text-generation או text2text)
generator = pipeline("text2text-generation", model="google/flan-t5-small")

def ai_reply(prompt):
    # שולח את הטקסט למודל ומחזיר פלט
    result = generator(prompt, max_length=100)[0]['generated_text']
    return result

iface = gr.Interface(
    fn=ai_reply,
    inputs=gr.Textbox(label="כתוב משהו"),
    outputs=gr.Textbox(label="תשובת AI")
)

iface.launch()