From 4c5bc9d059e4ff1cadba9fadbea88f41976e1bf5 Mon Sep 17 00:00:00 2001
From: aurelien <aurelien.pintrand@etu.uca.fr>
Date: Tue, 13 Feb 2024 15:03:15 +0100
Subject: [PATCH] ajout recherche plus rapide

---
 interface.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/interface.py b/interface.py
index b026d51..42ee64c 100644
--- a/interface.py
+++ b/interface.py
@@ -26,6 +26,9 @@ def levenshtein_distance(s1, s2):
 
 
 def find_closest_actor_name(input_name, actor_names):
+    if input_name in actor_names:
+        return input_name
+        
     closest_name = None
     min_distance = float('inf')
     for i in range(len(actor_names)):
@@ -81,6 +84,12 @@ def add_actor(actor_names):
             return None
 
 
+def find_direct_actor_name(df, searchedValue):
+    if 'primaryName' in df.columns:
+        if searchedValue in df['primaryName'].values:
+            return searchedValue
+    return None
+
 if __name__ == "__main__":
     mustContinue = True
     if not os.path.exists("processedData/uniqueActorNames.tsv"):