Update app.py

app.py

@@ -1,99 +1,123 @@
 """
-app.py — Simple Test Version
-Sri Yantra basic render — no quantum interference
-Use this to diagnose HuggingFace rendering issues
+app.py — HuggingFace Gradio Space
+Vedic Simulation Visualizer: Sri Yantra ↔ Quantum Field
+Gradio 6.0 compatible
 """
 
 import matplotlib
 matplotlib.use('Agg')
 
-import numpy as np
-import matplotlib.pyplot as plt
 import gradio as gr
+import matplotlib.pyplot as plt
 import io
 from PIL import Image
 
-
-def equilateral_triangle(cx, cy, r, pointing='up'):
-    offset = np.pi if pointing == 'down' else 0
-    angles = [np.pi/2 + offset + i * 2*np.pi/3 for i in range(3)]
-    return np.array([(cx + r*np.cos(a), cy + r*np.sin(a)) for a in angles])
-
-
-def draw_simple_yantra(phase_shift=0.0):
-    """Simple Sri Yantra — no quantum overlay, basic triangles only."""
-    fig, ax = plt.subplots(figsize=(6, 6), facecolor='#0d0010')
-    ax.set_facecolor('#0d0010')
-    ax.set_aspect('equal')
-    ax.axis('off')
-    ax.set_xlim(-1.5, 1.5)
-    ax.set_ylim(-1.5, 1.5)
-
-    # Outer circle
-    t = np.linspace(0, 2*np.pi, 200)
-    ax.plot(np.cos(t)*1.2, np.sin(t)*1.2, color='#DAA520', linewidth=1.0)
-
-    # Shiva triangles (orange, pointing up)
-    shiva = [(0.00,0.95),(0.10,0.60),(-0.05,0.38),(0.08,0.22)]
-    for cy, r in shiva:
-        v = equilateral_triangle(0, cy, r, 'up')
-        poly = plt.Polygon(v, fill=True, facecolor='#FF6B35',
-                           alpha=0.25, edgecolor='#FF6B35', linewidth=1.5)
-        ax.add_patch(poly)
-
-    # Shakti triangles (pink, pointing down)
-    shakti = [(-0.08,0.85),(0.00,0.65),(-0.10,0.48),(0.05,0.32),(-0.02,0.18)]
-    for cy, r in shakti:
-        v = equilateral_triangle(0, cy, r, 'down')
-        poly = plt.Polygon(v, fill=True, facecolor='#C2185B',
-                           alpha=0.25, edgecolor='#C2185B', linewidth=1.5)
-        ax.add_patch(poly)
-
-    # Bindu
-    ax.add_patch(plt.Circle((0, 0), 0.04, color='white', zorder=10))
-    ax.add_patch(plt.Circle((0, 0), 0.10, color='#FFD700', alpha=0.3, zorder=9))
-
-    # Title
-    fig.text(0.5, 0.96, 'SRI YANTRA — QUANTUM FIELD',
-             ha='center', color='#FFD700', fontsize=11,
-             fontweight='bold', fontfamily='monospace')
-    fig.text(0.5, 0.02, f'Phase: {phase_shift:.2f} | vedic-logic.blogspot.com',
-             ha='center', color='#888888', fontsize=7, fontfamily='monospace')
-
-    plt.tight_layout(pad=0)
-
+from sri_yantra.visualizer import draw_sri_yantra
+from sri_yantra.quantum_mapping import (
+    all_qubit_states, bindu_ground_state
+)
+
+
+def generate_sri_yantra(phase_shift, show_quantum, show_entanglement, show_labels):
+    """Gradio handler: render Sri Yantra and return as PIL image."""
+    fig = draw_sri_yantra(
+        phase_shift=phase_shift,
+        show_quantum=show_quantum,
+        show_entanglement=show_entanglement,
+        show_labels=show_labels,
+        figsize=(8, 8)
+    )
     buf = io.BytesIO()
-    fig.savefig(buf, format='png', dpi=100,
-                bbox_inches='tight', facecolor='#0d0010')
+    fig.savefig(buf, format='png', dpi=120, bbox_inches='tight', facecolor='#0d0010')
     buf.seek(0)
     plt.close(fig)
     return Image.open(buf)
 
 
+def qubit_table(phase_shift):
+    """Return markdown table of all 9 qubit states."""
+    states = all_qubit_states(phase_shift)
+    rows = [
+        "| Layer | Type | P(0) Shiva | P(1) Shakti | Vedic Name |",
+        "|-------|------|-----------|------------|------------|"
+    ]
+    vedic_names = [
+        "Sarva Siddhi Prada", "Sarva Shakti Mayi", "Sarva Raksha Kara",
+        "Sarva Rogahara", "Sarva Siddhimaya", "Sarva Anandamaya",
+        "Sarva Raksha Kara II", "Sarva Siddhiprada", "Bindu Mandala"
+    ]
+    for i, s in enumerate(states):
+        name = vedic_names[i] if i < len(vedic_names) else "-"
+        tri_type = "Shiva" if s['type'] == 'shiva' else "Shakti"
+        rows.append(
+            f"| {s['layer']} | {tri_type} "
+            f"| {s['prob_0']:.3f} | {s['prob_1']:.3f} | {name} |"
+        )
+    alpha, beta = bindu_ground_state(phase_shift)
+    rows.append(
+        f"\n**Bindu Ground State:** P(0) = {abs(alpha)**2:.3f}, "
+        f"P(1) = {abs(beta)**2:.3f}"
+    )
+    return "\n".join(rows)
+
+
+# ── Gradio UI ──────────────────────────────────────────────────────────────────
+
 with gr.Blocks(title="Vedic Simulation Visualizer") as demo:
 
     gr.Markdown("""
     # 🕉 Sri Yantra — Quantum Field Visualizer
     **vedic-logic.blogspot.com** | Branch 2: Simulation Theory
+
+    The Sri Yantra encodes a **9-qubit quantum register** in sacred geometry.
+    Adjust the quantum phase shift to animate interference patterns.
     """)
 
     with gr.Row():
         with gr.Column(scale=1):
-            phase = gr.Slider(0, 6.28, value=0.0, step=0.1,
-                              label="Phase Shift")
-            btn = gr.Button("🔮 Render Yantra", variant="primary")
+            phase = gr.Slider(
+                0, 6.28, value=0.0, step=0.05,
+                label="Quantum Phase Shift (0 to 2pi radians)"
+            )
+            show_q = gr.Checkbox(value=True,  label="Show Interference Field")
+            show_e = gr.Checkbox(value=True,  label="Show Entanglement Nodes")
+            show_l = gr.Checkbox(value=True,  label="Show Labels")
+            btn    = gr.Button("🔮 Render Yantra", variant="primary")
+
         with gr.Column(scale=2):
-            image_out = gr.Image(label="Sri Yantra", type="pil")
+            image_out = gr.Image(label="Sri Yantra Visualization", type="pil")
+
+    qubit_md = gr.Markdown()
 
-    btn.click(fn=draw_simple_yantra, inputs=[phase], outputs=image_out)
+    btn.click(
+        fn=generate_sri_yantra,
+        inputs=[phase, show_q, show_e, show_l],
+        outputs=image_out
+    )
+    btn.click(
+        fn=qubit_table,
+        inputs=[phase],
+        outputs=qubit_md
+    )
 
     gr.Markdown("""
     ---
-    | Vedic | Quantum |
+    ### Vedic to Quantum Mapping
+
+    | Vedic Concept | Quantum Equivalent |
     |---|---|
-    | Bindu | Wave function collapse |
-    | 4 Shiva triangles | Qubit 0 basis |
-    | 5 Shakti triangles | Qubit 1 basis |
+    | Bindu (central point) | Ground state — wave function collapse |
+    | 4 Shiva triangles (up) | Qubit in 0 basis |
+    | 5 Shakti triangles (down) | Qubit in 1 basis |
+    | Intersection points | Entanglement nodes |
+    | Phase shift slider | Time evolution unitary operator |
+    | Sri Yantra whole | Hilbert space manifold |
+
+    ---
+    Blog: vedic-logic.blogspot.com |
+    GitHub: github.com/kalpeshnitore/vedic-simulation
     """)
 
+
 demo.launch()
+