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.9):
+def string_similarity_index(original_text, comparison_text, threshold=0.75):
     """
     Calculate the similarity index between two strings based on Levenshtein distance
     and compare it to a threshold.
@@ -67,13 +67,18 @@ def string_similarity_index(original_text, comparison_text, threshold=0.9):
     # Compare the similarity score to the threshold
     return similarity_score >= threshold
 
-def remove_suffix(input_string):
+def clean_title(input_string):
     if input_string.endswith(" | Dhar Mann"):
-        return input_string[:-12]
+        input_string = input_string[:-12]
     elif input_string.endswith(" | Dhar Mann Studios"):
-        return input_string[:-20]
-    else:
-        return input_string
+        input_string = input_string[:-20]
+    
+    # Attempt to remove all text after the first comma
+    comma_index = input_string.find(',')
+    if comma_index != -1:
+        input_string = input_string[:comma_index]
+    
+    return input_string
 
 # Function to generate an AI title
 def generate_ai_title():
@@ -86,7 +91,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(remove_suffix(generated_title), remove_suffix(title)):
+            if string_similarity_index(clean_title(generated_title), clean_title(title)):
                 is_unique = False
                 print(f"Regenerating! Generated title was: \"{generated_title}\", and the real title was \"{title}\"")
                 break