from flask import Flask, request, jsonify
import pandas as pd
from transformers import pipeline
import io

app = Flask(__name__)
model = pipeline("sentiment-analysis")

@app.route("/predict", methods=["POST"])
def predict():
    data = request.json
    text = data.get("text")
    if not text:
        return jsonify({"error": "No text provided"}), 400
    result = model(text)
    return jsonify(result)

@app.route("/predict_csv", methods=["POST"])
def predict_csv():
    file = request.files.get("file")
    if not file:
        return jsonify({"error": "No CSV uploaded"}), 400
    df = pd.read_csv(file)
    df["prediction"] = df["text"].apply(lambda x: model(x)[0]['label'])
    return df.to_json(orient="records")

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