Update app.py

app.py

@@ -156,9 +156,12 @@ Non-organised entities with entiities.json
         entities = self.entity_df.entity
         self.related_entity = []
         match_scores = []
-        for name, entity in zip(names, entities):
+        for i, (name, entity) in enumerate(zip(names, entities)):
             if entity in ('PERSON','ORG','GPE','NORP','LOC'):
                 related_names = wikipedia.search(name, 10)
+                # Implementing logic for getting related names instead of original text
+                if related_names:
+                    names.iloc[i] = related_names[0]
                 self.related_entity.append(related_names)
                 matches = process.extract(name, related_names)
                 match_scores.append([match[0] for match in matches if match[1]>= 90 ])
@@ -170,6 +173,7 @@ Non-organised entities with entiities.json
         self.entity_df['Wikipedia Entity'] = self.related_entity
         self.entity_df['Matched Entity'] = match_scores
 
+
     def fuzzy_disambiguation(self):
         # Load the entity data
         self.entity_df['fuzzy_match'] = ''