Update app.py

app.py

@@ -154,12 +154,13 @@ entities_data = {}
 
 if text_input and model is not None:
     try:
+        # This block for German/English spaCy is correct
         if selected_language in ["German", "English - spaCy"]:
-            # Process the text with error handling for spaCy
             doc = model(text_input)
             entities = []
             for ent in doc.ents:
                 try:
+                    # Safely get the custom attributes from entity-fishing
                     kb_qid = getattr(ent._, 'kb_qid', None)
                     url_wikidata = getattr(ent._, 'url_wikidata', None)
                     entities.append((ent.text, ent.label_, kb_qid, url_wikidata))
@@ -175,14 +176,15 @@ if text_input and model is not None:
         
         else:
             # === CORRECTED ReFinED PROCESSING LOGIC ===
-            entities = model.process_text(text_input)
-
-            # Iterate through the entity objects directly and safely
-            for entity in entities:
-                # Check if the entity has a wikidata_id before processing
-                if entity.wikidata_id:
-                    entity_text = entity.text
-                    entity_id = entity.wikidata_id
+            # The model processes the text and returns doc.ents, which is a tuple of Spans
+            spans = model.process_text(text_input)
+
+            # Iterate through the Span objects and use the correct custom attribute
+            for ent in spans:
+                # Safely check if the custom attribute exists and has a value
+                if hasattr(ent._, 'wikidata_id') and ent._.wikidata_id:
+                    entity_text = ent.text
+                    entity_id = ent._.wikidata_id  # <-- THE FIX: Access via ._.
                     entity_link = f"http://www.wikidata.org/entity/{entity_id}"
 
                     # Populate your dictionaries
@@ -191,7 +193,6 @@ if text_input and model is not None:
                     if entity_data is not None:
                         entities_data[entity_text] = entity_data
 
-
     except Exception as e:
         st.error(f"Error processing text: {e}")
         if "entityfishing" in str(e).lower():