Update main.py

main.py

@@ -59,9 +59,26 @@ def analyze_sentiment(request: SentimentRequest):
     else:
         result = multilingual_model(text)
 
+    # Debugging: Print model output
+    print("Model Output:", result)
+
+    # Handle label mapping if model uses generic labels
+    label_map = {
+        "LABEL_0": "negative",
+        "LABEL_1": "positive",
+        "0": "negative",
+        "1": "positive",
+        "NEGATIVE": "negative",
+        "POSITIVE": "positive",
+    }
+
+    # Ensure label exists and map it correctly
+    sentiment_label = result[0]["label"].upper()  # Convert to uppercase for consistency
+    sentiment = label_map.get(sentiment_label, sentiment_label).lower()  # Default fallback
+
     return SentimentResponse(
         original_text=text,
         language_detected=language,
-        sentiment=result[0]["label"].lower(),
+        sentiment=sentiment,  # Always "positive" or "negative"
         confidence_score=result[0]["score"],
     )