models.py

from transformers import pipeline

def load_model():
    return pipeline("text-classification", model="tabularisai/multilingual-sentiment-analysis")

def analyze_sentiments(df, text_column, classifier):
    if text_column not in df.columns:
        raise ValueError(f"Column '{text_column}' not found in DataFrame.")
    
    texts = df[text_column].astype(str).tolist()
    try:
        sentiments = classifier(texts)
        df['sentiment'] = [result['label'] for result in sentiments]
    except Exception as e:
        print(f"Error processing text. Error: {e}")
        df['sentiment'] = 'UNKNOWN'
    return df