Update app.py

app.py

@@ -20,12 +20,12 @@ def generate_strange_attractor(num_points, a, b, c, d):
 
 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
+        return z[0]**2 - z[1]**2 + c[0], 2*z[0]*z[1] + c[1]
+    x, y, z = np.zeros(num_points), np.zeros(num_points), np.zeros((num_points, 2))
+    z[0] = (1, 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 (x, y, z)
+        x[i], y[i], z[i] = f(z[i-1], c)
+    return (x, y, z[:, 0])
 
 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'))
@@ -47,13 +47,10 @@ if fractal_type == 'Strange Attractor':
 else:
     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)
+    c = (real_part, imag_part)
     x, y, z = generate_julia_set(num_points, c)
     fig = plt.figure()
     ax = fig.add_subplot(111, projection='3d')
     ax.plot(x, y, z, linewidth=1)
     ax.set_title('Julia Set Fractal')
     ax.set_xlabel('X')
-    ax.set_ylabel('Y')
-    ax.set_zlabel('Z')
-    st.pyplot(fig)