fix: gr.Plot.select n'existe pas en Gradio 6 — UI grille de boutons (8 az + 4 el + 3 di)

app.py

@@ -164,14 +164,18 @@ def infer(image, az, el, di, seed, randomize_seed):
 
 # ── UI Gradio ─────────────────────────────────────────────────────────────────
 
-with gr.Blocks(title="Angle Studio", theme=gr.themes.Base()) as demo:
+AZ_LABELS = {0: "⬆ Front", 45: "↗ Front-R", 90: "➡ Right", 135: "↘ Back-R",
+             180: "⬇ Back", 225: "↙ Back-L", 270: "⬅ Left", 315: "↖ Front-L"}
+EL_LABELS = {-30: "⬇ Low (-30°)", 0: "➡ Eye (0°)", 30: "⬈ Elev (+30°)", 60: "⬆ High (+60°)"}
+DI_LABELS = {0.6: "🔍 Close-up", 1.0: "📷 Medium", 1.8: "🌄 Wide"}
+
+with gr.Blocks(title="Angle Studio") as demo:
     gr.Markdown("""
 # 🎥 Angle Studio
-**Cliquez un point sur le diagramme pour choisir l'angle, puis générez.**
-*Click a dot on the diagram to pick the angle, then generate.*
+**Cliquez 1 azimut, 1 élévation, 1 distance — puis générez.**
+*Pick one azimuth, one elevation, one distance — then generate.*
 """)
 
-    # États courants
     az_state = gr.State(0)
     el_state = gr.State(0)
     di_state = gr.State(1.0)
@@ -180,11 +184,23 @@ with gr.Blocks(title="Angle Studio", theme=gr.themes.Base()) as demo:
         with gr.Column(scale=1):
             input_image = gr.Image(label="Image source / Source image", type="pil")
 
-            gr.Markdown("**Distance**")
+            gr.Markdown("### 🧭 Azimut")
+            with gr.Row():
+                az_buttons = [gr.Button(AZ_LABELS[a], size="sm",
+                                        variant=("primary" if a == 0 else "secondary"))
+                              for a in AZIMUTHS]
+
+            gr.Markdown("### 📐 Élévation")
+            with gr.Row():
+                el_buttons = [gr.Button(EL_LABELS[e], size="sm",
+                                        variant=("primary" if e == 0 else "secondary"))
+                              for e in ELEVATIONS]
+
+            gr.Markdown("### 🔭 Distance")
             with gr.Row():
-                btn_close  = gr.Button("🔍 Close-up (0.6)", size="sm")
-                btn_medium = gr.Button("📷 Medium (1.0)",   size="sm", variant="primary")
-                btn_wide   = gr.Button("🌄 Wide (1.8)",     size="sm")
+                di_buttons = [gr.Button(DI_LABELS[d], size="sm",
+                                        variant=("primary" if d == 1.0 else "secondary"))
+                              for d in DISTANCES]
 
             prompt_preview = gr.Textbox(
                 label="Prompt", interactive=False,
@@ -196,7 +212,7 @@ with gr.Blocks(title="Angle Studio", theme=gr.themes.Base()) as demo:
             generate_btn = gr.Button("▶ Générer / Generate", variant="primary", size="lg")
 
         with gr.Column(scale=1):
-            viewer = gr.Plot(value=build_viewer(0, 0, 1.0), label=None, show_label=False)
+            viewer = gr.Plot(value=build_viewer(0, 0, 1.0), show_label=False)
             output_image = gr.Image(label="Résultat / Result", type="pil")
             output_seed  = gr.Number(label="Seed utilisé", interactive=False)
 
@@ -204,43 +220,32 @@ with gr.Blocks(title="Angle Studio", theme=gr.themes.Base()) as demo:
     gallery        = gr.Gallery(columns=4, height=260)
     session_images = gr.State([])
 
-    # ── Click sur le viewer → set az + el ────────────────────────────────────
-    def on_pick(di, evt: gr.SelectData):
-        # Le clic peut tomber sur n'importe quelle trace : on ne traite que la
-        # trace des 32 poses (curveNumber ≥ 6 : 4 cercles + sujet + labels).
-        idx = evt.index
-        if idx is None or not isinstance(idx, int):
-            return gr.update(), gr.update(), gr.update(), gr.update()
-        # On accepte uniquement les index 0..31 venant de la trace "poses".
-        # evt.value[2] contient le hovertext si on veut filtrer plus finement,
-        # mais l'index suffit dans la trace cliquée.
-        n = len(AZIMUTHS) * len(ELEVATIONS)
-        if idx < 0 or idx >= n:
-            return gr.update(), gr.update(), gr.update(), gr.update()
-        az = AZIMUTHS[idx // len(ELEVATIONS)]
-        el = ELEVATIONS[idx % len(ELEVATIONS)]
-        return az, el, build_viewer(az, el, di), build_prompt(az, el, di)
-
-    viewer.select(
-        fn=on_pick,
-        inputs=[di_state],
-        outputs=[az_state, el_state, viewer, prompt_preview],
-    )
+    # ── Handlers : chaque bouton met à jour son état + viewer + prompt ───────
+    def make_az_handler(a):
+        def _fn(el, di):
+            return a, build_viewer(a, el, di), build_prompt(a, el, di)
+        return _fn
+
+    def make_el_handler(e):
+        def _fn(az, di):
+            return e, build_viewer(az, e, di), build_prompt(az, e, di)
+        return _fn
 
-    # ── Boutons distance ───────────────────────────────────────────��──────────
-    def set_distance(d):
+    def make_di_handler(d):
         def _fn(az, el):
             return d, build_viewer(az, el, d), build_prompt(az, el, d)
         return _fn
 
-    btn_close.click(fn=set_distance(0.6), inputs=[az_state, el_state],
-                    outputs=[di_state, viewer, prompt_preview])
-    btn_medium.click(fn=set_distance(1.0), inputs=[az_state, el_state],
-                     outputs=[di_state, viewer, prompt_preview])
-    btn_wide.click(fn=set_distance(1.8), inputs=[az_state, el_state],
-                   outputs=[di_state, viewer, prompt_preview])
+    for a, btn in zip(AZIMUTHS, az_buttons):
+        btn.click(fn=make_az_handler(a), inputs=[el_state, di_state],
+                  outputs=[az_state, viewer, prompt_preview])
+    for e, btn in zip(ELEVATIONS, el_buttons):
+        btn.click(fn=make_el_handler(e), inputs=[az_state, di_state],
+                  outputs=[el_state, viewer, prompt_preview])
+    for d, btn in zip(DISTANCES, di_buttons):
+        btn.click(fn=make_di_handler(d), inputs=[az_state, el_state],
+                  outputs=[di_state, viewer, prompt_preview])
 
-    # ── Génération ────────────────────────────────────────────────────────────
     def run_and_append(image, az, el, di, seed, rand, history):
         result, used_seed, _ = infer(image, az, el, di, seed, rand)
         history = history + [result]
@@ -252,4 +257,4 @@ with gr.Blocks(title="Angle Studio", theme=gr.themes.Base()) as demo:
         outputs=[output_image, output_seed, session_images, gallery],
     )
 
-demo.launch()
+demo.launch(theme=gr.themes.Base())