test traj

app.py

@@ -1,7 +1,73 @@
 import gradio as gr
+import numpy as np
+import matplotlib.pyplot as plt
+from mpl_toolkits.mplot3d import Axes3D
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+# -----------------------------
+# Generate 3 synthetic trajectories
+# -----------------------------
+def generate_trajectories(n=200):
+    t = np.linspace(0, 10, n)
+
+    traj1 = np.vstack([
+        np.sin(t),
+        np.cos(t),
+        t * 0.2
+    ]).T
+
+    traj2 = np.vstack([
+        np.sin(t) * np.cos(t),
+        np.sin(t),
+        np.cos(t * 0.5)
+    ]).T
+
+    traj3 = np.vstack([
+        t * 0.1,
+        np.sin(t * 2),
+        np.cos(t * 2) + np.sin(t * 0.5)
+    ]).T
+
+    return {
+        "Trajectory 1 (helix-like)": traj1,
+        "Trajectory 2 (oscillating)": traj2,
+        "Trajectory 3 (mixed wave)": traj3
+    }
+
+
+TRAJECTORIES = generate_trajectories()
+
+
+# -----------------------------
+# Plot function
+# -----------------------------
+def plot_trajectory(choice):
+    traj = TRAJECTORIES[choice]
+
+    fig = plt.figure()
+    ax = fig.add_subplot(111, projection='3d')
+
+    ax.plot(traj[:, 0], traj[:, 1], traj[:, 2])
+
+    ax.set_title(choice)
+    ax.set_xlabel("X")
+    ax.set_ylabel("Y")
+    ax.set_zlabel("Z")
+
+    return fig
+
+
+# -----------------------------
+# Gradio UI
+# -----------------------------
+choices = list(TRAJECTORIES.keys())
+
+demo = gr.Interface(
+    fn=plot_trajectory,
+    inputs=gr.Dropdown(choices, label="Choose trajectory"),
+    outputs=gr.Plot(label="3D Trajectory"),
+    title="3D Trajectory Viewer",
+    description="Select one of 3 generated trajectories and visualize it in 3D."
+)
+
 demo.launch()