import gradio as gr
from transformers import pipeline
import torch

# 🔥 Modeli yükle (örnek olarak gpt2)
generator = pipeline("text-generation", model="gpt2")

def generate_text(prompt, temperature, top_k):
    output = generator(
        prompt,
        max_length=200,
        temperature=temperature,
        top_k=int(top_k),
        do_sample=True,
        pad_token_id=50256
    )[0]["generated_text"]
    return output

with gr.Blocks() as app:
    gr.Markdown("# 🧠 Hugging Face Text Generator (BrtAI Edition)")

    with gr.Row():
        prompt = gr.Textbox(label="Input", placeholder="Bir şey yaz...", lines=3)
    with gr.Row():
        temperature = gr.Slider(0.1, 2.0, value=1.0, label="Temperature")
        top_k = gr.Slider(10, 100, value=50, step=1, label="Top-K")
    with gr.Row():
        output = gr.Textbox(label="Output", lines=6)
    with gr.Row():
        btn = gr.Button("Generate")

    btn.click(generate_text, inputs=[prompt, temperature, top_k], outputs=output)

# 🚀 API açık!
app.launch()