# -*- coding:utf-8 -*-
# @Time: 2023/8/22 14:44
# @Author:ZYP
# @File：word2vec_api.py
# @mail：zypsunshine1@gmail.com
# @Software: PyCharm
import json

import numpy as np
from gensim.models.keyedvectors import KeyedVectors
import time
from flask import Flask, request

app_word2vec = Flask(__name__)

word2vec_path = "/home/zc-nlp-zyp/work_file/ssd_data/public_data/word2vec_model/word2vec.vector"
model_word2vec = KeyedVectors.load_word2vec_format(word2vec_path, binary=True)


@app_word2vec.route('/word2vec', methods=['POST'])
def get_word2vec():
    word_dict = json.loads(request.data.decode())
    try:
        vec = model_word2vec.get_vector(word_dict["word"])
        str_vec = ','.join([str(i) for i in vec])
        # vec1 = np.array([float(j) for j in str_vec.split(',')], dtype=np.float64)
        vec_dict = {'vec': str_vec}
        return json.dumps(vec_dict)
    except:
        return 'error_word2vec'


# if __name__ == '__main__':
#     app.run(host='0.0.0.0', port=50001, debug=False)