import torch
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
import gradio as gr

MODEL_NAME = "cagrigungor/adrest0"

# -------------------------------
# MODEL ve TOKENIZER YÜKLEME
# -------------------------------
device = "cuda" if torch.cuda.is_available() else "cpu"

tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)

#  TÜRKÇE KARAKTER PROBLEMİNİ ÇÖZEN SATIR:
try:
    tokenizer.backend_tokenizer.normalizer = None
except:
    pass  # bazı tokenizer sürümleri için zorunlu değil

model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME).to(device)


# -------------------------------
# ADRES DÜZELTME FONKSİYONU
# -------------------------------
def duzelt(adres: str) -> str:
    encoded = tokenizer(
        adres,
        return_tensors="pt",
        truncation=True,
        max_length=64
    ).to(device)

    with torch.no_grad():
        output = model.generate(
            **encoded,
            max_length=64,
            num_beams=4,
            early_stopping=True
        )

    text = tokenizer.decode(output[0], skip_special_tokens=True)
    return text


# -------------------------------
# GRADIO ARAYÜZ
# -------------------------------
with gr.Blocks(title="Türkçe Adres Düzeltme - adrest5") as demo:

    gr.Markdown("""
    # 🏡 Türkçe Adres Düzeltme  
    Bu uygulama, bozuk veya eksik yazılmış **Türkçe adresleri** otomatik olarak
    standart ve okunabilir biçime dönüştürür.
    **Model:** `cagrigungor/adrest5`  
    """)

    input_text = gr.Textbox(
        label="Bozuk Adres Girin",
        placeholder="ör: atatrk cad cafega mahs num 18 kadikoy",
        lines=2
    )

    output_text = gr.Textbox(
        label="Düzeltilmiş Adres",
        lines=2
    )

    btn = gr.Button("Düzelt")
    btn.click(duzelt, inputs=input_text, outputs=output_text)

    gr.Markdown("""
    ---  
    **Geliştirici:** @cagrigungor  
    """)


demo.launch()