Update app.py

app.py

@@ -42,7 +42,7 @@ def levenshtein_distance(s1, s2):
 
     return previous_row[-1]
 
-def string_similarity_index(original_text, comparison_text, threshold=0.65):
+def string_similarity_index(original_text, comparison_text, threshold=0.605):
     """
     Calculate the similarity index between two strings based on Levenshtein distance
     and compare it to a threshold.
@@ -67,6 +67,14 @@ def string_similarity_index(original_text, comparison_text, threshold=0.65):
     # Compare the similarity score to the threshold
     return similarity_score >= threshold
 
+def remove_suffix(input_string):
+    if input_string.endswith(" | Dhar Mann"):
+        return input_string[:-12]
+    elif input_string.endswith(" | Dhar Mann Studios"):
+        return input_string[:-20]
+    else:
+        return input_string
+
 # Function to generate an AI title
 def generate_ai_title():
     while True:
@@ -78,7 +86,7 @@ def generate_ai_title():
         is_unique = True
         for title in dhar_mann_titles:
             title = title.strip()  # Remove any extra whitespace characters like newlines
-            if string_similarity_index(generated_title, title):
+            if string_similarity_index(remove_suffix(generated_title), remove_suffix(title)):
                 is_unique = False
                 print(f"Regenerating! Generated title was: \"{generated_title}\", and the real title was \"{title}\"")
                 break