visual hinge and slider

app.py

@@ -4,112 +4,145 @@ import json
 import plotly.graph_objects as go
 
 # -----------------------------
-# Data path
+# DATA PATHS (tes "documents 000")
 # -----------------------------
-DATA_PATH = "000_hinge_0/000_hinge_0_Gen175_Pop10000_baseline_MAP-Elite-Explore_APGen_initState_max_v0/Archive_space_Object_000_JointNature_hinge_Item0_Generation_175_PopSize_10000_baseline_MAP-Elite-Explore_APGen_initState_max_v0.jsonl"
+DATA_PATH_1 = "000_hinge_0/000_hinge_0_Gen175_Pop10000_baseline_MAP-Elite-Explore_APGen_initState_max_v0/Archive_space_Object_000_JointNature_hinge_Item0_Generation_175_PopSize_10000_baseline_MAP-Elite-Explore_APGen_initState_max_v0.jsonl"
+
+DATA_PATH_2 = "000_slider_0/000_slider_0_Gen175_Pop10000_baseline_MAP-Elite-Explore_APGen_initState_max_v0/Archive_space_Object_000_JointNature_slider_Item0_Generation_175_PopSize_10000_baseline_MAP-Elite-Explore_APGen_initState_max_v0.jsonl"
+
 
 # -----------------------------
-# Load trajectories
+# LOAD
 # -----------------------------
 def load_all_trajectories(path):
     traj_list = []
-    count =0
-    nb_traj_stored = 0
+    count = 0
+
     with open(path, "r") as f:
         for line in f:
-            if count %50==0:
+            if count % 50 == 0:
                 obj = json.loads(line)
                 traj = obj.get("traj_list")
-                if traj is None:
-                    continue
-                traj = [x[:3] for x in traj]
-                traj_list.append(traj)
-                nb_traj_stored += 1
+
+                if traj is not None:
+                    traj = [x[:3] for x in traj]
+                    traj_list.append(traj)
+
             count += 1
-    print(f"Number of trajectories stored: {nb_traj_stored}")
+
     return traj_list
 
-def generate_trajectories():
-    trajs = load_all_trajectories(DATA_PATH)
+
+def build_dict(path):
+    trajs = load_all_trajectories(path)
     return {f"Trajectory {i}": traj for i, traj in enumerate(trajs)}
 
-TRAJECTORIES = generate_trajectories()
+
+TRAJ_1 = build_dict(DATA_PATH_1)
+TRAJ_2 = build_dict(DATA_PATH_2)
+
+choices_1 = list(TRAJ_1.keys())
+choices_2 = list(TRAJ_2.keys())
+
 
 # -----------------------------
-# Plot function
+# PLOT FUNCTION (generic)
 # -----------------------------
-def update_plot(choice, progress):
+def plot_traj(trajs_dict, choice, progress, title):
+
     fig = go.Figure()
 
-    for name, traj in TRAJECTORIES.items():
+    for name, traj in trajs_dict.items():
         traj = np.array(traj)
 
         if len(traj) == 0:
             continue
 
-        # other trajectories
+        # background
         if name != choice:
             fig.add_trace(go.Scatter3d(
                 x=traj[:, 0],
                 y=traj[:, 1],
                 z=traj[:, 2],
                 mode="lines",
-                line=dict(width=4, color="red"),
-                opacity=0.1,
+                line=dict(width=4, color="green"),
+                opacity=0.05,
                 showlegend=False
             ))
 
-        # selected trajectory (animated by slider)
+        # selected (progressive)
         else:
-            n = len(traj)
-            k = max(2, int(n * progress))
-            traj_partial = traj[:k]
+            k = max(2, int(len(traj) * progress))
+            traj_part = traj[:k]
 
             fig.add_trace(go.Scatter3d(
-                x=traj_partial[:, 0],
-                y=traj_partial[:, 1],
-                z=traj_partial[:, 2],
+                x=traj_part[:, 0],
+                y=traj_part[:, 1],
+                z=traj_part[:, 2],
                 mode="lines",
-                line=dict(width=6, color="green"),
+                line=dict(width=6, color="red"),
                 name=name
             ))
 
     fig.update_layout(
-        title=f"{choice} | progress={progress:.2f}",
-        scene=dict(
-            xaxis_title="X",
-            yaxis_title="Y",
-            zaxis_title="Z"
-        ),
+        title=f"{title} | {choice} | progress={progress:.2f}",
+        scene=dict(xaxis_title="X", yaxis_title="Y", zaxis_title="Z"),
         margin=dict(l=0, r=0, b=0, t=40)
     )
 
     return fig
 
+
 # -----------------------------
-# Gradio Blocks UI (LIVE)
+# WRAPPERS
 # -----------------------------
-choices = list(TRAJECTORIES.keys())
+def update_1(choice, progress):
+    return plot_traj(TRAJ_1, choice, progress, "Action primitive 000_hinge_0")
 
+def update_2(choice, progress):
+    return plot_traj(TRAJ_2, choice, progress, "Action primitive 000_slider_0")
+
+
+# -----------------------------
+# UI
+# -----------------------------
 with gr.Blocks() as demo:
 
-    gr.Markdown("# 🔴 3D Trajectory Viewer (LIVE SLIDER)")
+    gr.Markdown("# 🔴 Trajectory Explorer - 2 Documents (000 / 000)")
+
+    # -------------------------
+    # DOCUMENT 1
+    # -------------------------
+    gr.Markdown("## 📄 Document 000 - A")
+
+    with gr.Row():
+        dd1 = gr.Dropdown(choices_1, label="Trajectory A")
+        sl1 = gr.Slider(0, 1, value=1, step=0.01, label="Progress A")
+
+    plot1 = gr.Plot()
+
+    dd1.change(update_1, [dd1, sl1], plot1)
+    sl1.change(update_1, [dd1, sl1], plot1)
+
+    demo.load(update_1, [dd1, sl1], plot1)
+
+
+    # -------------------------
+    # DOCUMENT 2
+    # -------------------------
+    gr.Markdown("## 📄 Document 000 - B")
 
     with gr.Row():
-        traj_dropdown = gr.Dropdown(choices, label="Choose trajectory")
-        slider = gr.Slider(0, 1, value=1, step=0.01, label="Progress")
+        dd2 = gr.Dropdown(choices_2, label="Trajectory B")
+        sl2 = gr.Slider(0, 1, value=1, step=0.01, label="Progress B")
 
-    plot = gr.Plot()
+    plot2 = gr.Plot()
 
-    # live binding (IMPORTANT)
-    def refresh(choice, progress):
-        return update_plot(choice, progress)
+    dd2.change(update_2, [dd2, sl2], plot2)
+    sl2.change(update_2, [dd2, sl2], plot2)
 
-    traj_dropdown.change(refresh, [traj_dropdown, slider], plot)
-    slider.change(refresh, [traj_dropdown, slider], plot)
+    demo.load(update_2, [dd2, sl2], plot2)
 
-    # initial render
-    demo.load(refresh, [traj_dropdown, slider], plot)
 
 if __name__ == "__main__":
     demo.launch()