from flask import Flask
from flask import request
from pinecone import Pinecone, ServerlessSpec

import os


PINECONE_DB_API = os.environ.get('PINECONE_DB_API')
pc = Pinecone(api_key=PINECONE_DB_API)
index = pc.Index('fury-actor-py')       # NOTE: FOR NOW

app = Flask(__name__)

@app.route("/query", methods=['POST'])
def query():
    """
    {
        embeddings: <embeddings_list>
    }
    """

    embeddings = request.get_json()['embeddings']

    matches = index.query(
        vector = embeddings,
        top_k = 3,
        include_metadata = True
    )

    return matches.to_dict()