Fix UI state reset by using st.session_state for portfolio variables

app.py

@@ -9,6 +9,15 @@ from src.strategies.arbitrage import CrossPlatformArbitrage
 
 st.set_page_config(page_title="ArbIntel Scanner", layout="wide", page_icon="📈")
 
+if "capital" not in st.session_state:
+    st.session_state.capital = 10000.00
+if "pnl" not in st.session_state:
+    st.session_state.pnl = 0.00
+if "positions" not in st.session_state:
+    st.session_state.positions = []
+if "trades" not in st.session_state:
+    st.session_state.trades = []
+
 st.title("ArbIntel: Prediction Markets Alpha Engine")
 st.markdown("### Live Cross-Platform Arbitrage Scanner")
 st.write("Detecting price inefficiencies between Polymarket and Kalshi in real-time.")
@@ -54,6 +63,13 @@ if st.button("Run Live Arbitrage Scan", type="primary"):
             
             st.markdown("#### Execute Paper Trade")
             if st.button("Auto-Execute All"):
+                for o in opps:
+                    profit = o.expected_profit_margin * o.buy_size
+                    st.session_state.pnl += profit
+                    st.session_state.capital += profit
+                    if o.market_id_pm not in st.session_state.positions:
+                        st.session_state.positions.append(o.market_id_pm)
+                    st.session_state.trades.append(o)
                 st.info("Paper Trading Engine simulated execution successfully. Portfolios updated.")
         else:
             st.info("No active opportunities detected above risk-free threshold (markets efficient).")
@@ -61,7 +77,8 @@ if st.button("Run Live Arbitrage Scan", type="primary"):
 st.markdown("---")
 st.markdown("### Portfolio & Risk Metrics")
 col1, col2, col3, col4 = st.columns(4)
-col1.metric("Available Capital", "$10,000.00", "+$0.00")
-col2.metric("Active Positions", "0")
-col3.metric("24h PnL", "$0.00")
+pnl_sign = "+" if st.session_state.pnl >= 0 else "-"
+col1.metric("Available Capital", f"${st.session_state.capital:,.2f}", f"{pnl_sign}${abs(st.session_state.pnl):,.2f}")
+col2.metric("Active Positions", str(len(st.session_state.positions)))
+col3.metric("24h PnL", f"${st.session_state.pnl:,.2f}")
 col4.metric("Market Regime", "Stable", "-Vol")