import os
# Default /.cache/huggingface ci vengono scaricati i modelli
os.environ["HF_HOME"] = "/tmp/huggingface"

from flask import Flask, jsonify, request
from flask_cors import CORS

from model.sentiment import Sentiment
from urllib.parse import unquote

app = Flask(__name__)
CORS(app)

# Endpoint '/'
@app.route('/')
def home():
    return jsonify({"message": "API is running!"})

# Endpoint '/analisis'
@app.route('/analisis', methods=['GET'])
def analisis():
    prompt = request.args.get('prompt', '')
    prompt = unquote(prompt)

    if not prompt:
        return jsonify({"error": "No prompt provided"}), 400
    
    model = Sentiment(prompt)
    return jsonify({"message": model.result}), 200

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=7860)