small fix

handler.py

@@ -181,9 +181,11 @@ class Transcript:
         for word in teacher_dict.keys():
             teacher_dict_list.append(
                 {'text': word, 'value': teacher_dict[word], 'category': 'general'})
+            dict_list.append({'text': word, 'value': teacher_dict[word], 'category': 'general'})
         for word in student_dict.keys():
             student_dict_list.append(
                 {'text': word, 'value': student_dict[word], 'category': 'general'})
+            dict_list.append({'text': word, 'value': student_dict[word], 'category': 'general'})
         sorted_dict_list = sorted(dict_list, key=lambda x: x['value'], reverse=True)
         sorted_uptake_dict_list = sorted(uptake_dict_list, key=lambda x: x['value'], reverse=True)
         return sorted_dict_list[:50], sorted_uptake_dict_list[:50], teacher_dict_list, student_dict_list
@@ -407,15 +409,18 @@ def run_math_density(transcript):
         utt.math_terms = match_list
     teacher_dict_list = []
     student_dict_list = []
+    dict_list = []
     for word in teacher_math_word_cloud.keys():
         teacher_dict_list.append(
             {'text': word, 'value': teacher_math_word_cloud[word], 'category': "math"})
+        dict_list.append({'text': word, 'value': teacher_math_word_cloud[word], 'category': "math"})
     for word in student_math_word_cloud.keys():
         student_dict_list.append(
-            {'text': word, 'value': student_math_word_cloud[word], 'category': "math"})        
+            {'text': word, 'value': student_math_word_cloud[word], 'category': "math"}) 
+        dict_list.append({'text': word, 'value': student_math_word_cloud[word], 'category': "math"})
     sorted_dict_list = sorted(dict_list, key=lambda x: x['value'], reverse=True)
     # return sorted_dict_list[:50]
-    return sorted_dict_list, teacher_dict_list, student_dict_list
+    return sorted_dict_list[:50], teacher_dict_list, student_dict_list
 
 class EndpointHandler():
     def __init__(self, path="."):