Update main.py

main.py

@@ -7,37 +7,45 @@ app = FastAPI()
 class SentencesInput(BaseModel):
     sentences: List[str]
 
+def detect_sentiment(text: str) -> str:
+    text = text.lower()
+
+    happy_words = [
+        "love", "like", "enjoy", "great", "awesome", "amazing", "good", "fantastic", 
+        "best", "wonderful", "happy", "pleased", "delight", "joy", "joyful", "excited", 
+        "excellent", "positive", "thrilled", "glad"
+    ]
+    sad_words = [
+        "hate", "bad", "terrible", "awful", "horrible", "sad", "worst", "disappointed", 
+        "angry", "upset", "depress", "cry", "pain", "poor", "unhappy", "miserable", 
+        "lonely", "heartbroken", "sucks"
+    ]
+
+    # Rule-based detection with negations
+    for word in happy_words:
+        if word in text:
+            if "not " + word in text or "no " + word in text:
+                return "sad"
+            return "happy"
+
+    for word in sad_words:
+        if word in text:
+            if "not " + word in text or "no " + word in text:
+                return "happy"
+            return "sad"
+
+    return "neutral"
+
 @app.get("/")
 async def home():
-    return {"message": "API is running successfully!"}
+    return {"message": "Sentiment API is running successfully!"}
 
-# ✅ Add a POST handler for root (same logic as /sentiment)
+# Allow POST at both '/' and '/sentiment'
 @app.post("/")
-async def analyze_root(data: SentencesInput):
-    happy = ["love", "great", "awesome", "amazing", "good", "fantastic", "best", "wonderful", "enjoy"]
-    sad = ["hate", "bad", "terrible", "awful", "horrible", "sad", "worst", "disappointed", "angry"]
-    results = []
-    for s in data.sentences:
-        text = s.lower()
-        sentiment = "neutral"
-        if any(w in text for w in happy):
-            sentiment = "happy"
-        elif any(w in text for w in sad):
-            sentiment = "sad"
-        results.append({"sentence": s, "sentiment": sentiment})
-    return {"results": results}
-
 @app.post("/sentiment")
 async def analyze_sentiment(data: SentencesInput):
-    happy = ["love", "great", "awesome", "amazing", "good", "fantastic", "best", "wonderful", "enjoy"]
-    sad = ["hate", "bad", "terrible", "awful", "horrible", "sad", "worst", "disappointed", "angry"]
     results = []
-    for s in data.sentences:
-        text = s.lower()
-        sentiment = "neutral"
-        if any(w in text for w in happy):
-            sentiment = "happy"
-        elif any(w in text for w in sad):
-            sentiment = "sad"
-        results.append({"sentence": s, "sentiment": sentiment})
+    for sentence in data.sentences:
+        sentiment = detect_sentiment(sentence)
+        results.append({"sentence": sentence, "sentiment": sentiment})
     return {"results": results}