Viewing example clusters in main view

src/view/modals.py

@@ -1,6 +1,7 @@
 from lib.snabbdom import h
 import message
 from view.modal_templates import *
+from view.utils import show_toggle_cluster_buttons
 
 
 def edit_translation(translation, cluster_idx, num_clusters, cls):
@@ -63,6 +64,13 @@ def edit_example(example):
             h("div.three-fifth", {}, middle), 
             h("div.one-fifth", {}, buttons_right(idx))]))
                 
+    cluster = example.get_cluster()
+    if cluster is not None:
+        divs.append(h("hr", {}, []))
+        divs.append(h("div.flex.five.example-component", {}, [
+            h("div.one-fifth", {}, "Cluster"), 
+            h("div.four-fifth", {}, show_toggle_cluster_buttons(list(range(10))))]))
+        
     return modal_template(divs, "Edit Example", message.EditExampleText(example_original))
 
 
@@ -92,15 +100,24 @@ def edit_comment(comment):
 
 
 def do_chosen_examples(example_list, entry):
-    # assert len(example_list) > 0
-    
-    sense_of_first_example = None
+    example_senses = []
     for idx, sense in enumerate(entry.senses):
         for ex in sense.examples:
-            if ex == example_list[0]:
-                sense_of_first_example = idx
+            if ex in example_list:
+                example_senses.append(idx)
                 break
     
+    sense_of_first_example = example_senses[0]
+    
+    # determine if one can choose clusters
+    # this can happen ef every example from same sense and every example a collocation
+    can_choose_cluster = len(set(example_senses)) == 1
+    idx = 0
+    while can_choose_cluster and idx < len(example_list):
+        can_choose_cluster = example_list[idx].is_collocation()
+        idx += 1
+    console.log(can_choose_cluster)
+    
     options = [h("p", {}, "Choose sense for examples")]
     for idx, sense in enumerate(entry.senses):
         text = "{}: {}".format(idx + 1, sense.definition["indicator"])