Update app.py

app.py

@@ -2,7 +2,6 @@ from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from transformers import pipeline
 from typing import List
-import re
 
 app = FastAPI(title="NER + Emotion API")
 
@@ -48,7 +47,7 @@ def home():
 
 
 # ---------------------------------------------------------
-# NER ENDPOINT (FAST)
+# NER ENDPOINT
 # ---------------------------------------------------------
 @app.post("/analyze/ner")
 def analyze_ner(data: TextInput):
@@ -61,14 +60,14 @@ def analyze_ner(data: TextInput):
 
         for entity in results:
             label = entity["entity_group"]
-            text = entity["word"]
+            word = entity["word"]
 
             if label == "PER":
-                persons.append(text)
+                persons.append(word)
             elif label == "LOC":
-                locations.append(text)
+                locations.append(word)
             elif label == "ORG":
-                organizations.append(text)
+                organizations.append(word)
 
         return {
             "persons": list(set(persons)),
@@ -115,7 +114,7 @@ def analyze_sentiment(data: SentimentInput):
         return {"results": processed_results}
 
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e)}
+        raise HTTPException(status_code=500, detail=str(e))
 
 
 if __name__ == "__main__":