Update app.py

app.py

@@ -17,83 +17,122 @@ def draw_circle(ax, center, radius, color='black', fill=False, lw=1.5):
 def draw_line(ax, start, end, color='black', lw=1.5):
     ax.plot([start[0], end[0]], [start[1], end[1]], color=color, lw=lw)
 
-def setup_plot(xlim=(-1.5, 1.5), ylim=(-1.5, 1.5), aspect='equal'):
-    fig, ax = plt.subplots(figsize=(6, 6))
+def setup_plot(xlim=(-2, 2), ylim=(-2, 2), aspect='equal'):
+    fig, ax = plt.subplots(figsize=(8, 8))
     ax.set_xlim(xlim)
     ax.set_ylim(ylim)
     ax.set_aspect(aspect)
     ax.axis('off')
     return fig, ax
 
-def draw_shree_yantra(ax, radius=1, sides=12, line_skips=None, circle_fill=False, line_color='black', circle_color='black'):
+def draw_shree_yantra(ax, radius=1, sides=12, line_skips=None, circle_fill=False,
+                     line_color='black', circle_color='black', layers=2):
     if line_skips is None:
         line_skips = [2, 5, 7, 10]
 
-    # Step 1: Draw outer circle
-    draw_circle(ax, (0, 0), radius, color=circle_color, fill=circle_fill, lw=2)
-
-    # Step 2: Draw polygon (e.g., dodecagon for 12 sides)
-    points = [np.array([np.cos(theta), np.sin(theta)]) for theta in np.linspace(0, 2*np.pi, sides, endpoint=False)]
-    polygon = plt.Polygon(points, closed=True, fill=False, edgecolor=circle_color, lw=2)
-    ax.add_artist(polygon)
-
-    # Step 3: Draw connecting lines based on skips
-    for i, p1 in enumerate(points):
-        for j, p2 in enumerate(points):
-            if i != j and abs(i - j) in line_skips:
-                draw_line(ax, p1 * radius, p2 * radius, color=line_color, lw=1.5)
-
-    # Additional layers and intricacies can be added here following traditional Shree Yantra construction
+    # Step 1: Draw outer circles
+    outer_circles = [radius * (i+1) / layers for i in range(layers)]
+    for r in outer_circles:
+        draw_circle(ax, (0, 0), r, color=circle_color, fill=circle_fill, lw=2)
+
+    # Step 2: Draw polygons and connecting lines for each layer
+    for layer in range(1, layers + 1):
+        current_radius = radius * layer / layers
+        # Draw polygon
+        points = [np.array([np.cos(theta), np.sin(theta)]) for theta in np.linspace(0, 2*np.pi, sides, endpoint=False)]
+        scaled_points = [p * current_radius for p in points]
+        polygon = plt.Polygon(scaled_points, closed=True, fill=False, edgecolor=circle_color, lw=2)
+        ax.add_artist(polygon)
+
+        # Draw symmetric connecting lines
+        for i, p1 in enumerate(scaled_points):
+            for skip in line_skips:
+                j = (i + skip) % sides
+                p2 = scaled_points[j]
+                # To ensure symmetry, draw each line only once
+                if i < j:
+                    draw_line(ax, p1, p2, color=line_color, lw=1.5)
+
+    # Step 3: Draw central circle
+    draw_circle(ax, (0, 0), radius / layers, color=circle_color, fill=False, lw=2)
+
+def enforce_symmetry(selected_skips, sides):
+    """
+    Ensure that for every skip, its complementary skip is also selected.
+    For example, if skip=2 is selected in a 12-sided polygon, skip=10 (12-2) is also selected.
+    """
+    symmetric_skips = set()
+    for skip in selected_skips:
+        symmetric_skips.add(skip)
+        complement = sides - skip
+        if complement != skip:
+            symmetric_skips.add(complement)
+    return sorted(symmetric_skips)
 
 def main():
-    st.title("Shree Yantra Diagram Generator")
+    st.title("🕉️ Shree Yantra Diagram Generator")
 
-    st.sidebar.header("Customize Your Shree Yantra")
+    st.sidebar.header("🔧 Customize Your Shree Yantra")
 
     # Sidebar controls
-    radius = st.sidebar.slider("Outer Circle Radius", 0.5, 2.0, 1.0, 0.1)
-    sides = st.sidebar.slider("Number of Sides (Polygon)", 6, 20, 12, 1)
+    radius = st.sidebar.slider("🔵 Outer Circle Radius", 0.5, 3.0, 1.5, 0.1)
+    sides = st.sidebar.slider("🔺 Number of Sides (Polygon)", 6, 20, 12, 1)
+    layers = st.sidebar.slider("📚 Number of Layers", 1, 5, 2, 1)
 
     # Define default skips
     predefined_skips = [2, 5, 7, 10]
 
-    # Generate available options based on the number of sides
+    # Generate available skips based on the number of sides
     max_skip = sides // 2
     available_skips = list(range(1, max_skip + 1))
 
     # Adjust default skips to be within available options
     default_skips = [skip for skip in predefined_skips if skip in available_skips]
+    if not default_skips:
+        default_skips = [2] if 2 in available_skips else [1]
 
     # Provide user flexibility to select skips
-    line_skips = st.sidebar.multiselect(
-        "Line Skips",
+    selected_skips = st.sidebar.multiselect(
+        "🔀 Line Skips",
         options=available_skips,
         default=default_skips
     )
 
+    # Enforce symmetry in skips
+    symmetric_skips = enforce_symmetry(selected_skips, sides)
+
+    # Display selected symmetric skips
+    st.sidebar.write("**🔄 Symmetric Skips Applied:**", symmetric_skips)
+
     # Additional sidebar controls
-    line_color = st.sidebar.color_picker("Line Color", "#000000")
-    circle_color = st.sidebar.color_picker("Circle & Polygon Color", "#000000")
-    circle_fill = st.sidebar.checkbox("Fill Outer Circle", value=False)
+    line_color = st.sidebar.color_picker("✏️ Line Color", "#000000")
+    circle_color = st.sidebar.color_picker("🖌️ Circle & Polygon Color", "#000000")
+    circle_fill = st.sidebar.checkbox("🎨 Fill Outer Circles", value=False)
 
-    st.write("### Shree Yantra Diagram")
+    st.write("### 🕉️ Shree Yantra Diagram")
 
     fig, ax = setup_plot()
 
-    if not line_skips:
-        st.warning("Please select at least one line skip to draw connecting lines.")
+    if not symmetric_skips:
+        st.warning("⚠️ Please select at least one line skip to draw connecting lines.")
     else:
         draw_shree_yantra(
             ax,
             radius=radius,
             sides=sides,
-            line_skips=line_skips,
+            line_skips=symmetric_skips,
             circle_fill=circle_fill,
             line_color=line_color,
-            circle_color=circle_color
+            circle_color=circle_color,
+            layers=layers
         )
 
     st.pyplot(fig)
 
+    st.markdown("""
+    ---
+    **ℹ️ About Shree Yantra**: The Shree Yantra is a sacred geometrical diagram used in Hinduism for meditation and worship. It consists of nine interlocking triangles forming 43 smaller triangles, symbolizing the union of the divine and the material world.
+    """)
+
 if __name__ == "__main__":
     main()