Create app.py

app.py

@@ -0,0 +1,60 @@
+import streamlit as st
+import numpy as np
+import plotly.graph_objects as go
+
+st.title('Plant Fractal')
+
+def generate_strange_attractor(num_points, a, b, c, d):
+    x, y, z = 0.1, 0.0, 0.0
+    points = []
+    for i in range(num_points):
+        x_dot = np.sin(y * a) - np.cos(x * b)
+        y_dot = np.sin(z * c) - np.cos(y * a)
+        z_dot = np.sin(x * d) - np.cos(z * c)
+        x += 0.1 * x_dot
+        y += 0.1 * y_dot
+        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))
+
+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)
+    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))
+
+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'))
+
+if fractal_type == 'Strange Attractor':
+    a = st.slider('a', 0.0, 2.0, 1.2)
+    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
+    )
+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)