HakshaySundar commited on
Commit
d59d12f
1 Parent(s): 7d82239

Update text_converter.py

Browse files

Bugfixes:
1) Making reading and inverse reading dictionaries local since it was not able to access global (need to find out why)
2) converting input to int before verifying.

Files changed (1) hide show
  1. text_converter.py +7 -6
text_converter.py CHANGED
@@ -22,7 +22,9 @@ def generate_user_prompt(prompt_type, base_text):
22
 
23
  return prompts[prompt_type].format(base_text=base_text)
24
 
25
- reading_levels = {
 
 
26
  "5th Grade (90-100)": (90, 100),
27
  "6th Grade (80-90)": (80, 90),
28
  "7th Grade (70-80)": (70, 80),
@@ -30,11 +32,10 @@ reading_levels = {
30
  "10th - 12th Grade (50-60)": (50, 60),
31
  "College (30-50)": (30, 50),
32
  "College Graduate + Professionals (0-30)": (0, 30)
33
- }
34
-
35
- inverse_reading_levels = {v: k for k, v in reading_levels.items()}
36
 
37
- def user_input_readability_level(input_text):
 
38
  current_score = flesch_reading_ease(input_text)
39
  print(f'Reading score for user input is: {current_score}')
40
  current_level = ''
@@ -57,7 +58,7 @@ def generate_similar_sentence(input_text, min_reading_level, max_reading_level,
57
  result_index = -1
58
  closeness = float('inf')
59
 
60
- if min_reading_level < curr_reading_level < max_reading_level:
61
  return input_text, 1, curr_reading_level, "Input text was already within the target reading level!"
62
 
63
  else:
 
22
 
23
  return prompts[prompt_type].format(base_text=base_text)
24
 
25
+
26
+ def user_input_readability_level(input_text):
27
+ reading_levels = {
28
  "5th Grade (90-100)": (90, 100),
29
  "6th Grade (80-90)": (80, 90),
30
  "7th Grade (70-80)": (70, 80),
 
32
  "10th - 12th Grade (50-60)": (50, 60),
33
  "College (30-50)": (30, 50),
34
  "College Graduate + Professionals (0-30)": (0, 30)
35
+ }
 
 
36
 
37
+ inverse_reading_levels = {v: k for k, v in reading_levels.items()}
38
+
39
  current_score = flesch_reading_ease(input_text)
40
  print(f'Reading score for user input is: {current_score}')
41
  current_level = ''
 
58
  result_index = -1
59
  closeness = float('inf')
60
 
61
+ if min_reading_level < int(curr_reading_level) < max_reading_level:
62
  return input_text, 1, curr_reading_level, "Input text was already within the target reading level!"
63
 
64
  else: