Update app.py

app.py

@@ -32,16 +32,64 @@ def generate_symmetrical_circle_layout(num_layers):
             ax.add_artist(circle)
     return fig
 
+def generate_fibonacci_spiral_layout(num_points):
+    fig, ax = plt.subplots(figsize=(6, 6))
+    ax.axis('off')
+    n = num_points
+    golden_ratio = (1 + np.sqrt(5)) / 2
+    radius = 0.05
+    for i in range(n):
+        angle = i * 2 * np.pi / golden_ratio
+        distance = np.sqrt(i) * radius
+        x = 0.5 + distance * np.cos(angle)
+        y = 0.5 + distance * np.sin(angle)
+        circle = plt.Circle((x, y), radius, color=np.random.rand(3,), fill=True)
+        ax.add_artist(circle)
+    ax.set_aspect('equal')
+    return fig
+
+def is_prime(n):
+    if n <= 1:
+        return False
+    for i in range(2, int(n**0.5) + 1):
+        if n % i == 0:
+            return False
+    return True
+
+def generate_prime_number_spiral(num_points):
+    fig, ax = plt.subplots(figsize=(6, 6))
+    ax.axis('off')
+    radius = 0.05
+    for i in range(1, num_points + 1):
+        if is_prime(i):
+            angle = i * 2 * np.pi / golden_ratio
+            distance = np.sqrt(i) * radius
+            x = 0.5 + distance * np.cos(angle)
+            y = 0.5 + distance * np.sin(angle)
+            circle = plt.Circle((x, y), radius, color=np.random.rand(3,), fill=True)
+            ax.add_artist(circle)
+    ax.set_aspect('equal')
+    return fig
+
 # Streamlit UI
 st.title("Circle Packings Visualization")
 
-mode = st.radio("Choose a visualization mode:", ("Random Circle Packings", "Symmetrical Circle Layouts"))
+mode = st.radio(
+    "Choose a visualization mode:",
+    ("Random Circle Packings", "Symmetrical Circle Layouts", "Fibonacci Spiral Layout", "Prime Number Spiral")
+)
 
 if mode == "Random Circle Packings":
     num_circles = st.slider("Number of Circles", 5, 50, 10)
     fig = generate_colored_circle_template(num_circles)
-    st.pyplot(fig)
 elif mode == "Symmetrical Circle Layouts":
     num_layers = st.slider("Number of Symmetrical Layers", 1, 5, 3)
     fig = generate_symmetrical_circle_layout(num_layers)
-    st.pyplot(fig)
+elif mode == "Fibonacci Spiral Layout":
+    num_points = st.slider("Number of Points", 10, 100, 30)
+    fig = generate_fibonacci_spiral_layout(num_points)
+elif mode == "Prime Number Spiral":
+    num_points = st.slider("Number of Points", 10, 1000, 200)
+    fig = generate_prime_number_spiral(num_points)
+
+st.pyplot(fig)