Switch to Gradio version

app.py

@@ -2,22 +2,32 @@ import gradio as gr
 import matplotlib.pyplot as plt
 import numpy as np
 
-def plot_3d():
-    # Example: simple spiral
-    t = np.linspace(0, 20, 500)
+def plot_3d(max_t, num_points):
+    # Generate sequence
+    t = np.linspace(0, max_t, num_points)
     x = np.sin(t)
     y = np.cos(t)
     z = t
 
+    # Plot
     fig = plt.figure()
-    ax = fig.add_subplot(111, projection='3d')
+    ax = fig.add_subplot(111, projection="3d")
     ax.plot(x, y, z, label="3D Spiral")
     ax.legend()
 
     return fig
 
-# Build Gradio interface
-demo = gr.Interface(fn=plot_3d, inputs=None, outputs="plot")
+# Build Gradio interface with sliders
+demo = gr.Interface(
+    fn=plot_3d,
+    inputs=[
+        gr.Slider(5, 50, value=20, step=1, label="Spiral Length"),
+        gr.Slider(100, 2000, value=500, step=50, label="Number of Points")
+    ],
+    outputs="plot",
+    title="3D Hidden States Visualization",
+    description="Adjust the spiral length and number of points to explore the 3D curve."
+)
 
 if __name__ == "__main__":
     demo.launch()