from sqlalchemy import create_engine, text
from sentence_transformers import SentenceTransformer
import os

DATABASE_URL = os.getenv("DATABASE_URL")
engine = create_engine(DATABASE_URL)

model = SentenceTransformer("all-MiniLM-L6-v2")

name = "Range"
embedding = model.encode(name).tolist()

with engine.connect() as conn:
    conn.execute(text("""
        INSERT INTO attributes (name, embedding)
        VALUES (:name, :embedding)
    """), {"name": name, "embedding": embedding})
    conn.commit() 

print("Inserted successfully")