Update text_converter.py

text_converter.py

@@ -22,21 +22,43 @@ def generate_user_prompt(prompt_type, base_text):
 
     return prompts[prompt_type].format(base_text=base_text)
 
+reading_levels = {
+    "5th Grade (90-100)": (90, 100),
+    "6th Grade (80-90)": (80, 90),
+    "7th Grade (70-80)": (70, 80),
+    "8th - 9th Grade (60-70)": (60, 70),
+    "10th - 12th Grade (50-60)": (50, 60),
+    "College (30-50)": (30, 50),
+    "College Graduate + Professionals (0-30)": (0, 30)
+}
 
-def generate_similar_sentence(input_text, min_reading_level, max_reading_level, min_entailment, system_prompt, max_iter):
+inverse_reading_levels = {v: k for k, v in reading_levels.items()}
+
+def user_input_readability_level(input_text):
+    current_score = flesch_reading_ease(input_text)
+    print(f'Reading score for user input is: {current_score}')
+    current_level = ''
+    for (min, max), level in inverse_reading_levels.items():
+        if min <= current_score <= max:
+            print(f'Reading level for user input is: {level}')
+            current_level = level
+            break
+    return current_score, current_level
+
+def generate_similar_sentence(input_text, min_reading_level, max_reading_level, min_entailment, system_prompt, max_iter, curr_reading_level):
     i = 0
     completed = False
     user_prompt = ""
-    curr_reading_level = flesch_reading_ease(input_text)
-    input_reading_level = flesch_reading_ease(input_text)
     curr_text = input_text
     response = None
     similarity = 0
     reading_level = 0
-    print(f"Current reading level is: {curr_reading_level}")
+    generated_texts = []
+    result_index = -1
+    closeness = float('inf')
 
-    if curr_reading_level > min_reading_level and curr_reading_level < max_reading_level:
-        return input_text, 1, curr_reading_level, input_reading_level, "Input text was already within the target reading level!"
+    if min_reading_level < curr_reading_level < max_reading_level:
+        return input_text, 1, curr_reading_level, "Input text was already within the target reading level!"
 
     else:
         while i < max_iter and not completed:
@@ -49,20 +71,39 @@ def generate_similar_sentence(input_text, min_reading_level, max_reading_level,
             elif similarity < min_entailment:
                 print(f"Entailment level is too low: {similarity}")
                 user_prompt = f"Can you convert this text '{input_text}' to a grade level more similar to this text '{curr_text}'"
+            else: 
+                print(f"Reading level is within target range: {curr_reading_level}")
+                completed = True
+                break
             
             response = get_gpt_response(user_prompt, system_prompt)
+
+            # First we add the generated text to the list of generated texts.
+            generated_texts.append(response)
+
             similarity = get_similarity(response, input_text)
             reading_level = flesch_reading_ease(response)
 
+            # Determine the closeness of the reading level to the target reading level and store.
             if similarity >= min_entailment and min_reading_level <= reading_level <= max_reading_level:
+                result_index = i
                 completed = True
+                break
+            elif (reading_level < min_reading_level and abs(reading_level - min_reading_level) < closeness) \
+                or (reading_level > max_reading_level and abs(reading_level - max_reading_level) < closeness):
+                closeness = abs(reading_level - min_reading_level) if reading_level < min_reading_level else abs(reading_level - max_reading_level)
+                result_index = i
 
             curr_text = response
             curr_reading_level = reading_level
-            print(response)
             i += 1
 
+        # Printing all generated texts to console.
+        for i, text in enumerate(generated_texts):
+            print(f"Generated text {i}: {text}")
+
+        # Returning the final result.
         if completed:
-            return response, similarity, reading_level, input_reading_level, "Success! Please see the converted text at your target reading level."
+            return generated_texts[result_index], similarity, reading_level, "Success! Please see the converted text at your target reading level."
         else:
-            return response, similarity, reading_level, input_reading_level, "Failed. We could not reach the target reading level while maintaining the text meaning."
+            return generated_texts[result_index], similarity, reading_level, "Failed. We could not reach the target reading level while maintaining the text meaning."