Update app.py

app.py

@@ -75,14 +75,21 @@ def string_similarity_index(original_text, comparison_text, threshold=0.6):
 
 # Function to generate an AI title
 def generate_ai_title():
-    inputs = tokenizer(["<|startoftext|>"]*1, return_tensors = "pt")
-    outputs = model.generate(**inputs, max_new_tokens=50, use_cache=True, temperature=0.85, do_sample=True)
-    generated_title = (tokenizer.batch_decode(outputs)[0])[15:-13]
-    for title in dhar_mann_titles:
-        title = title.strip()  # Remove any extra whitespace characters like newlines
-        if string_similarity_index(generated_title, title):
-            return generate_ai_title()
-    return generated_title
+    while True:
+        inputs = tokenizer(["<|startoftext|>"]*1, return_tensors="pt")
+        outputs = model.generate(**inputs, max_new_tokens=48, use_cache=True, temperature=0.85, do_sample=True)
+        generated_title = (tokenizer.batch_decode(outputs)[0])[15:-13].strip()
+
+        # Check for similarity with existing titles
+        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):
+                is_unique = False
+                break
+
+        if is_unique:
+            return generated_title
 
 # Function to check user's answer and update score
 def check_answer(user_choice, real_index, score):