Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import numpy as np
-import plotly.graph_objects as go
+import matplotlib.pyplot as plt
+import seaborn as sns
 
 st.title('Plant Fractal')
 
@@ -16,15 +17,16 @@ def generate_strange_attractor(num_points, a, b, c, d):
         z += 0.1 * z_dot
         points.append((x, y, z))
     x, y, z = zip(*points)
-    return go.Scatter3d(x=x, y=y, z=z, mode='lines', line=dict(width=1))
+    return (x, y, z)
 
 def generate_julia_set(num_points, c):
     def f(z, c):
         return z**2 + c
     x, y, z = np.zeros(num_points), np.zeros(num_points), np.zeros(num_points)
+    z[0] = 1
     for i in range(1, num_points):
         x[i], y[i], z[i] = f((x[i-1], y[i-1], z[i-1]), c)
-    return go.Scatter3d(x=x, y=y, z=z, mode='lines', line=dict(width=1))
+    return (x, y, z)
 
 num_points = st.slider('How many points do you want to generate?', 1000, 100000, 10000)
 fractal_type = st.selectbox('Select a fractal type', ('Strange Attractor', 'Julia Set'))
@@ -34,27 +36,25 @@ if fractal_type == 'Strange Attractor':
     b = st.slider('b', 0.0, 2.0, 0.6)
     c = st.slider('c', 0.0, 2.0, 1.7)
     d = st.slider('d', 0.0, 2.0, 1.5)
-    fig = go.Figure(generate_strange_attractor(num_points, a, b, c, d))
-    fig.update_layout(
-        title='Strange Attractor Fractal',
-        scene=dict(
-            xaxis_title='X',
-            yaxis_title='Y',
-            zaxis_title='Z'
-        ),
-        showlegend=False
-    )
+    x, y, z = generate_strange_attractor(num_points, a, b, c, d)
+    fig = plt.figure()
+    ax = fig.add_subplot(111, projection='3d')
+    sns.lineplot(x=x, y=y, z=z, ax=ax, linewidth=1)
+    ax.set_title('Strange Attractor Fractal')
+    ax.set_xlabel('X')
+    ax.set_ylabel('Y')
+    ax.set_zlabel('Z')
+    st.pyplot(fig)
 else:
-    c = complex(st.slider('Real part of c', -2.0, 2.0, 0.4), st.slider('Imaginary part of c', -2.0, 2.0, 0.1))
-    fig = go.Figure(generate_julia_set(num_points, c))
-    fig.update_layout(
-        title='Julia Set Fractal',
-        scene=dict(
-            xaxis_title='X',
-            yaxis_title='Y',
-            zaxis_title='Z'
-        ),
-        showlegend=False
-    )
-
-st.plotly_chart(fig)
+    real_part = st.slider('Real part of c', -2.0, 2.0, 0.4)
+    imag_part = st.slider('Imaginary part of c', -2.0, 2.0, 0.1)
+    c = complex(real_part, imag_part)
+    x, y, z = generate_julia_set(num_points, c)
+    fig = plt.figure()
+    ax = fig.add_subplot(111, projection='3d')
+    sns.lineplot(x=x, y=y, z=z, ax=ax, linewidth=1)
+    ax.set_title('Julia Set Fractal')
+    ax.set_xlabel('X')
+    ax.set_ylabel('Y')
+    ax.set_zlabel('Z')
+    st.pyplot(fig)