from flask import Flask, request, jsonify
from flask_cors import CORS  
import nltk
from nltk.corpus import stopwords
from nltk.tokenize import word_tokenize
from nltk.sentiment.vader import SentimentIntensityAnalyzer

# 指定本地 NLTK 数据目录
nltk.data.path.append('/root/nltk_data')

def preprocess_text(text):
    stop_words = set(stopwords.words('english'))
    word_tokens = word_tokenize(text)
    filtered_text = [word for word in word_tokens if word.lower() not in stop_words and word.isalpha()]
    return ' '.join(filtered_text)

def analyze_sentiment_vader(text):
    analyzer = SentimentIntensityAnalyzer()
    sentiment = analyzer.polarity_scores(text)['compound']
    return sentiment

def emotional_division(sentiment_score_vader):
    if(sentiment_score_vader > 0.25):
        return 1
    elif(sentiment_score_vader < -0.25):
        return 2
    else:
        return 0

app = Flask(__name__)
CORS(app)

@app.route('/Computed', methods=['POST'])
def get_text():
    value = request.json
    value = value['value']
    print(f"Received text: {value}")
    processed_text = preprocess_text(value)
    print("processed_text：", processed_text)
    sentiment_score_vader = analyze_sentiment_vader(processed_text)
    print("sentiment_score_vader：", sentiment_score_vader)
    res = emotional_division(sentiment_score_vader)
    print("res：", res)
    return jsonify(res)

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