fix bug

app.py

@@ -34,6 +34,17 @@ EXAMPLE_CLUES = [
     (['ARCTIC', 'SHUT', 'STAMP'], 'SEAL', 'such as the Arctic being home to seals, or shutting a seal on an envelope, or a stamp being a type of seal'),
 ]
 
+def create_random_word_groups(clues: List[Tuple[List[str], str, str]], num_groups: int = 10) -> List[Tuple[List[str], List[int]]]:
+    """Creates random groups of words from the given clues."""
+    word_groups = []
+    while len(word_groups) < num_groups:
+        group_size = random.choice([3, 4])
+        selected_indices = random.sample(range(len(clues)), group_size)
+        words = [word for row in [clues[i][0] for i in selected_indices] for word in row]
+        if len(words) in [8, 9]:
+            word_groups.append((words, selected_indices))
+    return word_groups
+
 def create_example_groupings(clues: List[Tuple[List[str], str, str]], num_groups: int = 5) -> List[Tuple[List[str], str]]:
     """Creates example groupings from the given clues."""
     merged = create_random_word_groups(clues, num_groups)
@@ -51,17 +62,6 @@ def create_example_groupings(clues: List[Tuple[List[str], str, str]], num_groups
 
 EXAMPLE_GROUPINGS = create_example_groupings(EXAMPLE_CLUES)
 
-def create_random_word_groups(clues: List[Tuple[List[str], str, str]], num_groups: int = 10) -> List[Tuple[List[str], List[int]]]:
-    """Creates random groups of words from the given clues."""
-    word_groups = []
-    while len(word_groups) < num_groups:
-        group_size = random.choice([3, 4])
-        selected_indices = random.sample(range(len(clues)), group_size)
-        words = [word for row in [clues[i][0] for i in selected_indices] for word in row]
-        if len(words) in [8, 9]:
-            word_groups.append((words, selected_indices))
-    return word_groups
-
 def render_template(template: str, system_prompt: str, history: List[Tuple], query: str) -> str:
     """Renders a Jinja2 template with the given parameters."""
     env = Environment(loader=FileSystemLoader('.'))
@@ -149,36 +149,35 @@ def generate_clue_callback(group: List[str]) -> gr.update:
     return gr.update(value=clue['clue'], info=clue['explanation'])
 
 
-if __name__ == "__main__":
-    # UI Setup
-    with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
-        gr.Markdown("# *Codenames* Clue Generator")
-        gr.Markdown("Provide a list of words to generate clues")
+# UI Setup
+with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
+    gr.Markdown("# *Codenames* Clue Generator")
+    gr.Markdown("Provide a list of words to generate clues")
 
-        with gr.Row():
-            game_image = gr.Image(type="pil")
-            word_list_input = gr.Dropdown(label="Detected words", choices=[], multiselect=True, interactive=True)
+    with gr.Row():
+        game_image = gr.Image(type="pil")
+        word_list_input = gr.Dropdown(label="Detected words", choices=[], multiselect=True, interactive=True)
 
-        with gr.Row():
-            detect_words_button = gr.Button("Detect Words")
-            group_words_button = gr.Button("Group Words")
+    with gr.Row():
+        detect_words_button = gr.Button("Detect Words")
+        group_words_button = gr.Button("Group Words")
 
-        group_inputs, clue_buttons, clue_outputs = [], [], []
+    group_inputs, clue_buttons, clue_outputs = [], [], []
 
-        for i in range(4):
-            with gr.Row():
-                group_input = gr.Dropdown(label=f"Group {i + 1}", choices=[], allow_custom_value=True, multiselect=True, interactive=True)
-                clue_button = gr.Button("Generate Clue", size='sm')
-                clue_output = gr.Textbox(label=f"Clue {i + 1}")
-                group_inputs.append(group_input)
-                clue_buttons.append(clue_button)
-                clue_outputs.append(clue_output)
+    for i in range(4):
+        with gr.Row():
+            group_input = gr.Dropdown(label=f"Group {i + 1}", choices=[], allow_custom_value=True, multiselect=True, interactive=True)
+            clue_button = gr.Button("Generate Clue", size='sm')
+            clue_output = gr.Textbox(label=f"Clue {i + 1}")
+            group_inputs.append(group_input)
+            clue_buttons.append(clue_button)
+            clue_outputs.append(clue_output)
 
-        # Event handlers
-        detect_words_button.click(fn=process_image, inputs=game_image, outputs=[word_list_input])
-        group_words_button.click(fn=group_words_callback, inputs=word_list_input, outputs=group_inputs)
+    # Event handlers
+    detect_words_button.click(fn=process_image, inputs=game_image, outputs=[word_list_input])
+    group_words_button.click(fn=group_words_callback, inputs=word_list_input, outputs=group_inputs)
 
-        for i in range(4):
-            clue_buttons[i].click(generate_clue_callback, inputs=group_inputs[i], outputs=clue_outputs[i])
+    for i in range(4):
+        clue_buttons[i].click(generate_clue_callback, inputs=group_inputs[i], outputs=clue_outputs[i])
 
-    demo.launch(share=True, debug=True)
+demo.launch(share=True, debug=True)