import gradio as gr
from transformers import AutoTokenizer, AutoModelForCausalLM

model_name = "ytu-ce-cosmos/turkish-gpt2"  # veya "redrussianarmy/gpt2-turkish-cased"

tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)

def generate_text(prompt):
    input_ids = tokenizer.encode(prompt, return_tensors="pt")
    outputs = model.generate(
        input_ids,
        max_length=100,
        do_sample=True,
        top_p=0.95,
        temperature=0.8,
        pad_token_id=tokenizer.eos_token_id
    )
    return tokenizer.decode(outputs[0], skip_special_tokens=True)

demo = gr.Interface(
    fn=generate_text,
    inputs=gr.Textbox(label="Türkçe Başlangıç Metni", placeholder="Bir cümle yaz..."),
    outputs=gr.Textbox(label="Üretilen Türkçe Metin"),
    title="Türkçe GPT‑2 Metin Üretici",
    description="Türkçe GPT‑2 ile metin devam ettirme"
)

demo.launch()