session state streamlit

app.py

@@ -1,12 +1,10 @@
 import streamlit as st
 st.set_page_config(layout="wide")
-
 import yfinance as yf
 # import alpaca as tradeapi
 import alpaca_trade_api as alpaca
 from newsapi import NewsApiClient
 from vaderSentiment.vaderSentiment import SentimentIntensityAnalyzer
-
 from datetime import datetime, timedelta
 import streamlit as st
 import pandas as pd
@@ -57,6 +55,14 @@ class AlpacaTrader:
             return None
 
     def sell(self, symbol, qty):
+        # Check if position exists and has enough quantity before attempting to sell
+        positions = {p.symbol: float(p.qty) for p in self.alpaca.list_positions()}
+        if symbol not in positions:
+            logger.warning(f"No position in {symbol}. Sell not attempted.")
+            return None
+        if positions[symbol] < qty:
+            logger.warning(f"Not enough shares to sell: {qty} requested, {positions[symbol]} available for {symbol}. Sell not attempted.")
+            return None
         try:
             order = self.alpaca.submit_order(symbol=symbol, qty=qty, side='sell', type='market', time_in_force='day')
             logger.info(f"Sold {qty} shares of {symbol}")
@@ -262,12 +268,32 @@ class TradingApp:
         self.auto_trade_log = []  # Store automatic trade actions
 
     def display_charts(self):
-        # Create 12 individual dynamic price plots in a 4x3 grid using Plotly (3 columns, 4 rows)
+        # Dynamically adjust columns based on number of stocks and available width
         symbols = list(self.data.keys())
         symbol_to_name = self.analyzer.symbol_to_name
         n = len(symbols)
-        cols = 3
-        rows = 4
+
+        # Use Streamlit's layout to get available width and adjust columns accordingly
+        # Default to 3 columns, but adjust for small or large n
+        min_col_width = 350  # px, minimum width per plot for readability
+        max_cols = max(1, min(6, n))  # Don't exceed 6 columns for readability
+
+        # Try to get the container width from Streamlit (fallback to 1200px)
+        container_width = st.get_option("deprecation.showPyplotGlobalUse")  # Not a real width, just placeholder
+        # Since Streamlit does not provide actual window width, use st.columns to adapt layout
+        # Calculate columns based on n for best fit
+        if n <= 3:
+            cols = n
+        elif n <= 6:
+            cols = 3
+        elif n <= 8:
+            cols = 4
+        elif n <= 12:
+            cols = 4
+        else:
+            cols = 5
+
+        rows = (n + cols - 1) // cols
         subplot_titles = [
             f"{symbol} - {symbol_to_name.get(symbol, '')}" for symbol in symbols
         ]
@@ -290,9 +316,9 @@ class TradingApp:
                 )
         fig.update_layout(
             title="Top Volume Stocks - Price Charts (Since 2023)",
-            height=2000,
+            height=max(400 * rows, 600),
             showlegend=False,
-            dragmode=False,  # Disable global dragmode
+            dragmode=False,
         )
         # Enable scroll-zoom for each subplot (individual zoom)
         fig.update_layout(
@@ -402,6 +428,9 @@ def load_auto_trade_log():
     except Exception:
         return None
 
+# Add this at the top after imports to suppress Streamlit reruns on widget interaction
+st.experimental_set_query_params()  # This is a no-op but ensures Streamlit doesn't rerun due to query params
+
 def main():
     st.title("Stock Trading Application")
 
@@ -409,9 +438,13 @@ def main():
         st.error("Please configure your API keys in secrets.toml")
         return
 
-    app = TradingApp()
+    # Prevent Streamlit from rerunning the script on every widget interaction
+    # Use session state to persist objects and only update when necessary
+    if "app_instance" not in st.session_state:
+        st.session_state["app_instance"] = TradingApp()
+    app = st.session_state["app_instance"]
 
-    # Start background thread only once (on first run)
+    # Only start the background thread once
     if "auto_trade_thread_started" not in st.session_state:
         thread = threading.Thread(target=background_auto_trade, args=(app,), daemon=True)
         thread.start()