refactoring

app.py

@@ -148,35 +148,37 @@ def generate_clue_callback(group: List[str]) -> gr.update:
     clue = generate_clue(group)
     return gr.update(value=clue['clue'], info=clue['explanation'])
 
-# 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)
+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")
 
-    with gr.Row():
-        detect_words_button = gr.Button("Detect Words")
-        group_words_button = gr.Button("Group Words")
-
-    group_inputs, clue_buttons, clue_outputs = [], [], []
+        with gr.Row():
+            game_image = gr.Image(type="pil")
+            word_list_input = gr.Dropdown(label="Detected words", choices=[], multiselect=True, interactive=True)
 
-    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)
+            detect_words_button = gr.Button("Detect Words")
+            group_words_button = gr.Button("Group Words")
+
+        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)
 
-    # 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)