from flask import jsonify
from main import *
import torch

def analyze_sentiment(text):
    if sentiment_model is None:
        return {"error": "Sentiment model not initialized."}

    features = [ord(c) for c in text[:10]]
    while len(features) < 10:
        features.append(0)
    features_tensor = torch.tensor(features, dtype=torch.float32).unsqueeze(0).to(device)

    with torch.no_grad():
        output = sentiment_model(features_tensor)
        sentiment_idx = torch.argmax(output, dim=1).item()
        sentiment_label = "positive" if sentiment_idx == 1 else "negative"

    return {"sentiment": sentiment_label}

def sentiment_api():
    data = request.get_json()
    text = data.get('text')
    if not text:
        return jsonify({"error": "Text is required"}), 400
    output = analyze_sentiment(text)
    if "error" in output:
        return jsonify({"error": output["error"]}), 500
    return jsonify(output)