Adding reading level based synonym generation

app.py

@@ -1,5 +1,6 @@
 import csv
 import string
+import json
 
 import gensim.downloader as api
 import matplotlib.pyplot as plt
@@ -405,57 +406,82 @@ def speech_to_score(speech):
 def my_i_func(text):
     return {"original": "", "interpretation": [('', 0.0), ('what', -0.2), ('great', 0.3), ('day', 0.5), ('', 0.0)]}
 
-
-inter = {"original": "what a wonderful day", "interpretation": [0, .2, .3, .5]}
+def gen_syns(word, level):
+  with open('balanced_synonym_data.json') as f:
+    data = json.loads(f.read())
+    level = str(level)
+    pins = wn_syns(word)
+    reko = []
+    for i in pins:
+      if i in data[level]:
+        reko.append(i)
+    str_reko = ""
+    for idx, i in enumerate(reko):
+      if idx != len(reko) -1:
+        str_reko+= i + ' | '
+      else:
+        str_reko+= i
+    return str_reko
 
 with gr.Blocks() as demo:
-    with gr.Column():
-        with gr.Row():
-            with gr.Box():
-                with gr.Column():
-                    with gr.Group():
-                        with gr.Tabs():
-                            with gr.TabItem("Text"):
-                                in_text = gr.Textbox(label="In Text")
-                                grade = gr.Button("Grade Your Text")
-                            with gr.TabItem("Speech"):
-                                audio_file = gr.Audio(source="microphone", type="filepath")
-                                audio_trans = gr.Textbox(label="Transcription")
-                                grade1 = gr.Button("Grade Your Speech")
-
-            with gr.Box():
-                gr.Markdown("Requires 25-50 Words")
-                diff_output = gr.Label(label='Difficulty Level', show_label=True)
-                plotter = gr.Plot()
-
-        with gr.Row():
-            with gr.Box():
-                div_output = gr.Label(label='Lexical Diversity Score', show_label=False)
-                gr.Markdown("Lexical Diversity Heatmap | Blue cells are omitted from score")
-                interpretation = gr.components.Interpretation(in_text, label="Diversity Heatmap")
-                
-            with gr.Box():
-                gr.Markdown("Relative Difficulty Heatmap")
-                interpretation2 = gr.components.Interpretation(in_text, label="Difficulty Heatmap")
+  with gr.Column():
     with gr.Row():
-        with gr.Box():
-            with gr.Group():
-                target = gr.Textbox(label="Target Text")
-            with gr.Group():
-                audio_file1 = gr.Audio(source="microphone", type="filepath")
-                b1 = gr.Button("Grade Your Pronunciation")
-        with gr.Box():
-            some_val = gr.Label()
-            text = gr.Textbox()
-            phones = gr.Textbox()
-
-    grade.click(reading_difficulty, inputs=in_text, outputs=diff_output)
-    grade.click(get_mean_score, inputs=in_text, outputs=div_output)
-    grade.click(diversity_inter, inputs=in_text, outputs=interpretation)
-    grade.click(get_dif_inter, inputs=in_text, outputs=interpretation2)
-    grade.click(get_plot, inputs=in_text, outputs=plotter)
-    grade1.click(speech_to_score, inputs=audio_file, outputs=[diff_output, audio_trans])
-    
-    
-    b1.click(speech_to_text, inputs=[audio_file1, target], outputs=[text, some_val, phones])
+      with gr.Box():
+
+        with gr.Column():
+          with gr.Group():
+            with gr.Tabs():
+              
+                with gr.TabItem("Text"):
+                    in_text = gr.Textbox(label="In Text")
+                    grade = gr.Button("Grade Your Text")
+                with gr.TabItem("Speech"):
+                    audio_file = gr.Audio(source="microphone",type="filepath")
+                    grade1 = gr.Button("Grade Your Speech")
+            with gr.Group():     
+              gr.Markdown("Reading Level Based Synonyms")
+              words = gr.Textbox(label="Word For Synonyms")
+              lvl = gr.Slider(minimum=1, maximum=4, step=1, label="Intended Reading Level For Synonym")
+              get_syns = gr.Button("Get Synonyms")
+              reccos = gr.Label()
+              
+
+      with gr.Box():
+          diff_output = gr.Label(label='Difficulty Level',show_label=True)
+          gr.Markdown("Diversity Score Across Text")
+          plotter = gr.Plot()
+
+
+
+
+    with gr.Row():
+      with gr.Box():
+        div_output = gr.Label(label='Diversity Score', show_label=False)
+        gr.Markdown("Diversity Heamap | Blue cells are omitted from score")
+        interpretation = gr.components.Interpretation(in_text, label="Diversity Heapmap")
+      with gr.Box(): 
+        interpretation2 = gr.components.Interpretation(in_text, label="Difficulty Heapmap")
+  with gr.Row():
+    with gr.Box():
+      with gr.Group():      
+        target = gr.Textbox(label="Target Text")
+      with gr.Group():      
+        audio_file1 = gr.Audio(source="microphone",type="filepath")
+        b1 = gr.Button("Grade Your Pronunciation")
+    with gr.Box():
+      some_val = gr.Label()
+      text = gr.Textbox()
+      phones = gr.Textbox()
+        
+  
+
+
+  grade.click(reading_difficulty, inputs=in_text, outputs=diff_output)
+  grade.click(get_mean_score, inputs=in_text, outputs=div_output)
+  grade.click(diversity_inter, inputs=in_text, outputs=interpretation)
+  grade.click(get_dif_inter, inputs=in_text, outputs=interpretation2)
+  grade.click(get_plot, inputs=in_text, outputs=plotter)
+  grade1.click(speech_to_score, inputs=audio_file, outputs=diff_output)
+  b1.click(speech_to_text, inputs=[audio_file1, target], outputs=[text, some_val, phones])
+  get_syns.click(gen_syns, inputs=[words, lvl], outputs=reccos)
 demo.launch(debug=True)