from insert_into_milvus_db import (TextVectorizer, get_secrets, 
                                   create_schema, upsert_db)
from flask import Flask, Response
from flask_cors import cross_origin, CORS
import logging


app = Flask(__name__)
CORS(app)
logging.info('Initiated')


vectorizer = TextVectorizer(checkpoint='all-mpnet-base-v2')
uri, token, collection_name = get_secrets()
collection = create_schema(uri, token, collection_name)


@app.route("/")
@cross_origin()
def update_news_in_db():
    status_json = "{'status':'success'}"
    status_code = 200
    try:
        upsert_db(vectorizer, collection)
    except:
        status_json = "{'status':'failure'}"
        status_code = 500
    return Response(status_json, status=status_code, mimetype='application/json')


if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860, timeout=120, workers=1, threads=1)