from flask import Flask, jsonify, request
from dotenv import load_dotenv
from services.embedding_service import Embedding


load_dotenv()
app = Flask(__name__)

# @app.route("/")
# def hello():
#     return jsonify({"message":"Hello, World!"})

@app.get("/")
def hello():
    return jsonify({"message":"Hello, World!"})


@app.post("/get_embedding")
def get_embedding():
    # Get the query text from request JSON
    data = request.get_json()
    query = data.get("query", "")

    if not query:
        return jsonify({"error": "Query text is required"}), 400

    # Call the embedding function
    vector = Embedding.call(query, is_query=True)
    
    return jsonify({"query": query, "number of embedding": len(vector), "one embedding":len(vector[0])})

print(__name__)

if __name__=="__main__":
    app.run(debug=True, port=8000)