display explanations

app.py

@@ -115,7 +115,7 @@ def group_words(words):
     print("Grouping words:", words)
 
     print("Got groupings: ", json.dumps(response, indent=4))
-    return [group["words"] for group in response]
+    return response
 
 
 def generate_clues(group):
@@ -154,7 +154,6 @@ def generate_clues(group):
 
 
 def process_image(img):
-    print(img, type(img))
     raw_response = model.generate_content(['Identify the words in this game of Codenames. Provide only a list of '
                                            'words. Provide the words in capital letters only. Group these words into '
                                            '6 or 8 groups that can be guessed together using a single word clue in '
@@ -204,10 +203,10 @@ with gr.Blocks() as demo:
 
 
     def group_words_callback(words):
-        word_groups = group_words(words)
-        word_groups = pad_or_truncate(word_groups, 4)
-        print("Got groups: ", word_groups, type(word_groups))
-        return [gr.update(value=word_groups[i], choices=words) for i in range(4)]
+        groups = group_words(words)
+        groups = pad_or_truncate(groups, 4)
+        print("Got groups: ", groups, type(groups))
+        return [gr.update(value=groups[i]["words"], choices=words, info=groups[i]["explanation"]) for i in range(4)]
 
 
     def generate_clues_callback(group):