Update app.py

app.py

@@ -198,7 +198,7 @@ def generate_images(prompt, quantization_choice, progress=gr.Progress(track_tqdm
     random.shuffle(shuffled_results)
     shuffled_data_for_gallery = [(res["image"], f"Image {i+1}") for i, res in enumerate(shuffled_results)]
     correct_mapping = {i: res["label"] for i, res in enumerate(shuffled_results)}
-    # print("Correct mapping (hidden):", correct_mapping)
+    print("Correct mapping (hidden):", correct_mapping)
 
     return shuffled_data_for_gallery, correct_mapping, prompt, seed, results, "Generation complete! Make your guess.", None, gr.update(interactive=True), gr.update(interactive=True)
 
@@ -233,12 +233,14 @@ EXAMPLES = [
         "files": ["astronauts_seed_6456306350371904162.png", "astronauts_bnb_8bit.png"],
         "quantized_idx": 1,
         "quant_method": "8-bit bnb",
+        "summary": "Astronaut on Mars",
     },
     {
         "prompt": "Water-color painting of a cat wearing sunglasses",
         "files": ["watercolor_cat_bnb_8bit.png", "watercolor_cat_seed_14269059182221286790.png"],
         "quantized_idx": 0,
         "quant_method": "8-bit bnb",
+        "summary": "Cat with Sunglasses",
     },
     # {
     #     "prompt": "Neo-tokyo cyberpunk cityscape at night, rain-soaked streets, 8-K",
@@ -337,7 +339,7 @@ with gr.Blocks(title="FLUX Quantization Challenge", theme=gr.themes.Soft()) as d
 
             gr.Markdown("### Examples")
             ex_selector = gr.Radio(
-                choices=[f"Example {i+1}" for i in range(len(EXAMPLES))],
+                choices=[ex["summary"] for ex in EXAMPLES],
                 label="Choose an example prompt",
                 interactive=True,
             )
@@ -403,8 +405,15 @@ with gr.Blocks(title="FLUX Quantization Challenge", theme=gr.themes.Soft()) as d
             seed_state = gr.State(None)
             results_state = gr.State([])
 
-            def _load_example_and_update_dfs(sel):
-                idx = int(sel.split()[-1]) - 1
+            def _load_example_and_update_dfs(sel_summary):
+                # Find the index of the selected example by its summary
+                idx = next((i for i, ex in enumerate(EXAMPLES) if ex["summary"] == sel_summary), -1)
+                if idx == -1:
+                    # Fallback or error handling if summary not found
+                    print(f"Error: Example with summary '{sel_summary}' not found.")
+                    return (gr.update(), gr.update(), gr.update(), False, gr.update(), gr.update(), "", None, [])
+
+                ex = EXAMPLES[idx]
                 gallery_items, mapping, prompt = load_example(idx)
                 quant_data, overall_user_data, _ = update_leaderboards_data()
                 return gallery_items, mapping, prompt, True, quant_data, overall_user_data, "", None, []