Update app.py

app.py

@@ -6,8 +6,12 @@ from flair.data import Sentence
 
 app = FastAPI()
 
-# Ladda Flair NER-modell för svenska
-tagger = SequenceTagger.load("flair/ner-swedish")
+# Ladda Flair flerspråkig NER-modell
+try:
+    tagger = SequenceTagger.load("flair/ner-multi")
+except Exception as e:
+    print(f"Error loading model: {str(e)}")
+    raise e
 
 # Ladda entiteter från entities.json
 with open("entities.json") as f:
@@ -35,7 +39,7 @@ async def parse_user_request(request: str):
         
         # Extrahera entiteter
         result_entities = {}
-        # Först, kolla färger och priser i hela meningen
+        # Kolla färger och priser i hela meningen
         words = request.lower().split()
         for word in words:
             if word in COLORS:
@@ -45,11 +49,11 @@ async def parse_user_request(request: str):
         
         # Extrahera varor från NER
         for entity in sentence.get_spans("ner"):
-            if entity.tag in ["OBJ", "MISC", "PER"]:  # Objekt, diverse, eller potentiella produkter
+            if entity.tag in ["MISC", "ORG", "LOC"]:  # Diverse, organisationer, platser som potentiella objekt
                 corrected = correct_spelling(entity.text.lower(), ITEMS)
                 if corrected in ITEMS:
                     result_entities["VARA"] = corrected
-                elif not result_entities.get("VARA"):  # Fallback om ingen match i ITEMS
+                elif not result_entities.get("VARA"):
                     result_entities["VARA"] = entity.text.lower()
         
         # Om ingen vara hittades