Update pages/VIBRATION(Price&Angle).py

pages/VIBRATION(Price&Angle).py

@@ -57,7 +57,10 @@ with left:
                                     is_swing=is_swing,
                                     swing_type=(swing_type if swing_type != "" else None),
                                     swing_date=datetime.now())
-        # display
+        # persist summary + base price so it survives reruns
+        st.session_state["gann_summary"] = summary
+        st.session_state["gann_base_price"] = float(price_input)
+        # display (same as before)
         zone = summary["zone"]
         color = {"Very Close": "✅", "Near": "⚠️", "Far": "🔴"}.get(zone, "ℹ️")
         st.metric(label=f"Angle ({color} {zone})", value=f"{summary['angle']:.4f}°" if summary["angle"] is not None else "N/A",
@@ -68,6 +71,7 @@ with left:
         st.markdown("**Explanation**")
         st.write(summary["explanation"])
 
+
 # CSV batch widget
 with right:
     st.subheader("Batch: Upload CSV (Stock + Price)")
@@ -179,13 +183,15 @@ def forward_price_angle_table(base_price: float, price_list: list):
     return df
 
 # --- UI for both tables (only show if we have summary) ---
-if 'summary' in locals() and summary is not None:
+if "summary" in st.session_state and st.session_state["gann_summary"] is not None:
+    summary = st.session_state["gann_summary"]
     st.markdown("---")
     st.subheader("🔯 Square-of-9 — Reverse (Angle → Price)")
 
-    base_price = float(summary.get("price") or price_input or 0.0)
+    base_price = float(st.session_state.get("gann_base_price", summary.get("price") or price_input or 0.0))
     base_angle = float(summary.get("angle")) if summary.get("angle") is not None else None
 
+
     # Reverse table controls
     std_angles = [0,45,90,135,180,225,270,315]
     use_std = st.checkbox("Use standard harmonics for reverse table", value=True)