# Semantic similarity ile örnek seçimi (mini)
# pip install sentence-transformers
from sentence_transformers import SentenceTransformer, util
import torch
 
model = SentenceTransformer("paraphrase-multilingual-MiniLM-L12-v2")
 
EXAMPLES = [
    ("Ürün geç geldi ama içerik mükemmel", "nötr"),
    ("Hatalı ürün, iade etmek zorunda kaldım", "olumsuz"),
    ("Tam beklediğim gibi, mükemmel kalite", "olumlu"),
    ("Kargo hızlı, ürün güzel", "olumlu"),
    ("Bir kez kullandım, fonksiyonu öyle iyi değil", "olumsuz"),
    ("Fiyat-performans iyi", "olumlu"),
    ("Beklediğim ile gelen aynı değil", "olumsuz"),
    ("Düşündüğümden büyük geldi ama beğendim", "nötr"),
]
 
def select_top_k(query: str, k: int = 3):
    inputs = [e[0] for e in EXAMPLES]
    emb_examples = model.encode(inputs, convert_to_tensor=True)
    emb_query = model.encode(query, convert_to_tensor=True)
    sims = util.cos_sim(emb_query, emb_examples)[0]
    top_idx = torch.topk(sims, k).indices.tolist()
    return [EXAMPLES[i] for i in top_idx]
 
query = "Geç geldi ama kalite iyiydi"
selected = select_top_k(query, 3)
for ex in selected:
    print(f"  {ex}")